diff options
| author | reivilibre <oliverw@matrix.org> | 2022-09-07 11:03:32 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-09-07 12:03:32 +0100 |
| commit | d3d9ca156e323fe194b1bcb1af1628f65a2f3c1c (patch) | |
| tree | 74e19b799192c8be69179e27ceadda7a8beac18c /synapse/storage/databases/state | |
| parent | Rename the `EventFormatVersions` enum values so that they line up with room v... (diff) | |
| download | synapse-d3d9ca156e323fe194b1bcb1af1628f65a2f3c1c.tar.xz | |
Cancel the processing of key query requests when they time out. (#13680)
Diffstat (limited to 'synapse/storage/databases/state')
| -rw-r--r-- | synapse/storage/databases/state/store.py | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/synapse/storage/databases/state/store.py b/synapse/storage/databases/state/store.py index bb64543c1f..f8cfcaca83 100644 --- a/synapse/storage/databases/state/store.py +++ b/synapse/storage/databases/state/store.py @@ -31,6 +31,7 @@ from synapse.storage.util.sequence import build_sequence_generator from synapse.types import MutableStateMap, StateKey, StateMap from synapse.util.caches.descriptors import cached from synapse.util.caches.dictionary_cache import DictionaryCache +from synapse.util.cancellation import cancellable if TYPE_CHECKING: from synapse.server import HomeServer @@ -156,6 +157,7 @@ class StateGroupDataStore(StateBackgroundUpdateStore, SQLBaseStore): "get_state_group_delta", _get_state_group_delta_txn ) + @cancellable async def _get_state_groups_from_groups( self, groups: List[int], state_filter: StateFilter ) -> Dict[int, StateMap[str]]: @@ -235,6 +237,7 @@ class StateGroupDataStore(StateBackgroundUpdateStore, SQLBaseStore): return state_filter.filter_state(state_dict_ids), not missing_types + @cancellable async def _get_state_for_groups( self, groups: Iterable[int], state_filter: Optional[StateFilter] = None ) -> Dict[int, MutableStateMap[str]]: |