feat: server refactoring

This commit is contained in:
Swann
2019-07-05 21:17:37 +02:00
parent a87d3c33ba
commit c5dc46f63d
2 changed files with 12 additions and 13 deletions

View File

@ -2,6 +2,7 @@ import threading
import logging import logging
import zmq import zmq
import time import time
import replication
logging.basicConfig(level=logging.DEBUG) logging.basicConfig(level=logging.DEBUG)
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -46,8 +47,6 @@ class ClientNetService(threading.Thread):
def run(self): def run(self):
log.debug("Running Net service")
poller = zmq.Poller() poller = zmq.Poller()
poller.register(self.snapshot, zmq.POLLIN) poller.register(self.snapshot, zmq.POLLIN)
poller.register(self.subscriber, zmq.POLLIN) poller.register(self.subscriber, zmq.POLLIN)
@ -61,6 +60,12 @@ class ClientNetService(threading.Thread):
if not items: if not items:
log.error("No request ") log.error("No request ")
time.sleep(1)
def send(data):
assert(issubclass(data, ReplicatedDatablock))
data.push(self.publish)
def stop(self): def stop(self):
self.exit_event.set() self.exit_event.set()
@ -119,8 +124,6 @@ class ServerNetService(threading.Thread):
self.state = 0 self.state = 0
def run(self): def run(self):
log.debug("Running Server Net service")
poller = zmq.Poller() poller = zmq.Poller()
poller.register(self.snapshot, zmq.POLLIN) poller.register(self.snapshot, zmq.POLLIN)
poller.register(self.pull, zmq.POLLIN) poller.register(self.pull, zmq.POLLIN)
@ -131,4 +134,6 @@ class ServerNetService(threading.Thread):
items = dict(poller.poll(10)) items = dict(poller.poll(10))
if not items: if not items:
log.info("No request ") pass
time.sleep(1)

View File

@ -35,6 +35,7 @@ class TestData(unittest.TestCase):
self.server_api = Server() self.server_api = Server()
def test_server_launching(self): def test_server_launching(self):
log.info("test_server_launching")
self.server_api.serve() self.server_api.serve()
time.sleep(1) time.sleep(1)
self.assertEqual(self.server_api.state(),1) self.assertEqual(self.server_api.state(),1)
@ -48,6 +49,7 @@ class TestData(unittest.TestCase):
self.assertEqual(isinstance(rep_sample,RepSampleData), True) self.assertEqual(isinstance(rep_sample,RepSampleData), True)
def test_client_connect(self): def test_client_connect(self):
log.info("test_client_connect")
self.client_api.connect() self.client_api.connect()
time.sleep(1) time.sleep(1)
self.assertEqual(self.client_api.state(),1) self.assertEqual(self.client_api.state(),1)
@ -57,14 +59,6 @@ class TestData(unittest.TestCase):
time.sleep(1) time.sleep(1)
self.assertEqual(self.client_api.state(),0) self.assertEqual(self.client_api.state(),0)
def test_client_add_rep(self):
pass
def test_add_replicated_value(self):
pass