add local broker and republisher

This commit is contained in:
Wolfgang Hottgenroth
2016-07-03 22:27:58 +02:00
parent 6c6f6c7e62
commit d3aa892f37
3 changed files with 82 additions and 22 deletions

View File

@ -6,6 +6,9 @@ Created on 06.03.2016
import Queue
import threading
from logger import Logger
class BrokerException(Exception): pass
@ -16,15 +19,33 @@ class BrokerNotSubscribedException(BrokerException): pass
class BrokerOverflowException(BrokerException): pass
class Broker(object):
class Broker(threading.Thread):
def __init__(self):
threading.Thread.__init__(self)
self.setDaemon(True)
self.outQueues = {}
self.inQueue = Queue.Queue
self.inQueue = Queue.Queue()
def run(self):
while True:
try:
item = self.inQueue.get(True, None)
Logger.log("got an item")
for (n, q) in self.outQueues.iteritems():
Logger.log("passed to %s" % n)
q.put_nowait(item)
except Queue.Full:
pass
def getInQueue(self):
return self.inQueue
def subscribe(self, name):
if self.outQueues.has_key(name):
raise BrokerDuplicateIdException()
self.outQueues[name] = Queue.Queue()
return self.outQueues[name]
def unsubscribe(self, name):
try:
@ -32,16 +53,3 @@ class Broker(object):
except KeyError:
raise BrokerNotSubscribedException
def get(self, name):
try:
return self.outQueues[name].get()
except KeyError:
raise BrokerNotSubscribedException
def put(self, msg):
try:
for q in self.outQueues.values():
q.put_nowait(msg)
except Queue.Full: