Do better

Author: Mathieu Velten

synapse/storage/databases/main/deviceinbox.py

@@ -477,14 +477,11 @@ async def delete_messages_for_device(
                 log_kv({"message": "No changes in cache since last check"})
                 return 0
 
-        ROW_ID_LITERAL = "ctid" if isinstance(self.database_engine, PostgresEngine) else "rowid"
-
         def delete_messages_for_device_txn(txn: LoggingTransaction) -> int:
             sql = (
-                f"DELETE FROM device_inbox WHERE {ROW_ID_LITERAL} IN ("
-                f"SELECT {ROW_ID_LITERAL} FROM device_inbox"
-                " WHERE user_id = ? AND device_id = ?"
-                " AND stream_id <= ?"
+                f"DELETE FROM device_inbox WHERE {self.database_engine.row_id_name} IN ("
+                f"SELECT {self.database_engine.row_id_name} FROM device_inbox"
+                " WHERE user_id = ? AND device_id = ? AND stream_id <= ?"
             )
             if limit:
                 sql += f" LIMIT {limit}"

synapse/storage/engines/_base.py

@@ -100,6 +100,12 @@ def server_version(self) -> str:
         """Gets a string giving the server version. For example: '3.22.0'"""
         ...
 
+    @property
+    @abc.abstractmethod
+    def row_id_name(self) -> str:
+        """Gets the literal name representing a row id for this engine."""
+        ...
+
     @abc.abstractmethod
     def in_transaction(self, conn: ConnectionType) -> bool:
         """Whether the connection is currently in a transaction."""

synapse/storage/engines/postgres.py

@@ -211,6 +211,10 @@ def server_version(self) -> str:
         else:
             return "%i.%i.%i" % (numver / 10000, (numver % 10000) / 100, numver % 100)
 
+    @property
+    def row_id_name(self) -> str:
+        return "ctid"
+
     def in_transaction(self, conn: psycopg2.extensions.connection) -> bool:
         return conn.status != psycopg2.extensions.STATUS_READY
 

synapse/storage/engines/sqlite.py

@@ -123,6 +123,10 @@ def server_version(self) -> str:
         """Gets a string giving the server version. For example: '3.22.0'."""
         return "%i.%i.%i" % sqlite3.sqlite_version_info
 
+    @property
+    def row_id_name(self) -> str:
+        return "rowid"
+
     def in_transaction(self, conn: sqlite3.Connection) -> bool:
         return conn.in_transaction
 

synapse/storage/schema/main/delta/48/group_unique_indexes.py

@@ -37,7 +37,7 @@
 
 
 def run_create(cur: LoggingTransaction, database_engine: BaseDatabaseEngine) -> None:
-    rowid = "ctid" if isinstance(database_engine, PostgresEngine) else "rowid"
+    rowid = database_engine.row_id_name
 
     # remove duplicates from group_users & group_invites tables
     cur.execute(