Add setdefault key to ExpiringCache

2024-10-01 11:49:51 -04:00 · 2017-03-10 10:30:38 +00:00 · 2017-03-10 10:30:38 +00:00 · 3545e17f43
commit 3545e17f43
parent 45d173a59a
1 changed files with 7 additions and 0 deletions
--- a/synapse/util/caches/expiringcache.py
+++ b/synapse/util/caches/expiringcache.py
@ -100,6 +100,13 @@ class ExpiringCache(object):
        except KeyError:
            return default

+    def setdefault(self, key, value):
+        try:
+            return self[key]
+        except KeyError:
+            self[key] = value
+            return value
+
    def _prune_cache(self):
        if not self._expiry_ms:
            # zero expiry time means don't expire. This should never get called