diff options
| author | Erik Johnston <erik@matrix.org> | 2019-12-06 13:08:40 +0000 |
|---|---|---|
| committer | Erik Johnston <erik@matrix.org> | 2019-12-06 13:30:06 +0000 |
| commit | 9a4fb457cf5918c85068ea249cd2d58b3e2e3cfc (patch) | |
| tree | 4bb85c953cb523732bcc53ec01b7c1f8209ac1ad /synapse/replication/slave/storage/receipts.py | |
| parent | Merge pull request #6469 from matrix-org/erikj/make_database_class (diff) | |
| download | synapse-9a4fb457cf5918c85068ea249cd2d58b3e2e3cfc.tar.xz | |
Change DataStores to accept 'database' param.
Diffstat (limited to 'synapse/replication/slave/storage/receipts.py')
| -rw-r--r-- | synapse/replication/slave/storage/receipts.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/synapse/replication/slave/storage/receipts.py b/synapse/replication/slave/storage/receipts.py index 43d823c601..d40dc6e1f5 100644 --- a/synapse/replication/slave/storage/receipts.py +++ b/synapse/replication/slave/storage/receipts.py @@ -15,6 +15,7 @@ # limitations under the License. from synapse.storage.data_stores.main.receipts import ReceiptsWorkerStore +from synapse.storage.database import Database from ._base import BaseSlavedStore from ._slaved_id_tracker import SlavedIdTracker @@ -29,14 +30,14 @@ from ._slaved_id_tracker import SlavedIdTracker class SlavedReceiptsStore(ReceiptsWorkerStore, BaseSlavedStore): - def __init__(self, db_conn, hs): + def __init__(self, database: Database, db_conn, hs): # We instantiate this first as the ReceiptsWorkerStore constructor # needs to be able to call get_max_receipt_stream_id self._receipts_id_gen = SlavedIdTracker( db_conn, "receipts_linearized", "stream_id" ) - super(SlavedReceiptsStore, self).__init__(db_conn, hs) + super(SlavedReceiptsStore, self).__init__(database, db_conn, hs) def get_max_receipt_stream_id(self): return self._receipts_id_gen.get_current_token() |