X-Git-Url: http://git.maemo.org/git/?p=jamaendo;a=blobdiff_plain;f=jamaui%2Fpostoffice.py;h=a84158b1a57af60ce02dfdbd287a45b5ad749c7f;hp=009c8846e51536f82b864067518c48ec7625f49a;hb=ae451be237b4622abd934a611f5e2dd4d8aec883;hpb=99b9e55a7165a70a2e2f8846a214246d49faf9d0 diff --git a/jamaui/postoffice.py b/jamaui/postoffice.py index 009c884..a84158b 100644 --- a/jamaui/postoffice.py +++ b/jamaui/postoffice.py @@ -23,35 +23,45 @@ # # message central +from __future__ import with_statement +import logging +import threading + +log = logging.getLogger(__name__) + class PostOffice(object): - class Client(object): - def __init__(self): - self.tags = {} - def has_tag(self, tag): - return tag in self.tags - def notify(self, tags, data): - for tag in tags: - cb = self.tags.get(tag) - if cb: - cb(data) - def register(self, tag, callback): - self.tags[tag] = callback def __init__(self): - self.clients = {} - - def notify(self, tags, data): - if not isinstance(tags, list): - tags = [tags] - for client in clients: - client.notify(tags, data) - - def register(self, client_id, tag, callback): - client = self.clients.get(client_id) - if not client: - client = Client() - self.clients[client_id] = client - client.register(tag, callback) + self.lock = threading.RLock() + self.tags = {} # tag -> [callback] + + def notify(self, tag, *data): + with self.lock: + log.info("(%s %s)", tag, " ".join(str(x) for x in data)) + clients = self.tags.get(tag) + if clients: + for ref, client in clients: + client(*data) + + def connect(self, tag, ref, callback): + with self.lock: + if not isinstance(tag, list): + tag = [tag] + for t in tag: + if t not in self.tags: + self.tags[t] = [] + clients = self.tags[t] + if callback not in clients: + clients.append((ref, callback)) + + def disconnect(self, tag, ref): + with self.lock: + if not isinstance(tag, list): + tag = [tag] + for t in tag: + if t not in self.tags: + self.tags[t] = [] + self.tags[t] = [(_ref, cb) for _ref, cb in self.tags[t] if _ref != ref] postoffice = PostOffice()