mvkcontroller.xml 9.7 KB

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