mvkcontroller.xml 9.2 KB

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