mirror of
				https://github.com/avatao-content/baseimage-tutorial-framework
				synced 2025-11-04 12:12:55 +00:00 
			
		
		
		
	Hide ZMQ serialization magic from ServerConnector clients
This commit is contained in:
		@@ -6,9 +6,12 @@ from functools import partial
 | 
			
		||||
import zmq
 | 
			
		||||
from zmq.eventloop.zmqstream import ZMQStream
 | 
			
		||||
 | 
			
		||||
from tfw.networking import serialize_tfw_msg
 | 
			
		||||
from tfw.networking import serialize_tfw_msg, with_deserialize_tfw_msg
 | 
			
		||||
from tfw.networking import ZMQConnectorBase
 | 
			
		||||
from tfw.config import TFWENV
 | 
			
		||||
from tfw.config.logs import logging
 | 
			
		||||
 | 
			
		||||
LOG = logging.getLogger(__name__)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ServerDownlinkConnector(ZMQConnectorBase):
 | 
			
		||||
@@ -20,7 +23,10 @@ class ServerDownlinkConnector(ZMQConnectorBase):
 | 
			
		||||
 | 
			
		||||
        self.subscribe = partial(self._zmq_sub_socket.setsockopt_string, zmq.SUBSCRIBE)
 | 
			
		||||
        self.unsubscribe = partial(self._zmq_sub_socket.setsockopt_string, zmq.UNSUBSCRIBE)
 | 
			
		||||
        self.register_callback = self._zmq_sub_stream.on_recv
 | 
			
		||||
 | 
			
		||||
    def register_callback(self, callback):
 | 
			
		||||
        callback = with_deserialize_tfw_msg(callback)
 | 
			
		||||
        self._zmq_sub_stream.on_recv(callback)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ServerUplinkConnector(ZMQConnectorBase):
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user