Set reactor clock into the class instance

realtyem · Nov 13, 2023 · ba5856c · ba5856c
1 parent f3ebc26
commit ba5856c
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/synapse/storage/databases/main/event_federation.py b/synapse/storage/databases/main/event_federation.py
@@ -130,6 +130,7 @@ def __init__(
         super().__init__(database, db_conn, hs)
 
         self.hs = hs
+        self._clock = hs.get_clock()
 
         if hs.config.worker.run_background_tasks:
             hs.get_clock().looping_call(
@@ -289,7 +290,7 @@ def _get_auth_chain_ids_using_cover_index_txn(
         s1 = self._authchain_event_id_to_chain_info.set
 
         with Measure(
-            self.hs.get_clock(),
+            self._clock,
             "_get_auth_chain_ids_using_cover_index_txn.section_1_all",
         ):
             initial_events_copy = set(initial_events)
@@ -366,7 +367,7 @@ def _get_auth_chain_ids_using_cover_index_txn(
         s2 = self._authchain_links_list.set
 
         with Measure(
-            self.hs.get_clock(),
+            self._clock,
             "_get_auth_chain_ids_using_cover_index_txn.section_2_all",
         ):
             # 'event_chains' represents the origin chain_id and maximum sequence number
@@ -468,7 +469,7 @@ def _get_auth_chain_ids_using_cover_index_txn(
         s3 = self._authchain_chain_info_to_event_id.set
 
         with Measure(
-            self.hs.get_clock(),
+            self._clock,
             "_get_auth_chain_ids_using_cover_index_txn.section_3_all",
         ):
             while len(chains) != 0: