summary refs log tree commit diff
path: root/synapse/app/user_dir.py
diff options
context:
space:
mode:
authorErik Johnston <erik@matrix.org>2019-12-06 13:08:40 +0000
committerErik Johnston <erik@matrix.org>2019-12-06 13:30:06 +0000
commit9a4fb457cf5918c85068ea249cd2d58b3e2e3cfc (patch)
tree4bb85c953cb523732bcc53ec01b7c1f8209ac1ad /synapse/app/user_dir.py
parentMerge pull request #6469 from matrix-org/erikj/make_database_class (diff)
downloadsynapse-9a4fb457cf5918c85068ea249cd2d58b3e2e3cfc.tar.xz
Change DataStores to accept 'database' param.
Diffstat (limited to 'synapse/app/user_dir.py')
-rw-r--r--synapse/app/user_dir.py5
1 files changed, 3 insertions, 2 deletions
diff --git a/synapse/app/user_dir.py b/synapse/app/user_dir.py
index b6d4481725..c01fb34a9b 100644
--- a/synapse/app/user_dir.py
+++ b/synapse/app/user_dir.py
@@ -43,6 +43,7 @@ from synapse.replication.tcp.streams.events import (
 from synapse.rest.client.v2_alpha import user_directory
 from synapse.server import HomeServer
 from synapse.storage.data_stores.main.user_directory import UserDirectoryStore
+from synapse.storage.database import Database
 from synapse.storage.engines import create_engine
 from synapse.util.caches.stream_change_cache import StreamChangeCache
 from synapse.util.httpresourcetree import create_resource_tree
@@ -60,8 +61,8 @@ class UserDirectorySlaveStore(
     UserDirectoryStore,
     BaseSlavedStore,
 ):
-    def __init__(self, db_conn, hs):
-        super(UserDirectorySlaveStore, self).__init__(db_conn, hs)
+    def __init__(self, database: Database, db_conn, hs):
+        super(UserDirectorySlaveStore, self).__init__(database, db_conn, hs)
 
         events_max = self._stream_id_gen.get_current_token()
         curr_state_delta_prefill, min_curr_state_delta_id = self.db.get_cache_dict(