source: ralphm-patches/pubsub_resource_example.patch @ 54:03ec57713c90

new file doc/examples/pubsub_resource.tac

@@ +1 @@
+from twisted.application import service
+from twisted.words.protocols.jabber.error import StanzaError
+from wokkel import pubsub
+
+class CounterService(service.Service):
+
+    pubsubService = None
+
+    def __init__(self):
+        self.counters = {}
+        self.lc = None
+
+
+    def startService(self):
+        self.lc = task.LoopingCall(self.updateCounters)
+        self.lc.start(1)
+
+
+    def updateCounters(self):
+        for key in self.counters:
+            self.counters[key] += 1
+            self.pubsubService.publishCounters(self.counters)
+
+
+class Backend(pubsub.PubSubResource):
+    """
+    Business logic for XMPP Publish-Subscribe nodes.
+    """
+
+    features = [
+            "subscribe",
+            ]
+
+    discoIdentity = disco.DiscoIdentity('pubsub', 'service',
+                                        'Example publish-subscribe service')
+
+    def __init__(self):
+       self.nodes = {}
+       self.subscriptions = {}
+
+    def subscribe(self, request):
+        if request.nodeIdentifier == "":
+            raise StanzaError('forbidden')
+
+        # TODO: check subscriber against sender
+
+        # Get or create subscription
+        try:
+            nodeSubscriptions = self.subscriptions[request.nodeIdentifier]
+        except KeyError:
+            nodeSubscriptions = {}
+            self.subscriptions[request.nodeIdentifier] = nodeSubscriptions
+
+        try:
+            subscription = nodeSubscriptions[request.subscriber]
+        except:
+            subscription = pubsub.Subscription(request.nodeIdentifier,
+                                               request.subscriber,
+                                               'subscribed')
+            nodeSubscriptions[request.subscriber] = subscription
+
+        # Initialize the node if it does not exist yet.
+        if request.nodeIdentifier not in self.nodes:
+            self.nodes[request.nodeIdentifier] = 0
+
+        return subscription
+
+
+    def publishCounters(self, counters):
+        for key in counters:
+            if key in self.subscriptions(
+