MirrorHub/synapse
0
0
Fork 1
mirror of https://mau.dev/maunium/synapse.git synced 2024-11-12 04:52:26 +01:00
Code Issues Projects Releases Wiki Activity

Revert all the bits changing keys of eeverything that used LRUCaches to tuples

Browse source
This commit is contained in:
David Baker 2016-01-22 12:18:14 +00:00
parent 10f76dc5da
commit d552861346
5 changed files with 44 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
synapse/push/push_rule_evaluator.py
View file

@ -309,14 +309,14 @@ def _flatten_dict(d, prefix=[], result={}):
return result return result
regex_cache = LruCache(5000, 1) regex_cache = LruCache(5000)
def _compile_regex(regex_str): def _compile_regex(regex_str):
r = regex_cache.get((regex_str,), None) r = regex_cache.get(regex_str, None)
if r: if r:
return r return r
r = re.compile(regex_str, flags=re.IGNORECASE) r = re.compile(regex_str, flags=re.IGNORECASE)
regex_cache[(regex_str,)] = r regex_cache[regex_str] = r
return r return r

10
synapse/util/caches/dictionary_cache.py
View file

@ -32,7 +32,7 @@ class DictionaryCache(object):
""" """
def __init__(self, name, max_entries=1000): def __init__(self, name, max_entries=1000):
self.cache = LruCache(max_size=max_entries, keylen=1) self.cache = LruCache(max_size=max_entries)
self.name = name self.name = name
self.sequence = 0 self.sequence = 0
@ -56,7 +56,7 @@ class DictionaryCache(object):
) )
def get(self, key, dict_keys=None): def get(self, key, dict_keys=None):
entry = self.cache.get((key,), self.sentinel) entry = self.cache.get(key, self.sentinel)
if entry is not self.sentinel: if entry is not self.sentinel:
cache_counter.inc_hits(self.name) cache_counter.inc_hits(self.name)
@ -78,7 +78,7 @@ class DictionaryCache(object):
# Increment the sequence number so that any SELECT statements that # Increment the sequence number so that any SELECT statements that
# raced with the INSERT don't update the cache (SYN-369) # raced with the INSERT don't update the cache (SYN-369)
self.sequence += 1 self.sequence += 1
self.cache.pop((key,), None) self.cache.pop(key, None)
def invalidate_all(self): def invalidate_all(self):
self.check_thread() self.check_thread()
@ -96,8 +96,8 @@ class DictionaryCache(object):
self._update_or_insert(key, value) self._update_or_insert(key, value)
def _update_or_insert(self, key, value): def _update_or_insert(self, key, value):
entry = self.cache.setdefault((key,), DictionaryEntry(False, {})) entry = self.cache.setdefault(key, DictionaryEntry(False, {}))
entry.value.update(value) entry.value.update(value)
def _insert(self, key, value): def _insert(self, key, value):
self.cache[(key,)] = DictionaryEntry(True, value) self.cache[key] = DictionaryEntry(True, value)

2
synapse/util/caches/lrucache.py
View file

@ -29,7 +29,7 @@ def enumerate_leaves(node, depth):
class LruCache(object): class LruCache(object):
"""Least-recently-used cache.""" """Least-recently-used cache."""
def __init__(self, max_size, keylen, cache_type=dict): def __init__(self, max_size, keylen=1, cache_type=dict):
cache = cache_type() cache = cache_type()
self.size = 0 self.size = 0
list_root = [] list_root = []

26
tests/storage/test__base.py
View file

@ -56,42 +56,42 @@ class CacheTestCase(unittest.TestCase):
def test_eviction(self): def test_eviction(self):
cache = Cache("test", max_entries=2) cache = Cache("test", max_entries=2)
cache.prefill((1,), "one") cache.prefill(1, "one")
cache.prefill((2,), "two") cache.prefill(2, "two")
cache.prefill((3,), "three") # 1 will be evicted cache.prefill(3, "three") # 1 will be evicted
failed = False failed = False
try: try:
cache.get((1,)) cache.get(1)
except KeyError: except KeyError:
failed = True failed = True
self.assertTrue(failed) self.assertTrue(failed)
cache.get((2,)) cache.get(2)
cache.get((3,)) cache.get(3)
def test_eviction_lru(self): def test_eviction_lru(self):
cache = Cache("test", max_entries=2, lru=True) cache = Cache("test", max_entries=2, lru=True)
cache.prefill((1,), "one") cache.prefill(1, "one")
cache.prefill((2,), "two") cache.prefill(2, "two")
# Now access 1 again, thus causing 2 to be least-recently used # Now access 1 again, thus causing 2 to be least-recently used
cache.get((1,)) cache.get(1)
cache.prefill((3,), "three") cache.prefill(3, "three")
failed = False failed = False
try: try:
cache.get((2,)) cache.get(2)
except KeyError: except KeyError:
failed = True failed = True
self.assertTrue(failed) self.assertTrue(failed)
cache.get((1,)) cache.get(1)
cache.get((3,)) cache.get(3)
class CacheDecoratorTestCase(unittest.TestCase): class CacheDecoratorTestCase(unittest.TestCase):

44
tests/util/test_lrucache.py
View file

@ -22,37 +22,37 @@ from synapse.util.caches.treecache import TreeCache
class LruCacheTestCase(unittest.TestCase): class LruCacheTestCase(unittest.TestCase):
def test_get_set(self): def test_get_set(self):
cache = LruCache(1, 1) cache = LruCache(1)
cache[("key",)] = "value" cache["key"] = "value"
self.assertEquals(cache.get(("key",)), "value") self.assertEquals(cache.get("key"), "value")
self.assertEquals(cache[("key",)], "value") self.assertEquals(cache["key"], "value")
def test_eviction(self): def test_eviction(self):
cache = LruCache(2, 1) cache = LruCache(2)
cache[(1,)] = 1 cache[1] = 1
cache[(2,)] = 2 cache[2] = 2
self.assertEquals(cache.get((1,)), 1) self.assertEquals(cache.get(1), 1)
self.assertEquals(cache.get((2,)), 2) self.assertEquals(cache.get(2), 2)
cache[(3,)] = 3 cache[3] = 3
self.assertEquals(cache.get((1,)), None) self.assertEquals(cache.get(1), None)
self.assertEquals(cache.get((2,)), 2) self.assertEquals(cache.get(2), 2)
self.assertEquals(cache.get((3,)), 3) self.assertEquals(cache.get(3), 3)
def test_setdefault(self): def test_setdefault(self):
cache = LruCache(1, 1) cache = LruCache(1)
self.assertEquals(cache.setdefault(("key",), 1), 1) self.assertEquals(cache.setdefault("key", 1), 1)
self.assertEquals(cache.get(("key",)), 1) self.assertEquals(cache.get("key"), 1)
self.assertEquals(cache.setdefault(("key",), 2), 1) self.assertEquals(cache.setdefault("key", 2), 1)
self.assertEquals(cache.get(("key",)), 1) self.assertEquals(cache.get("key"), 1)
def test_pop(self): def test_pop(self):
cache = LruCache(1, 1) cache = LruCache(1)
cache[("key",)] = 1 cache["key"] = 1
self.assertEquals(cache.pop(("key",)), 1) self.assertEquals(cache.pop("key"), 1)
self.assertEquals(cache.pop(("key",)), None) self.assertEquals(cache.pop("key"), None)
def test_del_multi(self): def test_del_multi(self):
cache = LruCache(4, 2, cache_type=TreeCache) cache = LruCache(4, 2, cache_type=TreeCache)