add local broker and republisher
This commit is contained in:
38
Broker.py
38
Broker.py
@ -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:
|
||||
|
||||
|
Reference in New Issue
Block a user