diff options
author | Erik Johnston <erik@matrix.org> | 2016-07-21 10:30:12 +0100 |
---|---|---|
committer | Erik Johnston <erik@matrix.org> | 2016-07-21 10:30:12 +0100 |
commit | 248e6770ca0faadf574cfd62f72d8e200cb5b57a (patch) | |
tree | 7bc7a6c0173fe0366c8c029d444da5d3f7609682 /synapse/util | |
parent | Merge pull request #938 from matrix-org/rav/add_device_id_to_client_ips (diff) | |
download | synapse-248e6770ca0faadf574cfd62f72d8e200cb5b57a.tar.xz |
Cache federation state responses
Diffstat (limited to 'synapse/util')
-rw-r--r-- | synapse/util/caches/response_cache.py | 13 |
1 files changed, 11 insertions, 2 deletions
diff --git a/synapse/util/caches/response_cache.py b/synapse/util/caches/response_cache.py index 36686b479e..00af539880 100644 --- a/synapse/util/caches/response_cache.py +++ b/synapse/util/caches/response_cache.py @@ -24,9 +24,12 @@ class ResponseCache(object): used rather than trying to compute a new response. """ - def __init__(self): + def __init__(self, hs, timeout_ms=0): self.pending_result_cache = {} # Requests that haven't finished yet. + self.clock = hs.get_clock() + self.timeout_sec = timeout_ms / 1000. + def get(self, key): result = self.pending_result_cache.get(key) if result is not None: @@ -39,7 +42,13 @@ class ResponseCache(object): self.pending_result_cache[key] = result def remove(r): - self.pending_result_cache.pop(key, None) + if self.timeout_sec: + self.clock.call_later( + self.timeout_sec, + self.pending_result_cache.pop, key, None, + ) + else: + self.pending_result_cache.pop(key, None) return r result.addBoth(remove) |