40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
|
from threading import currentThread
|
||
|
|
|
||
|
|
from django.core.cache.backends.locmem import LocMemCache
|
||
|
|
|
||
|
|
|
||
|
|
_request_cache = {}
|
||
|
|
|
||
|
|
|
||
|
|
class RequestCache(LocMemCache):
|
||
|
|
""" LocMemCache is a threadsafe local memory cache """
|
||
|
|
def __init__(self):
|
||
|
|
name = 'locmemcache@%i' % hash(currentThread())
|
||
|
|
super(RequestCache, self).__init__(name, {})
|
||
|
|
|
||
|
|
|
||
|
|
def get_request_cache():
|
||
|
|
"""
|
||
|
|
Returns per-request cache when running RequestCacheMiddleware otherwise a
|
||
|
|
new LocMemCache instance (when running periodic tasks or shell)
|
||
|
|
"""
|
||
|
|
try:
|
||
|
|
return _request_cache[currentThread()]
|
||
|
|
except KeyError:
|
||
|
|
cache = RequestCache()
|
||
|
|
_request_cache[currentThread()] = cache
|
||
|
|
return cache
|
||
|
|
|
||
|
|
|
||
|
|
class RequestCacheMiddleware(object):
|
||
|
|
def process_request(self, request):
|
||
|
|
cache = _request_cache.get(currentThread(), RequestCache())
|
||
|
|
_request_cache[currentThread()] = cache
|
||
|
|
cache.clear()
|
||
|
|
|
||
|
|
def process_response(self, request, response):
|
||
|
|
# TODO not sure if this actually saves memory, remove otherwise
|
||
|
|
if currentThread() in _request_cache:
|
||
|
|
_request_cache[currentThread()].clear()
|
||
|
|
return response
|