Add DataStores and Storage classes.

This commit is contained in:
Erik Johnston 2019-10-23 16:14:16 +01:00
parent 22a9847670
commit 73cf63784b
3 changed files with 34 additions and 4 deletions

View file

@ -29,6 +29,7 @@ from synapse.api.constants import EventTypes
from synapse.logging.context import PreserveLoggingContext, make_deferred_yieldable
from synapse.metrics.background_process_metrics import run_as_background_process
from synapse.state import StateResolutionStore
from synapse.storage.data_stores import DataStores
from synapse.util.async_helpers import ObservableDeferred
from synapse.util.metrics import Measure
@ -171,12 +172,12 @@ class _EventPeristenceQueue(object):
class EventsPersistenceStore(object):
def __init__(self, hs):
def __init__(self, hs, stores: DataStores):
# We ultimately want to split out the state store from the main store,
# so we use separate variables here even though they point to the same
# store for now.
self.main_store = hs.get_datastore()
self.state_store = hs.get_datastore()
self.main_store = stores.main
self.state_store = stores.main
self._clock = hs.get_clock()
self.is_mine_id = hs.is_mine_id