http_client.py 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371
  1. """
  2. Generated by Statechart compiler by Glenn De Jonghe, Joeri Exelmans, Simon Van Mierlo, and Yentl Van Tendeloo (for the inspiration)
  3. Date: Fri Aug 25 15:48:04 2017
  4. Model author: Yentl Van Tendeloo
  5. Model name: HTTP client
  6. Model description:
  7. HTTP client.
  8. """
  9. from sccd.runtime.statecharts_core import *
  10. import uuid
  11. # package "HTTP client"
  12. class HTTPClient(RuntimeClassBase):
  13. def __init__(self, controller):
  14. RuntimeClassBase.__init__(self, controller)
  15. self.semantics.big_step_maximality = StatechartSemantics.TakeMany
  16. self.semantics.internal_event_lifeline = StatechartSemantics.Queue
  17. self.semantics.input_event_lifeline = StatechartSemantics.FirstComboStep
  18. self.semantics.priority = StatechartSemantics.SourceParent
  19. self.semantics.concurrency = StatechartSemantics.Single
  20. # build Statechart structure
  21. self.build_statechart_structure()
  22. # call user defined constructor
  23. HTTPClient.user_defined_constructor(self)
  24. def user_defined_constructor(self):
  25. self.socket = None
  26. self.received_data = ""
  27. self.send_data = ""
  28. self.queue = []
  29. self.IDs = []
  30. def user_defined_destructor(self):
  31. pass
  32. # builds Statechart structure
  33. def build_statechart_structure(self):
  34. # state <root>
  35. self.states[""] = State(0, "", self)
  36. # state /init
  37. self.states["/init"] = State(1, "/init", self)
  38. self.states["/init"].setEnter(self._init_enter)
  39. # state /waiting
  40. self.states["/waiting"] = State(2, "/waiting", self)
  41. self.states["/waiting"].setEnter(self._waiting_enter)
  42. # state /connecting
  43. self.states["/connecting"] = State(3, "/connecting", self)
  44. self.states["/connecting"].setEnter(self._connecting_enter)
  45. self.states["/connecting"].setExit(self._connecting_exit)
  46. # state /connecting/connecting
  47. self.states["/connecting/connecting"] = State(4, "/connecting/connecting", self)
  48. self.states["/connecting/connecting"].setEnter(self._connecting_connecting_enter)
  49. # state /connecting/cooldown
  50. self.states["/connecting/cooldown"] = State(5, "/connecting/cooldown", self)
  51. self.states["/connecting/cooldown"].setEnter(self._connecting_cooldown_enter)
  52. self.states["/connecting/cooldown"].setExit(self._connecting_cooldown_exit)
  53. # state /connected
  54. self.states["/connected"] = ParallelState(6, "/connected", self)
  55. # state /connected/listening
  56. self.states["/connected/listening"] = State(7, "/connected/listening", self)
  57. # state /connected/listening/listen
  58. self.states["/connected/listening/listen"] = State(8, "/connected/listening/listen", self)
  59. self.states["/connected/listening/listen"].setEnter(self._connected_listening_listen_enter)
  60. # state /connected/listening/close
  61. self.states["/connected/listening/close"] = State(9, "/connected/listening/close", self)
  62. # state /connected/sending
  63. self.states["/connected/sending"] = State(10, "/connected/sending", self)
  64. # state /connected/sending/waiting_for_data
  65. self.states["/connected/sending/waiting_for_data"] = State(11, "/connected/sending/waiting_for_data", self)
  66. # state /connected/sending/transferring
  67. self.states["/connected/sending/transferring"] = State(12, "/connected/sending/transferring", self)
  68. # state /connected/queueing
  69. self.states["/connected/queueing"] = State(13, "/connected/queueing", self)
  70. # state /connected/queueing/queueing
  71. self.states["/connected/queueing/queueing"] = State(14, "/connected/queueing/queueing", self)
  72. self.states["/connected/queueing/queueing"].setEnter(self._connected_queueing_queueing_enter)
  73. # state /connected/parsing
  74. self.states["/connected/parsing"] = State(15, "/connected/parsing", self)
  75. # state /connected/parsing/wait_for_header
  76. self.states["/connected/parsing/wait_for_header"] = State(16, "/connected/parsing/wait_for_header", self)
  77. # state /connected/parsing/wait_for_payload
  78. self.states["/connected/parsing/wait_for_payload"] = State(17, "/connected/parsing/wait_for_payload", self)
  79. # add children
  80. self.states[""].addChild(self.states["/init"])
  81. self.states[""].addChild(self.states["/waiting"])
  82. self.states[""].addChild(self.states["/connecting"])
  83. self.states[""].addChild(self.states["/connected"])
  84. self.states["/connecting"].addChild(self.states["/connecting/connecting"])
  85. self.states["/connecting"].addChild(self.states["/connecting/cooldown"])
  86. self.states["/connected"].addChild(self.states["/connected/listening"])
  87. self.states["/connected"].addChild(self.states["/connected/sending"])
  88. self.states["/connected"].addChild(self.states["/connected/queueing"])
  89. self.states["/connected"].addChild(self.states["/connected/parsing"])
  90. self.states["/connected/listening"].addChild(self.states["/connected/listening/listen"])
  91. self.states["/connected/listening"].addChild(self.states["/connected/listening/close"])
  92. self.states["/connected/sending"].addChild(self.states["/connected/sending/waiting_for_data"])
  93. self.states["/connected/sending"].addChild(self.states["/connected/sending/transferring"])
  94. self.states["/connected/queueing"].addChild(self.states["/connected/queueing/queueing"])
  95. self.states["/connected/parsing"].addChild(self.states["/connected/parsing/wait_for_header"])
  96. self.states["/connected/parsing"].addChild(self.states["/connected/parsing/wait_for_payload"])
  97. self.states[""].fixTree()
  98. self.states[""].default_state = self.states["/init"]
  99. self.states["/connecting"].default_state = self.states["/connecting/connecting"]
  100. self.states["/connected/listening"].default_state = self.states["/connected/listening/listen"]
  101. self.states["/connected/sending"].default_state = self.states["/connected/sending/waiting_for_data"]
  102. self.states["/connected/queueing"].default_state = self.states["/connected/queueing/queueing"]
  103. self.states["/connected/parsing"].default_state = self.states["/connected/parsing/wait_for_header"]
  104. # transition /init
  105. _init_0 = Transition(self, self.states["/init"], [self.states["/waiting"]])
  106. _init_0.setAction(self._init_0_exec)
  107. _init_0.setTrigger(Event("created_socket", "socket_in"))
  108. _init_0.setGuard(self._init_0_guard)
  109. self.states["/init"].addTransition(_init_0)
  110. # transition /waiting
  111. _waiting_0 = Transition(self, self.states["/waiting"], [self.states["/connecting"]])
  112. _waiting_0.setAction(self._waiting_0_exec)
  113. _waiting_0.setTrigger(Event("connect", "request_in"))
  114. self.states["/waiting"].addTransition(_waiting_0)
  115. # transition /connecting/connecting
  116. _connecting_connecting_0 = Transition(self, self.states["/connecting/connecting"], [self.states["/connecting/cooldown"]])
  117. _connecting_connecting_0.setTrigger(Event("error_socket", "socket_in"))
  118. self.states["/connecting/connecting"].addTransition(_connecting_connecting_0)
  119. _connecting_connecting_1 = Transition(self, self.states["/connecting/connecting"], [self.states["/connected"]])
  120. _connecting_connecting_1.setAction(self._connecting_connecting_1_exec)
  121. _connecting_connecting_1.setTrigger(Event("connected_socket", "socket_in"))
  122. _connecting_connecting_1.setGuard(self._connecting_connecting_1_guard)
  123. self.states["/connecting/connecting"].addTransition(_connecting_connecting_1)
  124. # transition /connecting/cooldown
  125. _connecting_cooldown_0 = Transition(self, self.states["/connecting/cooldown"], [self.states["/connecting/connecting"]])
  126. _connecting_cooldown_0.setTrigger(Event("_1after"))
  127. self.states["/connecting/cooldown"].addTransition(_connecting_cooldown_0)
  128. # transition /connected/listening/listen
  129. _connected_listening_listen_0 = Transition(self, self.states["/connected/listening/listen"], [self.states["/connected/listening/listen"]])
  130. _connected_listening_listen_0.setAction(self._connected_listening_listen_0_exec)
  131. _connected_listening_listen_0.setTrigger(Event("received_socket", "socket_in"))
  132. _connected_listening_listen_0.setGuard(self._connected_listening_listen_0_guard)
  133. self.states["/connected/listening/listen"].addTransition(_connected_listening_listen_0)
  134. _connected_listening_listen_1 = Transition(self, self.states["/connected/listening/listen"], [self.states["/connected/listening/close"]])
  135. _connected_listening_listen_1.setTrigger(Event("received_socket", "socket_in"))
  136. _connected_listening_listen_1.setGuard(self._connected_listening_listen_1_guard)
  137. self.states["/connected/listening/listen"].addTransition(_connected_listening_listen_1)
  138. # transition /connected/sending/waiting_for_data
  139. _connected_sending_waiting_for_data_0 = Transition(self, self.states["/connected/sending/waiting_for_data"], [self.states["/connected/sending/transferring"]])
  140. _connected_sending_waiting_for_data_0.setAction(self._connected_sending_waiting_for_data_0_exec)
  141. _connected_sending_waiting_for_data_0.setTrigger(None)
  142. _connected_sending_waiting_for_data_0.setGuard(self._connected_sending_waiting_for_data_0_guard)
  143. self.states["/connected/sending/waiting_for_data"].addTransition(_connected_sending_waiting_for_data_0)
  144. # transition /connected/sending/transferring
  145. _connected_sending_transferring_0 = Transition(self, self.states["/connected/sending/transferring"], [self.states["/connected/sending/waiting_for_data"]])
  146. _connected_sending_transferring_0.setAction(self._connected_sending_transferring_0_exec)
  147. _connected_sending_transferring_0.setTrigger(Event("sent_socket", "socket_in"))
  148. _connected_sending_transferring_0.setGuard(self._connected_sending_transferring_0_guard)
  149. self.states["/connected/sending/transferring"].addTransition(_connected_sending_transferring_0)
  150. # transition /connected/queueing/queueing
  151. _connected_queueing_queueing_0 = Transition(self, self.states["/connected/queueing/queueing"], [self.states["/connected/queueing/queueing"]])
  152. _connected_queueing_queueing_0.setAction(self._connected_queueing_queueing_0_exec)
  153. _connected_queueing_queueing_0.setTrigger(Event("HTTP_input", "request_in"))
  154. self.states["/connected/queueing/queueing"].addTransition(_connected_queueing_queueing_0)
  155. # transition /connected/parsing/wait_for_header
  156. _connected_parsing_wait_for_header_0 = Transition(self, self.states["/connected/parsing/wait_for_header"], [self.states["/connected/parsing/wait_for_payload"]])
  157. _connected_parsing_wait_for_header_0.setAction(self._connected_parsing_wait_for_header_0_exec)
  158. _connected_parsing_wait_for_header_0.setTrigger(None)
  159. _connected_parsing_wait_for_header_0.setGuard(self._connected_parsing_wait_for_header_0_guard)
  160. self.states["/connected/parsing/wait_for_header"].addTransition(_connected_parsing_wait_for_header_0)
  161. # transition /connected/parsing/wait_for_payload
  162. _connected_parsing_wait_for_payload_0 = Transition(self, self.states["/connected/parsing/wait_for_payload"], [self.states["/connected/parsing/wait_for_header"]])
  163. _connected_parsing_wait_for_payload_0.setAction(self._connected_parsing_wait_for_payload_0_exec)
  164. _connected_parsing_wait_for_payload_0.setTrigger(None)
  165. _connected_parsing_wait_for_payload_0.setGuard(self._connected_parsing_wait_for_payload_0_guard)
  166. self.states["/connected/parsing/wait_for_payload"].addTransition(_connected_parsing_wait_for_payload_0)
  167. _connected_parsing_wait_for_payload_1 = Transition(self, self.states["/connected/parsing/wait_for_payload"], [self.states["/connected/parsing/wait_for_header"]])
  168. _connected_parsing_wait_for_payload_1.setAction(self._connected_parsing_wait_for_payload_1_exec)
  169. _connected_parsing_wait_for_payload_1.setTrigger(None)
  170. _connected_parsing_wait_for_payload_1.setGuard(self._connected_parsing_wait_for_payload_1_guard)
  171. self.states["/connected/parsing/wait_for_payload"].addTransition(_connected_parsing_wait_for_payload_1)
  172. # transition /connecting
  173. _connecting_0 = Transition(self, self.states["/connecting"], [self.states["/waiting"]])
  174. _connecting_0.setAction(self._connecting_0_exec)
  175. _connecting_0.setTrigger(Event("_0after"))
  176. self.states["/connecting"].addTransition(_connecting_0)
  177. def _connecting_enter(self):
  178. self.addTimer(0, self.timeout)
  179. def _connecting_exit(self):
  180. self.removeTimer(0)
  181. def _init_enter(self):
  182. self.ID = str(uuid.uuid4())
  183. self.big_step.outputEvent(Event("create_socket", "socket_out", [self.ID]))
  184. def _waiting_enter(self):
  185. self.big_step.outputEvent(Event("http_client_initialized", "request_out", []))
  186. def _connecting_connecting_enter(self):
  187. self.big_step.outputEvent(Event("connect_socket", "socket_out", [self.socket, self.address]))
  188. def _connecting_cooldown_enter(self):
  189. self.addTimer(1, 0.1)
  190. def _connecting_cooldown_exit(self):
  191. self.removeTimer(1)
  192. def _connected_listening_listen_enter(self):
  193. self.big_step.outputEvent(Event("recv_socket", "socket_out", [self.socket]))
  194. def _connected_queueing_queueing_enter(self):
  195. pass
  196. def _connecting_0_exec(self, parameters):
  197. self.big_step.outputEvent(Event("http_client_timeout", "request_out", []))
  198. def _init_0_exec(self, parameters):
  199. socket = parameters[0]
  200. ID = parameters[1]
  201. self.socket = socket
  202. def _init_0_guard(self, parameters):
  203. socket = parameters[0]
  204. ID = parameters[1]
  205. return self.ID == ID
  206. def _waiting_0_exec(self, parameters):
  207. address = parameters[0]
  208. timeout = parameters[1]
  209. self.address = address
  210. self.timeout = timeout
  211. def _connecting_connecting_1_exec(self, parameters):
  212. socket = parameters[0]
  213. self.big_step.outputEvent(Event("http_client_ready", "request_out", []))
  214. def _connecting_connecting_1_guard(self, parameters):
  215. socket = parameters[0]
  216. return self.socket == socket
  217. def _connected_listening_listen_0_exec(self, parameters):
  218. socket = parameters[0]
  219. data = parameters[1]
  220. self.received_data += data
  221. def _connected_listening_listen_0_guard(self, parameters):
  222. socket = parameters[0]
  223. data = parameters[1]
  224. return (self.socket == socket) and (len(data) > 0)
  225. def _connected_listening_listen_1_guard(self, parameters):
  226. socket = parameters[0]
  227. data = parameters[1]
  228. return (self.socket == socket) and (len(data) == 0)
  229. def _connected_sending_waiting_for_data_0_exec(self, parameters):
  230. self.big_step.outputEvent(Event("send_socket", "socket_out", [self.socket, self.send_data]))
  231. def _connected_sending_waiting_for_data_0_guard(self, parameters):
  232. return len(self.send_data) > 0
  233. def _connected_sending_transferring_0_exec(self, parameters):
  234. socket = parameters[0]
  235. sent_bytes = parameters[1]
  236. self.send_data = self.send_data[sent_bytes:]
  237. def _connected_sending_transferring_0_guard(self, parameters):
  238. socket = parameters[0]
  239. sent_bytes = parameters[1]
  240. return self.socket == socket
  241. def _connected_queueing_queueing_0_exec(self, parameters):
  242. data = parameters[0]
  243. ID = parameters[1]
  244. self.send_data += "POST / HTTP/1.0\r\n"
  245. self.send_data += "Content-Length: %i\r\n" % len(str(data))
  246. self.send_data += "\r\n"
  247. self.send_data += data
  248. self.IDs.append(ID)
  249. def _connected_parsing_wait_for_header_0_exec(self, parameters):
  250. header, self.received_data = self.received_data.split("\r\n\r\n", 1)
  251. header = header.lower()
  252. if "content-length" in header:
  253. _, after = header.split("content-length:", 1)
  254. after, _ = after.split("\r\n", 1)
  255. after = after.strip()
  256. self.length = int(after)
  257. else:
  258. self.length = float('inf')
  259. def _connected_parsing_wait_for_header_0_guard(self, parameters):
  260. return '\r\n\r\n' in self.received_data
  261. def _connected_parsing_wait_for_payload_0_exec(self, parameters):
  262. data = self.received_data[:self.length]
  263. self.received_data = self.received_data[self.length:]
  264. self.big_step.outputEvent(Event("HTTP_output", "request_out", [data, self.IDs.pop(0)]))
  265. def _connected_parsing_wait_for_payload_0_guard(self, parameters):
  266. return len(self.received_data) >= self.length and self.IDs[0] is not None
  267. def _connected_parsing_wait_for_payload_1_exec(self, parameters):
  268. # Drop data
  269. self.received_data = self.received_data[self.length:]
  270. self.IDs.pop(0)
  271. def _connected_parsing_wait_for_payload_1_guard(self, parameters):
  272. return len(self.received_data) >= self.length and self.IDs[0] is None
  273. def initializeStatechart(self):
  274. # enter default state
  275. self.default_targets = self.states["/init"].getEffectiveTargetStates()
  276. RuntimeClassBase.initializeStatechart(self)
  277. class ObjectManager(ObjectManagerBase):
  278. def __init__(self, controller):
  279. ObjectManagerBase.__init__(self, controller)
  280. def instantiate(self, class_name, construct_params):
  281. if class_name == "HTTPClient":
  282. instance = HTTPClient(self.controller)
  283. instance.associations = {}
  284. else:
  285. raise Exception("Cannot instantiate class " + class_name)
  286. return instance
  287. class Controller(ThreadsControllerBase):
  288. def __init__(self, keep_running = None, behind_schedule_callback = None):
  289. if keep_running == None: keep_running = True
  290. if behind_schedule_callback == None: behind_schedule_callback = None
  291. ThreadsControllerBase.__init__(self, ObjectManager(self), keep_running, behind_schedule_callback)
  292. self.addInputPort("socket_in")
  293. self.addInputPort("request_in")
  294. self.addOutputPort("socket_out")
  295. self.addOutputPort("request_out")
  296. self.object_manager.createInstance("HTTPClient", [])