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

Add subscription management to ComponentBase

Browse Source
This commit is contained in:
Bálint Bokros 2017-11-27 18:34:26 +01:00
parent 1d5abaaea0
commit 726fc165b0
1 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/components/component_base.py
View File

@ -14,7 +14,8 @@ class ComponentBase:
self.event_handler = event_handler self.event_handler = event_handler
self.zmq_context = zmq_context or zmq.Context.instance() self.zmq_context = zmq_context or zmq.Context.instance()
self.zmq_sub_socket = self.zmq_context.socket(zmq.SUB) self.zmq_sub_socket = self.zmq_context.socket(zmq.SUB)
self.zmq_sub_socket.setsockopt_string(zmq.SUBSCRIBE, anchor) self.subscriptions = {self.anchor}
self.zmq_sub_socket.setsockopt_string(zmq.SUBSCRIBE, self.anchor)
self.zmq_sub_socket.connect('tcp://localhost:{}'.format(PUBLISHER_PORT)) self.zmq_sub_socket.connect('tcp://localhost:{}'.format(PUBLISHER_PORT))
self.zmq_sub_stream = ZMQStream(self.zmq_sub_socket) self.zmq_sub_stream = ZMQStream(self.zmq_sub_socket)
self.zmq_push_socket = self.zmq_context.socket(zmq.PUSH) self.zmq_push_socket = self.zmq_context.socket(zmq.PUSH)
@ -29,3 +30,19 @@ class ComponentBase:
encoded_message = json.dumps(message).encode('utf-8') encoded_message = json.dumps(message).encode('utf-8')
self.zmq_push_socket.send_multipart([encoded_anchor, encoded_message]) self.zmq_push_socket.send_multipart([encoded_anchor, encoded_message])
def subscribe(self, anchor):
if anchor not in self.subscriptions:
self.subscriptions.add(anchor)
self.zmq_sub_socket.setsockopt_string(zmq.SUBSCRIBE, anchor)
def unsubscribe(self, anchor):
try:
self.subscriptions.remove(anchor)
self.zmq_sub_socket.setsockopt_string(zmq.UNSUBSCRIBE, anchor)
except KeyError:
pass
def unsubscribe_all(self):
for sub in self.subscriptions:
self.zmq_sub_socket.setsockopt_string(zmq.UNSUBSCRIBE, sub)
self.subscriptions.clear()