libervia-pubsub: idavoll/pgsql_backend.py comparison

Add get_affiliations() and get_subscriptions() to Storage. Add AffiliationsService as a subclass of backend.AffiliationsService.

comparison

equal deleted inserted replaced

-:ad08e021b4d5
+:8cdbc27c467a
 except Exception, e:
 print e
 return None
+def get_affiliations(self, entity):
+return self.dbpool.runQuery("""SELECT node, affiliation FROM entities
+JOIN affiliations ON
+(affiliations.entity_id=entities.id)
+JOIN nodes ON
+(nodes.id=affiliations.node_id)
+WHERE jid=%s""",
+(entity.full().encode('utf8'),))
+def get_subscriptions(self, entity):
+d = self.dbpool.runQuery("""SELECT node, jid, resource, subscription
+FROM entities JOIN subscriptions ON
+(subscriptions.entity_id=entities.id)
+JOIN nodes ON
+(nodes.id=subscriptions.node_id)
+WHERE jid=%s""",
+(entity.full().encode('utf8'),))
+d.addCallback(self._convert_subscription_jids)
+return d
+def _convert_subscription_jids(self, subscriptions):
+return [(node, jid.JID('%s/%s' % (subscriber, resource)), subscription)
+for node, subscriber, resource, subscription in subscriptions]
 class BackendService(backend.BackendService):
 """ PostgreSQL backend Service for a JEP-0060 pubsub service """
 class NodeCreationService(backend.NodeCreationService):
 class NotificationService(backend.NotificationService):
 pass
 class SubscriptionService(backend.SubscriptionService):
 pass
+class AffiliationsService(backend.AffiliationsService):
+pass

Mercurial > libervia-pubsub