|
@@ -698,10 +698,17 @@ class UserDirectoryBackgroundUpdateStore(StateDeltasStore):
|
|
|
"""Delete the entire user directory"""
|
|
|
|
|
|
def _delete_all_from_user_dir_txn(txn: LoggingTransaction) -> None:
|
|
|
- txn.execute("DELETE FROM user_directory")
|
|
|
- txn.execute("DELETE FROM user_directory_search")
|
|
|
- txn.execute("DELETE FROM users_in_public_rooms")
|
|
|
- txn.execute("DELETE FROM users_who_share_private_rooms")
|
|
|
+ # SQLite doesn't support TRUNCATE.
|
|
|
+ # On Postgres, DELETE FROM does a table scan but TRUNCATE is more efficient.
|
|
|
+ truncate = (
|
|
|
+ "DELETE FROM"
|
|
|
+ if isinstance(self.database_engine, Sqlite3Engine)
|
|
|
+ else "TRUNCATE"
|
|
|
+ )
|
|
|
+ txn.execute(f"{truncate} user_directory")
|
|
|
+ txn.execute(f"{truncate} user_directory_search")
|
|
|
+ txn.execute(f"{truncate} users_in_public_rooms")
|
|
|
+ txn.execute(f"{truncate} users_who_share_private_rooms")
|
|
|
txn.call_after(self.get_user_in_directory.invalidate_all)
|
|
|
|
|
|
await self.db_pool.runInteraction(
|