blender code refactoring

client.py

@@ -40,7 +40,7 @@ def main():
             break
 
         if updates in socks:
-            message = updates.recv_multipart()
+            message = updates.recv_multipart(zmq.NOBLOCK)
             print(message)
         
         # Send update

net_components.py

@@ -10,6 +10,7 @@ import random
 logger = logging.getLogger(__name__)
 logging.basicConfig(level=logging.DEBUG)
 
+
 class Session():
     def __init__(self, host='127.0.0.1', port=5555, is_hosting=False):
         self.host = host
@@ -89,6 +90,7 @@ class Session():
         del self.listen
         self.is_running = False
 
+
 class Client_poller():
     def __init__(self, id):
 
@@ -117,86 +119,119 @@ class Client_poller():
                 msg = socket.recv(zmq.NOBLOCK)
                 logger.info("{} received:{}".format(self.id, msg))
 
-
     def stop(self):
         logger.info("Stopping client {}".format(self.id))
         self.listen.cancel()
 
+
 class Client():
-    def __init__(self, context=None):
+    def __init__(self, context=zmq.Context(), id="default"):
 
-        if context is None:
-            logger.info("client init default context")
-            self.context = zmq.Context()
-        else:
         self.context = context
+        self.pull_sock = None
+        self.push_sock = None
+        self.poller = None
 
-        self.task = asyncio.ensure_future(self.run())
-        logger.info("client initiated")
+        self.id = id
+        self.bind_ports()
+        # Main client loop registration
+        self.task = asyncio.ensure_future(self.main())
 
-    async def run(self):
+        logger.info("{} client initialized".format(id))
+
+    def bind_ports(self):
+        # pull socket: get update FROM server
+        self.pull_sock = self.context.socket(zmq.SUB)
+        self.pull_sock.linger = 0
+        self.pull_sock.connect("tcp://localhost:5555")
+        self.pull_sock.setsockopt_string(zmq.SUBSCRIBE, '')
+
+        # push socket: push update TO server
+        self.push_sock = self.context.socket(zmq.DEALER)
+        self.push_sock.setsockopt(zmq.IDENTITY, self.id.encode())
+        self.push_sock.linger = 0
+        self.push_sock.connect("tcp://localhost:5556")
+
+        # Sockets aggregator, not really used for now
+        self.poller = zmq.Poller()
+        self.poller.register(self.pull_sock, zmq.POLLIN)
+
+    async def main(self):
+        logger.info("{} client launched".format(id))
        # Prepare our context and publisher socket
-        logger.info("configuring:")
-        updates = self.context.socket(zmq.SUB)
-        logger.info("..socket")
-        updates.linger = 0
-        logger.info("..linger")
-        updates.setsockopt(zmq.SUBSCRIBE,b"10001")
-        logger.info("client launched")
-        updates.connect("tcp://localhost:5556")
-
-        kvmap = {}
-        sequence = 0
-        
         while True:
+            # TODO: find a better way
             await asyncio.sleep(0.016)
             try:
-                kvmsg = kvsimple.KVMsg.recv(updates)
-            except:
-                break # Interrupted
-            kvmsg.store(kvmap)
-            sequence += 1
+                socks = dict(self.poller.poll(1))
+            except KeyboardInterrupt:
+                break
 
+            if self.pull_sock in socks:
+                message = self.pull_sock.recv_multipart(zmq.NOBLOCK)
+                print(message)
+
+    def send_msg(self, msg):
+        self.push_sock.send_multipart(msg.encode())
 
     def stop(self):
         logger.info("Stopping client")
         self.task.cancel()
+        self.push_sock.close()
+        self.pull_sock.close()
+        self.context.term()
 
 
-
-    kvmap = {}
-    sequence = 0
-
 class Server():
-    def __init__(self):
-        self.context = zmq.Context()
-        self.task = asyncio.ensure_future(self.run())
-        logger.info("server initiated ")
+    def __init__(self, context=zmq.Context(), id="admin"):
+        self.context = context
+        self.pub_sock = None
+        self.pull_sock = None
+        self.poller = None
 
-    async def run(self):
-        publisher = self.context.socket(zmq.PUB)
+        self.id = id
+        self.bind_ports()
+        # Main client loop registration
+        self.task = asyncio.ensure_future(self.main())
 
-        publisher.bind("tcp://*:5556")
+        logger.info("{} client initialized".format(id))
+
+    def bind_ports(self):
+        # Update all clients
+        self.pub_sock = self.context.socket(zmq.PUB)
+        self.pub_sock.bind("tcp://*:5555")
         time.sleep(0.2)
-        logger.info("server launched")
 
-        sequence = 0
-        random.seed(time.time())
-        kvmap = {}
+        # Update receiver
+        self.pull_sock = self.context.socket(zmq.ROUTER)
+        self.pull_sock.bind("tcp://*:5556")
 
+        # poller for socket aggregation
+        self.poller = zmq.Poller()
+        self.poller.register(self.pull_sock, zmq.POLLIN)
+
+    async def main(self):
+        logger.info("{} server launched".format(id))
+        # Prepare our context and publisher socket
         while True:
-            # Non blocking
+            # TODO: find a better way
             await asyncio.sleep(0.016)
 
-            # Distribute as key-value message
-            sequence += 1
-            kvmsg = kvsimple.KVMsg(sequence)
-            kvmsg.key = "%d" % random.randint(1,10000)
-            kvmsg.body = "%d" % random.randint(1,1000000)
-            kvmsg.send(publisher)
-            kvmsg.store(kvmap)
+            try:
+                socks = dict(self.poller.poll(1))
+            except KeyboardInterrupt:
+                break
 
+            if self.pull_sock in socks:
+                msg = self.pull_sock.recv_multipart(zmq.NOBLOCK)
+                print("{}:{}".format(msg[0].decode('ascii'), msg[1].decode()))
+
+                # Update all clients
+                self.pub_sock.send_multipart(msg)
 
     def stop(self):
         logger.info("Stopping server")
         self.task.cancel()
+        self.pub_sock.close()
+        self.pull_sock.close()
+        self.context.term()

server.py

@@ -31,9 +31,8 @@ def main():
             break
             
         if state_request in socks:
-            msg = state_request.recv_multipart()
-            print(msg[0].decode('ascii'))
-            print(msg[1].decode())
+            msg = state_request.recv_multipart(zmq.NOBLOCK)
+            print("{}:{}".format(msg[0].decode('ascii'),msg[1].decode()))
             publisher.send(b'Server update')
             
         # publisher.send_string('test')