From c25ec34d732952dcc1a4ecb89652f11a9cd43a48 Mon Sep 17 00:00:00 2001 From: Patrick Cloke Date: Wed, 14 Dec 2022 12:08:21 -0500 Subject: ✨ Magic ✨ MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- synapse/storage/_base.py | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'synapse/storage/_base.py') diff --git a/synapse/storage/_base.py b/synapse/storage/_base.py index 481fec72fe..a30ebda0c8 100644 --- a/synapse/storage/_base.py +++ b/synapse/storage/_base.py @@ -25,6 +25,7 @@ from synapse.util.caches.descriptors import CachedFunction if TYPE_CHECKING: from synapse.server import HomeServer + from synapse.storage.databases import DataStore logger = logging.getLogger(__name__) @@ -44,11 +45,14 @@ class SQLBaseStore(metaclass=ABCMeta): database: DatabasePool, db_conn: LoggingDatabaseConnection, hs: "HomeServer", + datastore: Optional["DataStore"] = None, ): self.hs = hs self._clock = hs.get_clock() self.database_engine = database.engine self.db_pool = database + # A reference back to the root datastore. + self.datastore = datastore self.external_cached_functions: Dict[str, CachedFunction] = {} -- cgit 1.5.1