mrtoth/baseimage-tutorial-framework
1
0
Fork 0
mirror of https://github.com/avatao-content/baseimage-tutorial-framework synced 2024-11-22 21:11:32 +00:00
Code Issues Releases Wiki Activity

Implement support for multiple websockets connecting to TFW

Browse Source
This commit is contained in:
Kristóf Tóth 2018-02-16 17:47:30 +01:00
parent 3e54733033
commit 24fd753679
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
lib/tfw/networking/server/zmq_websocket_handler.py
View File

@ -9,6 +9,13 @@ log = logging.getLogger(__name__)
class ZMQWebSocketHandler(WebSocketHandler): class ZMQWebSocketHandler(WebSocketHandler):
_event_handler_connector = EventHandlerConnector() _event_handler_connector = EventHandlerConnector()
instances = set()
def prepare(self):
ZMQWebSocketHandler.instances.add(self)
def on_close(self):
ZMQWebSocketHandler.instances.remove(self)
def __init__(self, application, request, **kwargs): def __init__(self, application, request, **kwargs):
super().__init__(application, request, **kwargs) super().__init__(application, request, **kwargs)
@ -17,10 +24,12 @@ class ZMQWebSocketHandler(WebSocketHandler):
log.debug('WebSocket connection initiated') log.debug('WebSocket connection initiated')
self._event_handler_connector.register_callback(self.zmq_callback) self._event_handler_connector.register_callback(self.zmq_callback)
def zmq_callback(self, msg_parts): @staticmethod
def zmq_callback(msg_parts):
anchor, data = deserialize_all(*msg_parts) anchor, data = deserialize_all(*msg_parts)
log.debug('Received on pull socket: {}'.format(data)) log.debug('Received on pull socket: {}'.format(data))
self.write_message(data) for instance in ZMQWebSocketHandler.instances:
instance.write_message(data)
def on_message(self, message): def on_message(self, message):
log.debug('Received on WebSocket: {}'.format(message)) log.debug('Received on WebSocket: {}'.format(message))
@ -40,6 +49,7 @@ class FSMManagingSocketHandler(ZMQWebSocketHandler):
self.fsm.subscribe_message_handler(self.handle_fsm_message) self.fsm.subscribe_message_handler(self.handle_fsm_message)
def on_close(self): def on_close(self):
super().on_close()
self.fsm.unsubscribe_message_handler(self.handle_fsm_message) self.fsm.unsubscribe_message_handler(self.handle_fsm_message)
def handle_fsm_message(self, message): def handle_fsm_message(self, message):