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

Close ZMQ sockets gracefully

Browse Source
This commit is contained in:
R. Richard 2019-05-20 11:06:57 +02:00
parent 2a3d29f080
commit 7d50ee607a
2 changed files with 15 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lib/tfw/event_handler_base/event_handler_base.py
View File

@ -106,6 +106,10 @@ class EventHandlerBase(ABC):
self.server_connector.unsubscribe(key)
self.keys.remove(key)
def stop(self):
self.server_connector.close()
self.cleanup()
def cleanup(self):
"""
Perform cleanup actions such as releasing database

12
lib/tfw/networking/event_handlers/server_connector.py
View File

@ -19,6 +19,7 @@ class ServerDownlinkConnector(ZMQConnectorBase):
super(ServerDownlinkConnector, self).__init__(zmq_context)
self._zmq_sub_socket = self._zmq_context.socket(zmq.SUB)
self._zmq_sub_socket.connect(f'tcp://localhost:{TFWENV.PUBLISHER_PORT}')
self._zmq_sub_socket.setsockopt(zmq.RCVHWM,0)
self._zmq_sub_stream = ZMQStream(self._zmq_sub_socket)
self.subscribe = partial(self._zmq_sub_socket.setsockopt_string, zmq.SUBSCRIBE)
@ -28,6 +29,9 @@ class ServerDownlinkConnector(ZMQConnectorBase):
callback = with_deserialize_tfw_msg(callback)
self._zmq_sub_stream.on_recv(callback)
def close(self):
self._zmq_sub_stream.close()
class ServerUplinkConnector(ZMQConnectorBase):
"""
@ -37,6 +41,7 @@ class ServerUplinkConnector(ZMQConnectorBase):
super(ServerUplinkConnector, self).__init__(zmq_context)
self._zmq_push_socket = self._zmq_context.socket(zmq.PUSH)
self._zmq_push_socket.connect(f'tcp://localhost:{TFWENV.RECEIVER_PORT}')
self._zmq_push_socket.setsockopt(zmq.SNDHWM,0)
def send_to_eventhandler(self, message):
"""
@ -74,6 +79,11 @@ class ServerUplinkConnector(ZMQConnectorBase):
'data': message
})
def close(self):
self._zmq_push_socket.close()
class ServerConnector(ServerUplinkConnector, ServerDownlinkConnector):
pass
def close(self):
ServerUplinkConnector.close(self)
ServerDownlinkConnector.close(self)