X-Git-Url: http://average.org/gitweb/?a=blobdiff_plain;f=gps303%2Fwsgateway.py;h=8a5af7f43ec37c94d1500c9ab3a46bf934991e33;hb=ccfb47bba216d7f1b99f1d2e4f488afec4c37738;hp=f9f5c6a2ad936f33c9961419245c65de14414dcd;hpb=c41f98b94321d35cf7f54a9c659fc36b13990f25;p=loctrkd.git diff --git a/gps303/wsgateway.py b/gps303/wsgateway.py index f9f5c6a..8a5af7f 100644 --- a/gps303/wsgateway.py +++ b/gps303/wsgateway.py @@ -17,6 +17,7 @@ from wsproto.utilities import RemoteProtocolError import zmq from . import common +from .backlog import blinit, backlog from .zmsg import LocEvt log = getLogger("gps303/wsgateway") @@ -196,14 +197,7 @@ class Clients: def recv(self, fd): clnt = self.by_fd[fd] - msgs = clnt.recv() - if msgs is None: - return None - result = [] - for msg in msgs: - log.debug("Received: %s", msg) - result.append(msg) - return result + return clnt.recv() def send(self, msg): towrite = set() @@ -230,6 +224,7 @@ class Clients: def runserver(conf): global htmlfile + blinit(conf.get("storage", "dbfn"), conf.get("opencellid", "dbfn")) htmlfile = conf.get("wsgateway", "htmlfile") zctx = zmq.Context() zsub = zctx.socket(zmq.SUB) @@ -280,6 +275,11 @@ def runserver(conf): else: for msg in received: log.debug("Received from %d: %s", sk, msg) + if msg.get("type", None) == "subscribe": + imei = msg.get("imei") + numback = msg.get("backlog", 5) + for elem in imei: + tosend.extend(backlog(elem, numback)) towrite.add(sk) elif fl & zmq.POLLOUT: log.debug("Write now open for fd %d", sk)