diff --git a/lib/tfw/networking/server/event_handler_connector.py b/lib/tfw/networking/server/event_handler_connector.py index 41b6b54..a735673 100644 --- a/lib/tfw/networking/server/event_handler_connector.py +++ b/lib/tfw/networking/server/event_handler_connector.py @@ -34,5 +34,5 @@ class EventHandlerConnector(EventHandlerDownlinkConnector, EventHandlerUplinkCon def register_callback(self, callback): self._zmq_pull_stream.on_recv(callback) - def send_message(self, message: dict, key: str = None): + def send_message(self, message: dict): self._zmq_pub_socket.send_multipart(serialize_tfw_msg(message)) diff --git a/lib/tfw/networking/server/zmq_websocket_handler.py b/lib/tfw/networking/server/zmq_websocket_handler.py index f0b865a..efd7b00 100644 --- a/lib/tfw/networking/server/zmq_websocket_handler.py +++ b/lib/tfw/networking/server/zmq_websocket_handler.py @@ -46,7 +46,7 @@ class ZMQWebSocketHandler(WebSocketHandler, ABC): def mirror(self, message): message = message['data'] - self._event_handler_connector.send_message(message, message['key']) + self._event_handler_connector.send_message(message) def on_message(self, message): LOG.debug('Received on WebSocket: %s', message) @@ -57,7 +57,7 @@ class ZMQWebSocketHandler(WebSocketHandler, ABC): raise NotImplementedError def send_message(self, message: dict, key: str = None): - self._event_handler_connector.send_message(message, key) + self._event_handler_connector.send_message(message) @abstractmethod def handle_trigger(self, message):