summary refs log tree commit diff
path: root/synapse/storage/pusher.py
diff options
context:
space:
mode:
Diffstat (limited to 'synapse/storage/pusher.py')
-rw-r--r--synapse/storage/pusher.py62
1 files changed, 30 insertions, 32 deletions
diff --git a/synapse/storage/pusher.py b/synapse/storage/pusher.py
index 90ec06809a..774eda0f3d 100644
--- a/synapse/storage/pusher.py
+++ b/synapse/storage/pusher.py
@@ -56,43 +56,41 @@ class PusherStore(SQLBaseStore):
         )
         defer.returnValue(ret is not None)
 
-    @defer.inlineCallbacks
     def get_pushers_by_app_id_and_pushkey(self, app_id, pushkey):
-        def r(txn):
-            sql = (
-                "SELECT * FROM pushers"
-                " WHERE app_id = ? AND pushkey = ?"
-            )
-
-            txn.execute(sql, (app_id, pushkey,))
-            rows = self.cursor_to_dict(txn)
-
-            return self._decode_pushers_rows(rows)
-
-        rows = yield self.runInteraction(
-            "get_pushers_by_app_id_and_pushkey", r
-        )
-
-        defer.returnValue(rows)
+        return self.get_pushers_by({
+            "app_id": app_id,
+            "pushkey": pushkey,
+        })
 
-    @defer.inlineCallbacks
     def get_pushers_by_user_id(self, user_id):
-        def r(txn):
-            sql = (
-                "SELECT * FROM pushers"
-                " WHERE user_name = ?"
-            )
-
-            txn.execute(sql, (user_id,))
-            rows = self.cursor_to_dict(txn)
-
-            return self._decode_pushers_rows(rows)
+        return self.get_pushers_by({
+            "user_name": user_id,
+        })
 
-        result = yield self.runInteraction(
-            "get_pushers_by_user_id", r
+    @defer.inlineCallbacks
+    def get_pushers_by(self, keyvalues):
+        ret = yield self._simple_select_list(
+            "pushers", keyvalues,
+            [
+                "id",
+                "user_name",
+                "access_token",
+                "profile_tag",
+                "kind",
+                "app_id",
+                "app_display_name",
+                "device_display_name",
+                "pushkey",
+                "ts",
+                "lang",
+                "data",
+                "last_stream_ordering",
+                "last_success",
+                "failing_since",
+
+            ]
         )
-
-        defer.returnValue(result)
+        defer.returnValue(self._decode_pushers_rows(ret))
 
     @defer.inlineCallbacks
     def get_all_pushers(self):