mirrors/synapse
1
0
Fork 0
mirror of https://github.com/element-hq/synapse.git synced 2024-11-22 09:35:45 +03:00
Code Issues Projects Releases Packages Wiki Activity

Check events_and_context for state events

Browse source 
See https://github.com/element-hq/synapse/pull/17512#discussion_r1726798803
This commit is contained in:
Eric Eastwood 2024-08-22 11:34:40 -05:00
parent 339500d067
commit 9b8d2017af
1 changed files with 23 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
synapse/storage/databases/main/events.py
View file

@ -468,14 +468,30 @@ class PersistEventsStore:
if state_key in SLIDING_SYNC_RELEVANT_STATE_SET:
current_state_ids_map[state_key] = event_id
fetched_events = await self.store.get_events(
current_state_ids_map.values()
)
current_state_map: MutableStateMap[EventBase] = {}
# In normal event persist scenarios, we probably won't be able to find
# these state events in `events_and_contexts` since we don't generally
# batch up local membership changes with other events, but it can
# happen.
missing_state_event_ids: Set[str] = set()
for state_key, event_id in current_state_ids_map.items():
event = event_map.get(event_id)
if event:
current_state_map[state_key] = event
else:
missing_state_event_ids.add(event_id)
current_state_map: StateMap[EventBase] = {
state_key: fetched_events[event_id]
for state_key, event_id in current_state_ids_map.items()
}
# Otherwise, we need to find a couple events
if missing_state_event_ids:
remaining_events = await self.store.get_events(
missing_state_event_ids
)
# There shouldn't be any missing events
assert (
remaining_events.keys() == missing_state_event_ids
), missing_state_event_ids.difference(remaining_events.keys())
for event in remaining_events.values():
current_state_map[(event.type, event.state_key)] = event
if current_state_map:
state_insert_values = PersistEventsStore._get_sliding_sync_insert_values_from_state_map(