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

View File

@ -78,7 +78,8 @@ class StreamChangeCache(object):
not_known_entities = set(entities) - set(self._entity_to_key) not_known_entities = set(entities) - set(self._entity_to_key)
result = ( 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) .intersection(entities)
.union(not_known_entities) .union(not_known_entities)
) )
@ -113,7 +114,8 @@ class StreamChangeCache(object):
assert type(stream_pos) is int assert type(stream_pos) is int
if stream_pos >= self._earliest_known_stream_pos: 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: else:
return None return None