mvkcontroller.xml 9.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  1. <class name="MvKController">
  2. <relationships>
  3. <association name="to_mvi" class="Server" min="1" max="1"/>
  4. </relationships>
  5. <constructor>
  6. <parameter name="params"/>
  7. <body>
  8. <![CDATA[
  9. self.mvs = ModelverseState("../bootstrap/bootstrap.m")
  10. # Enable Garbage Collection
  11. self.mvs.GC = True
  12. self.root = self.mvs.read_root()[0]
  13. self.mvk = ModelverseKernel(self.root)
  14. self.all_failed = False
  15. self.timeout = False
  16. self.init_time = time.time()
  17. self.users = set()
  18. self.input_queue = defaultdict(list)
  19. self.output_queue = defaultdict(list)
  20. self.source = None
  21. self.port = int(sys.argv[1])
  22. self.mvs_operations = {
  23. "CN": self.mvs.create_node,
  24. "CE": self.mvs.create_edge,
  25. "CNV": self.mvs.create_nodevalue,
  26. "CD": self.mvs.create_dict,
  27. "RV": self.mvs.read_value,
  28. "RO": self.mvs.read_outgoing,
  29. "RI": self.mvs.read_incoming,
  30. "RE": self.mvs.read_edge,
  31. "RD": self.mvs.read_dict,
  32. "RDN": self.mvs.read_dict_node,
  33. "RDNE": self.mvs.read_dict_node_edge,
  34. "RDE": self.mvs.read_dict_edge,
  35. "RRD": self.mvs.read_reverse_dict,
  36. "RR": self.mvs.read_root,
  37. "RDK": self.mvs.read_dict_keys,
  38. "DE": self.mvs.delete_edge,
  39. "DN": self.mvs.delete_node,
  40. }
  41. self.execute_modelverse("", "load_primitives", [])
  42. ]]>
  43. </body>
  44. </constructor>
  45. <method name="execute_modelverse">
  46. <parameter name="username"/>
  47. <parameter name="operation"/>
  48. <parameter name="params"/>
  49. <body>
  50. <![CDATA[
  51. reply = None
  52. commands = []
  53. while 1:
  54. commands = self.mvk.execute_yields(username, operation, params, reply)
  55. if commands is None:
  56. self.mvs.garbage_collect()
  57. break
  58. reply = [self.mvs_operations[command[0]](*(command[1]))[0] for command in commands]
  59. ]]>
  60. </body>
  61. </method>
  62. <scxml initial="init_server">
  63. <state id="init_server">
  64. <onentry>
  65. <raise scope="cd" event="create_instance">
  66. <parameter expr="'to_mvi'"/>
  67. <parameter expr="'Server'"/>
  68. <parameter expr="''"/>
  69. <parameter expr="self.port"/>
  70. </raise>
  71. </onentry>
  72. <transition event="instance_created" target="../running">
  73. <parameter name="instancename"/>
  74. <raise scope="cd" event="start_instance">
  75. <parameter expr="instancename"/>
  76. </raise>
  77. </transition>
  78. </state>
  79. <parallel id="running">
  80. <state id="wait_for_requests">
  81. <state id="wait">
  82. <transition event="from_mvi" target=".">
  83. <parameter name="source"/>
  84. <parameter name="data"/>
  85. <script>
  86. # No JSON encoding necessary, as it is not complex
  87. try:
  88. if data["op"] == "set_input":
  89. if "element_type" in data:
  90. if data["element_type"] == "V":
  91. value = json.loads(data["value"])
  92. else:
  93. value = data["value"]
  94. args = [data["element_type"], value]
  95. self.input_queue[data["username"]].append((source, [args]))
  96. else:
  97. d = []
  98. for t, v in json.loads(data["data"]):
  99. if t == "V":
  100. v = json.loads(v)
  101. d.append([t, v])
  102. self.input_queue[data["username"]].append((source, d))
  103. elif data["op"] == "get_output":
  104. self.output_queue[data["username"]].append(source)
  105. else:
  106. raise Exception("DROPPING unknown operation: " + str(data["op"]))
  107. except ValueError:
  108. print("Error when deserializing request: " + str(data))
  109. raise
  110. </script>
  111. </transition>
  112. </state>
  113. </state>
  114. <state id="execution">
  115. <state id="execution">
  116. <onentry>
  117. <script>
  118. self.timeout = False
  119. self.destination = None
  120. if self.users:
  121. user = self.users.pop()
  122. # Check if there are values to input
  123. if self.input_queue[user]:
  124. source, args = self.input_queue[user].pop(0)
  125. for args_entry in args:
  126. self.execute_modelverse(user, "set_input", args_entry)
  127. self.destination = source
  128. self.value = {"id": 200, "value": "OK"}
  129. self.all_failed = False
  130. # Now process for some steps, or until we are again blocked for input
  131. for x in xrange(200):
  132. self.execute_modelverse(user, "execute_rule", [])
  133. if not self.mvk.success:
  134. # Blocking or broken, so quit already to stop wasting CPU
  135. break
  136. self.all_failed = False
  137. # Check that we don't have anything to output yet, otherwise we wait
  138. if self.destination is None:
  139. # Perform output if there is anything
  140. if self.output_queue[user]:
  141. self.execute_modelverse(user, "get_output", [])
  142. if self.mvk.success:
  143. self.destination = self.output_queue[user].pop(0)
  144. self.value = self.mvk.returnvalue
  145. self.all_failed = False
  146. else:
  147. out = self.mvs.read_outgoing(self.root)[0]
  148. for m in out:
  149. src, user = self.mvs.read_edge(m)[0]
  150. outgoing = self.mvs.read_outgoing(m)[0]
  151. first = self.mvs.read_edge(outgoing[0])[0]
  152. dest = first[1]
  153. name = self.mvs.read_value(dest)[0]
  154. if name.startswith("__"):
  155. continue
  156. self.users.add(name)
  157. self.timeout = self.all_failed
  158. self.all_failed = True
  159. </script>
  160. </onentry>
  161. <transition cond="self.destination is not None" after="self.sccd_yield()" target=".">
  162. <raise event="HTTP_input" scope="narrow" target="'to_mvi/%s' % self.destination">
  163. <parameter expr="self.value"/>
  164. </raise>
  165. </transition>
  166. <transition cond="self.timeout and self.destination is None" after="self.sccd_yield() + 0.05" target="."/>
  167. <transition cond="not self.timeout and self.destination is None" after="self.sccd_yield()" target="."/>
  168. </state>
  169. </state>
  170. <state id="remove_sockets">
  171. <state id="remove_sockets">
  172. <transition event="delete_socket" target=".">
  173. <parameter name="socket"/>
  174. <script>
  175. for user in self.output_queue.keys():
  176. self.output_queue[user] = [s for s in self.output_queue[user] if s != socket]
  177. </script>
  178. </transition>
  179. </state>
  180. </state>
  181. </parallel>
  182. </scxml>
  183. </class>