diff options
| -rw-r--r-- | src/daq/flumotion-plug/sfive.py | 32 |
1 files changed, 28 insertions, 4 deletions
diff --git a/src/daq/flumotion-plug/sfive.py b/src/daq/flumotion-plug/sfive.py index a192f92..6292ad0 100644 --- a/src/daq/flumotion-plug/sfive.py +++ b/src/daq/flumotion-plug/sfive.py @@ -36,6 +36,8 @@ from flumotion.component.plugs import base from flumotion.common import messages, i18n, log from flumotion.common.poller import Poller +from twisted.internet import protocol, reactor + from flumotion.common.i18n import N_ T_ = i18n.gettexter() @@ -43,6 +45,26 @@ _DEFAULT_POLL_INTERVAL = 5 # in seconds __version__ = "$Rev$" +class SFiveProto(protocol.ConnectedDatagramProtocol): + plug = None + + def __init__(self, plug): + self.plug = plug + + def stopProtocol(self): + self.plug._socketError() + + def startProtocol(self): + self.plug._socketReady() + + def datagramReceived(self, data): + self.plug.debug('SFive: received datagram: "%s" (will get ignored)', data) + + def connectionFailed(self, failure): + self.plug.warning('SFive: "%s"', failure.getErrorMessage()) + self.plug._socketError() + + class ComponentSFivePlug(base.ComponentPlug): """Class to send statistics to the spreadspace streaming statistic suite""" @@ -76,12 +98,13 @@ class ComponentSFivePlug(base.ComponentPlug): def _initSocket(self): self.debug('SFive: connecting to %s...', self._socket) - ## TODO: init unix socket and call _socketReady when socket is ready - ## TODO: how handle connection drop??? - self._socketReady() + self._cp = SFiveProto(self) + c = reactor.connectUNIXDatagram(self._socket, self._cp) def _socketError(self): - self.warning('SFive: connection lost... trying reconnect') + self.warning('SFive: connection error ... trying reconnect') + if self._sfivepoller: + self._sfivepoller.stop() ## TODO: try reconnect... def _socketReady(self): @@ -109,6 +132,7 @@ class ComponentSFivePlug(base.ComponentPlug): self._old_bytes_sent = bytes_sent self.debug('SFive: sending data (client-count: %d, bytes_received: %d, bytes_sent: %d)', client_count, bytes_received_diff, bytes_sent_diff) + self._cp.transport.write('client-count: %d, bytes_received: %d, bytes_sent: %d\n' % (client_count, bytes_received_diff, bytes_sent_diff)); # TODO: create json and send it out # { # "start-time": "2014-08-03Z12:34:56.123", |