try and make loading items from the cache faster

This commit is contained in:
Amber Brown 2018-06-26 17:25:34 +01:00
parent 1d62c4a127
commit abfe4b2957
1 changed files with 4 additions and 2 deletions

View File

@ -78,7 +78,8 @@ class StreamChangeCache(object):
not_known_entities = set(entities) - set(self._entity_to_key)
result = (
set(self._cache.values()[self._cache.bisect_right(stream_pos) :])
{self._cache[k] for k in in self._cache.islice(
start=self._cache.bisect_right(stream_pos))}
.intersection(entities)
.union(not_known_entities)
)
@ -113,7 +114,8 @@ class StreamChangeCache(object):
assert type(stream_pos) is int
if stream_pos >= self._earliest_known_stream_pos:
return self._cache.values()[self._cache.bisect_right(stream_pos) :]
return {self._cache[k] for k in in self._cache.islice(
start=self._cache.bisect_right(stream_pos))}
else:
return None