Store absence of state in cache

This commit is contained in:
Erik Johnston 2015-08-07 10:17:38 +01:00
parent 1d08bf7c17
commit fe994e728f

View File

@ -234,7 +234,8 @@ class StateStore(SQLBaseStore):
])
sql = (
"SELECT sg.event_id FROM state_groups_state as sg"
"SELECT e.event_id, sg.state_group, sg.event_id"
" FROM state_groups_state as sg"
" INNER JOIN event_to_state_groups as e"
" ON e.state_group = sg.state_group"
" WHERE e.event_id = ? AND (%s)"
@ -342,8 +343,9 @@ class StateStore(SQLBaseStore):
defer.returnValue(state_dict)
if is_all or (types is not None and not missing_types):
def include(typ, state_key):
sentinel = object()
def include(typ, state_key):
valid_state_keys = type_to_key.get(typ, sentinel)
if valid_state_keys is sentinel:
return False
@ -356,20 +358,24 @@ class StateStore(SQLBaseStore):
defer.returnValue({
k: v
for k, v in state_dict.items()
if include(k[0], k[1])
if v and include(k[0], k[1])
})
# Okay, so we have some missing_types, lets fetch them.
cache_seq_num = self._state_group_cache.sequence
_, state_ids = yield self._get_state_groups_from_group(
group,
frozenset(types) if types else None
frozenset(missing_types) if types else None
)
state_events = yield self._get_events(state_ids, get_prev_content=False)
state_dict = {
key: None
for key in missing_types
}
state_dict.update({
(e.type, e.state_key): e
for e in state_events
}
})
# Update the cache
self._state_group_cache.update(
@ -379,7 +385,11 @@ class StateStore(SQLBaseStore):
full=(types is None),
)
defer.returnValue(state_dict)
defer.returnValue({
key: value
for key, value in state_dict.items()
if value
})
def _make_group_id(clock):