feat: factory get implementation by type name

(not sure about this...)

replication.py

@@ -35,6 +35,11 @@ class ReplicatedDataFactory(object):
         print("type not supported for replication")
         raise NotImplementedError
 
+    def match_type_by_name(self,type_name):
+        for stypes, implementation in self.supported_types: 
+            if type_name == implementation.__class__.__name__:
+                return implementation
+
     def construct_from_dcc(self,data):
         implementation = self.match_type_by_instance(data)
         return implementation
@@ -61,9 +66,7 @@ class ReplicatedDatablock(object):
         assert(owner)
         self.owner = owner
         self.pointer = data
-
+        self.str_type = self.data.__class__.__name__
-        if data:
-            self.str_type = self.data.__class__.__name__
         
 
     def push(self, socket):
@@ -76,8 +79,9 @@ class ReplicatedDatablock(object):
         assert(isinstance(data, bytes))
         owner = self.owner.encode()
         key = self.uuid.encode()
+        type = self.str_type.encode()
 
-        socket.send_multipart([key,owner,data])
+        socket.send_multipart([key,owner,str_type,data])
    
     @classmethod
     def pull(cls, socket, factory):
@@ -86,9 +90,9 @@ class ReplicatedDatablock(object):
             - read data from the socket
             - reconstruct an instance
         """
-        uuid, owner, data = socket.recv_multipart(zmq.NOBLOCK)
+        uuid, owner,str_type, data = socket.recv_multipart(zmq.NOBLOCK)
 
-        instance = factory.construct_from_net(data)(owner=owner.decode(), uuid=uuid.decode())
+        instance = factory.construct_from_net(str_type.decode())(owner=owner.decode(), uuid=uuid.decode())
 
         instance.data = instance.deserialize(data)
         return instance

replication_client.py

@@ -31,7 +31,7 @@ class Client(object):
         """
         assert(object)
         
-        new_item = self._factory.construct_from_dcc(object)(owner="client")
+        new_item = self._factory.construct_from_dcc(object)(owner="client", data=object)
 
         if new_item:
             log.info("Registering {} on {}".format(object,new_item.uuid))
@@ -107,9 +107,9 @@ class ClientNetService(threading.Thread):
 
 
 class Server():
-    def __init__(self,config=None):
+    def __init__(self,config=None, factory=None):
         self.rep_store = {}
-        self.net = ServerNetService(self.rep_store)
+        self.net = ServerNetService(store_reference=self.rep_store, factory=factory)
         # self.serve()
 
     def serve(self):
@@ -205,7 +205,7 @@ class ServerNetService(threading.Thread):
             # Regular update routing (Clients / Client)
             if self.pull in socks:
                 log.info("Receiving changes from client")
-                msg = ReplicatedDatablock.pull(self.pull)
+                msg = ReplicatedDatablock.pull(self.pull, self.factory)
 
                 msg.store(self.store)
                 # msg = message.Message.recv(self.collector_sock)

test_replication.py

@@ -72,7 +72,7 @@ class TestDataReplication(unittest.TestCase):
         factory = ReplicatedDataFactory()
         factory.register_type(SampleData, RepSampleData)
 
-        server_api = Server()
+        server_api = Server(factory=factory)
         server_api.serve()
         client_api = Client(factory=factory)
         client_api.connect()