Skip to content
Snippets Groups Projects
Commit d6c75cb7 authored by Erik Johnston's avatar Erik Johnston
Browse files

Rename and comment tree_to_leaves_iterator

parent 1ccd5676
No related branches found
No related tags found
No related merge requests found
......@@ -17,7 +17,7 @@ import logging
from synapse.util.async import ObservableDeferred
from synapse.util import unwrapFirstError
from synapse.util.caches.lrucache import LruCache
from synapse.util.caches.treecache import TreeCache, popped_to_iterator
from synapse.util.caches.treecache import TreeCache, iterate_tree_cache_entry
from synapse.util.logcontext import (
PreserveLoggingContext, preserve_context_over_deferred, preserve_context_over_fn
)
......@@ -181,7 +181,7 @@ class Cache(object):
val = self._pending_deferred_cache.pop(key, None)
if val is not None:
entry_dict, _ = val
for entry in popped_to_iterator(entry_dict):
for entry in iterate_tree_cache_entry(entry_dict):
entry.invalidate()
def invalidate_all(self):
......
......@@ -65,16 +65,19 @@ class TreeCache(object):
return popped
def values(self):
return list(popped_to_iterator(self.root))
return list(iterate_tree_cache_entry(self.root))
def __len__(self):
return self.size
def popped_to_iterator(d):
def iterate_tree_cache_entry(d):
"""Helper function to iterate over the leaves of a tree, i.e. a dict of that
can contain dicts.
"""
if isinstance(d, dict):
for value_d in d.itervalues():
for value in popped_to_iterator(value_d):
for value in iterate_tree_cache_entry(value_d):
yield value
else:
if isinstance(d, _Entry):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment