68 lines
1.6 KiB
Python
68 lines
1.6 KiB
Python
|
import threading
|
||
|
|
import logging
|
||
|
|
import zmq
|
||
|
|
import time
|
||
|
|
|
||
|
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
|
log = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
class Client(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.rep_store = {}
|
||
|
|
self.net = ClientNetService(self.rep_store)
|
||
|
|
|
||
|
|
def connect(self):
|
||
|
|
self.net.run()
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.net.stop()
|
||
|
|
|
||
|
|
class ClientNetService(threading.Thread):
|
||
|
|
def __init__(self,store_reference=None):
|
||
|
|
# Threading
|
||
|
|
threading.Thread.__init__(self)
|
||
|
|
self.name = "NetLink"
|
||
|
|
self.daemon = True
|
||
|
|
self.exit_event = threading.Event()
|
||
|
|
|
||
|
|
# Networking
|
||
|
|
self.context = zmq.Context.instance()
|
||
|
|
|
||
|
|
self.snapshot = self.context.socket(zmq.DEALER)
|
||
|
|
self.snapshot.connect("tcp://127.0.0.1:5560")
|
||
|
|
|
||
|
|
self.subscriber = self.context.socket(zmq.SUB)
|
||
|
|
self.subscriber.setsockopt_string(zmq.SUBSCRIBE, '')
|
||
|
|
self.subscriber.connect("tcp://127.0.0.1:5561")
|
||
|
|
self.subscriber.linger = 0
|
||
|
|
|
||
|
|
self.publish = self.context.socket(zmq.PULL)
|
||
|
|
self.publish.bind("tcp://*:5562")
|
||
|
|
|
||
|
|
# For teststing purpose
|
||
|
|
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
log.debug("Running Net service")
|
||
|
|
|
||
|
|
poller = zmq.Poller()
|
||
|
|
poller.register(self.snapshot, zmq.POLLIN)
|
||
|
|
poller.register(self.subscriber, zmq.POLLIN)
|
||
|
|
poller.register(self.publish, zmq.POLLOUT)
|
||
|
|
|
||
|
|
while not self.exit_event.is_set():
|
||
|
|
items = dict(poller.poll(10))
|
||
|
|
|
||
|
|
if not items:
|
||
|
|
log.error("No request ")
|
||
|
|
|
||
|
|
|
||
|
|
def stop(self):
|
||
|
|
self.exit_event.set()
|
||
|
|
|
||
|
|
#Wait the end of the run
|
||
|
|
while self.exit_event.is_set():
|
||
|
|
time.sleep(.1)
|
||
|
|
|
||
|
|
|