Add more caching

judge/caching.py

@@ -1,27 +1,32 @@
 from inspect import signature
-from django.core.cache import cache
+from django.core.cache import cache, caches
 from django.db.models.query import QuerySet
 from django.core.handlers.wsgi import WSGIRequest
 
 import hashlib
 
-MAX_NUM_CHAR = 15
+MAX_NUM_CHAR = 50
 NONE_RESULT = "__None__"
 
 
+def arg_to_str(arg):
+    if hasattr(arg, "id"):
+        return str(arg.id)
+    if isinstance(arg, list) or isinstance(arg, QuerySet):
+        return hashlib.sha1(str(list(arg)).encode()).hexdigest()[:MAX_NUM_CHAR]
+    if len(str(arg)) > MAX_NUM_CHAR:
+        return str(arg)[:MAX_NUM_CHAR]
+    return str(arg)
+
+
+def filter_args(args_list):
+    return [x for x in args_list if not isinstance(x, WSGIRequest)]
+
+
+l0_cache = caches["l0"] if "l0" in caches else None
+
+
 def cache_wrapper(prefix, timeout=None):
-    def arg_to_str(arg):
-        if hasattr(arg, "id"):
-            return str(arg.id)
-        if isinstance(arg, list) or isinstance(arg, QuerySet):
-            return hashlib.sha1(str(list(arg)).encode()).hexdigest()[:MAX_NUM_CHAR]
-        if len(str(arg)) > MAX_NUM_CHAR:
-            return str(arg)[:MAX_NUM_CHAR]
-        return str(arg)
-
-    def filter_args(args_list):
-        return [x for x in args_list if not isinstance(x, WSGIRequest)]
-
     def get_key(func, *args, **kwargs):
         args_list = list(args)
         signature_args = list(signature(func).parameters.keys())
@@ -32,23 +37,41 @@ def cache_wrapper(prefix, timeout=None):
         key = key.replace(" ", "_")
         return key
 
+    def _get(key):
+        if not l0_cache:
+            return cache.get(key)
+        print("GET", key, l0_cache.get(key))
+        return l0_cache.get(key) or cache.get(key)
+
+    def _set_l0(key, value):
+        if l0_cache:
+            print("SET", key, value)
+            l0_cache.set(key, value, 30)
+
+    def _set(key, value, timeout):
+        _set_l0(key, value)
+        cache.set(key, value, timeout)
+
     def decorator(func):
         def wrapper(*args, **kwargs):
             cache_key = get_key(func, *args, **kwargs)
-            result = cache.get(cache_key)
+            result = _get(cache_key)
             if result is not None:
+                _set_l0(cache_key, result)
                 if result == NONE_RESULT:
                     result = None
                 return result
             result = func(*args, **kwargs)
             if result is None:
                 result = NONE_RESULT
-            cache.set(cache_key, result, timeout)
+            _set(cache_key, result, timeout)
             return result
 
         def dirty(*args, **kwargs):
             cache_key = get_key(func, *args, **kwargs)
             cache.delete(cache_key)
+            if l0_cache:
+                l0_cache.delete(cache_key)
 
         wrapper.dirty = dirty