def setUp(self):
self.heartbeat_event = Event()
self.cfg = Mock()
self.cfg.data_port = DATA_PORT
self.q = Queue()
self.orbpktsrc = MagicMock()
self.orbpktsrc.subscription.return_value.__enter__.return_value = self.q
self.ds = DataServer(('localhost', DATA_PORT),
self.orbpktsrc.subscription, self.heartbeat_event, self.janitor)
class Test_DataServer(unittest.TestCase):
def janitor(self, *args, **kwargs):
pass
def setUp(self):
self.heartbeat_event = Event()
self.cfg = Mock()
self.cfg.data_port = DATA_PORT
self.q = Queue()
self.orbpktsrc = MagicMock()
self.orbpktsrc.subscription.return_value.__enter__.return_value = self.q
self.ds = DataServer(('localhost', DATA_PORT),
self.orbpktsrc.subscription, self.heartbeat_event, self.janitor)
def test_DataServer(self):
self.cfg.heartbeat_interval = 10
[self.q.put(('', float(n))) for n in xrange(15)]
received = []
def rx():
try:
self.ds.start()
sock = create_connection(('127.0.0.1', DATA_PORT), timeout=2)
while True:
received.append(sock.recv(0x10))
finally:
self.ds.stop()
rxg = spawn(rx)
sleep(1)
rxg.kill()
self.assertEquals(len(received), 15)
def test_DataServer_heartbeat(self):
self.cfg.heartbeat_interval = 1
received = []
def rx():
try:
self.ds.start()
sock = create_connection(('127.0.0.1', DATA_PORT), timeout=2)
while True:
received.append(sock.recv(0x10))
finally:
self.ds.stop()
rxg = spawn(rx)
sleep(0.1)
self.heartbeat_event.set()
self.heartbeat_event.clear()
sleep(0.1)
rxg.kill()
self.assertEquals(len(received), 1)
def state_configured(self):
# spawn orbreapthr
# link to it? what if it dies?
self.orbpktsrc = OrbPktSrc(
srcname = self.cfg.antelope_orb_name,
select = self.cfg.antelope_orb_select,
reject = self.cfg.antelope_orb_reject,
after = self.cfg.antelope_orb_after,
transformation = transform
)
self.orbpktsrc.link_exception(self.janitor)
self.orbpktsrc.start()
# spawn data server
self.dataserver = DataServer(
(LISTENING_ADDR, self.cfg.data_port),
self.orbpktsrc.subscription,
self.heartbeat_event, self.janitor)
self.dataserver.start()
# spawn state_connected
return self.state_connected
class PortAgent(Greenlet):
def __init__(self, cfg, cmdproc):
super(PortAgent, self).__init__()
self.cfg = cfg
self.cmdproc = cmdproc
self.heartbeat_event = Event()
self.cmdproc.setCmd('get_state', None, self.get_state)
self.cmdproc.setCmd('ping', None, self.ping)
self.cmdproc.setCmd('shutdown', None, self.shutdown)
self.states = {
self.state_startup: 'STARTUP',
self.state_unconfigured: 'UNCONFIGURED',
self.state_configured: 'CONFIGURED',
self.state_connected: 'CONNECTED',
}
@property
def state(self):
return self._state
@state.setter
def state(self, v):
self._state = v
log.debug("Transitioning to state %s" % v)
def janitor(self, src):
log.debug("Janitor, cleanup aisle 12")
self.kill(src.exception)
def heartbeat_timer(self):
try:
while True:
self.cfg.heartbeatactive.wait()
sleep(self.cfg.heartbeat_interval)
self.heartbeat_event.set()
self.heartbeat_event.clear()
except Exception:
log.critical("heartbeat_timer terminating due to exception", exc_info=True)
raise
def _run(self):
try:
self.state = self.state_startup
while True:
self.state = self.state()
except Exception:
log.critical("PortAgent terminating due to exception", exc_info=True)
raise
finally:
log.debug("Killing orbpkt src")
try: self.orbpktsrc.kill()
except: pass
log.debug("Stopping dataserver")
try: self.dataserver.stop()
except: pass
log.debug("Stopping cmdserver")
try: self.cmdserver.stop()
except: pass
log.debug("PortAgent finally done")
def state_startup(self):
# start cmdserver; err if not cmd port
spawn(self.heartbeat_timer).link_exception(self.janitor)
self.cmdserver = CmdServer((LISTENING_ADDR, self.cfg.command_port),
self.cmdproc.processCmds, self.janitor)
self.cmdserver.start()
return self.state_unconfigured
def state_unconfigured(self):
self.cfg.configuredevent.wait()
return self.state_configured
def state_configured(self):
# spawn orbreapthr
# link to it? what if it dies?
self.orbpktsrc = OrbPktSrc(
srcname = self.cfg.antelope_orb_name,
select = self.cfg.antelope_orb_select,
reject = self.cfg.antelope_orb_reject,
after = self.cfg.antelope_orb_after,
transformation = transform
)
self.orbpktsrc.link_exception(self.janitor)
self.orbpktsrc.start()
# spawn data server
self.dataserver = DataServer(
(LISTENING_ADDR, self.cfg.data_port),
self.orbpktsrc.subscription,
self.heartbeat_event, self.janitor)
self.dataserver.start()
# spawn state_connected
return self.state_connected
def state_connected(self):
# on dataserver config update event
self.cfg.dataserverconfigupdate.wait()
self.orbpktsrc.kill()
self.dataserver.stop()
return self.state_configured
def get_state(self, val, sock):
statestr = self.states[self.state]
sock.sendall(makepacket(PacketType.PORT_AGENT_STATUS, ntp.now(), statestr))
def ping(self, val, sock):
msg = "pong. version: " + __version__
sock.sendall(makepacket(PacketType.PORT_AGENT_STATUS, ntp.now(), msg))
def shutdown(self, val, sock):
self.kill()
def _start_data_server(self, endpoint):
data_server = DataServer()
port = endpoint.split(":")[1]
data_server.start(endpoint=port)