mvkcontroller.xml 9.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204
  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 "value" in data:
  92. value = [json.loads(data["value"])]
  93. else:
  94. value = json.loads(data["data"])
  95. self.input_queue[data["username"]].append((source, value))
  96. elif data["op"] == "get_output":
  97. self.output_queue[data["username"]].append(source)
  98. else:
  99. raise Exception("DROPPING unknown operation: " + str(data["op"]))
  100. except ValueError:
  101. print("Error when deserializing request: " + str(data))
  102. raise
  103. </script>
  104. </transition>
  105. </state>
  106. </state>
  107. <state id="execution">
  108. <state id="execution">
  109. <onentry>
  110. <script>
  111. self.timeout = False
  112. self.destination = None
  113. if self.users:
  114. user = self.users.pop()
  115. # Check if there are values to input
  116. if self.input_queue[user]:
  117. source, args = self.input_queue[user].pop(0)
  118. for args_entry in args:
  119. self.execute_modelverse(user, "set_input", [args_entry])
  120. self.destination = source
  121. self.value = "OK"
  122. self.all_failed = False
  123. # Now process for some steps, or until we are again blocked for input
  124. for x in xrange(200):
  125. self.execute_modelverse(user, "execute_rule", [])
  126. if not self.mvk.success:
  127. # Blocking or broken, so quit already to stop wasting CPU
  128. break
  129. # Could at least execute one instruction, so mark it as "not failed"
  130. self.all_failed = False
  131. # Check that we don't have anything to output yet, otherwise we wait
  132. if self.destination is None:
  133. # Perform output if there is anything
  134. if self.output_queue[user]:
  135. self.execute_modelverse(user, "get_output", [])
  136. if self.mvk.success:
  137. self.destination = self.output_queue[user].pop(0)
  138. self.value = self.mvk.returnvalue
  139. self.all_failed = False
  140. else:
  141. if self.count >= 2000:
  142. self.count = 0
  143. self.mvs.purge()
  144. else:
  145. self.count += 1
  146. self.mvs.garbage_collect()
  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="json.dumps(self.value)"/>
  164. </raise>
  165. </transition>
  166. <transition cond="self.timeout and self.destination is None" after="self.sccd_yield() + 0.10" 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>