53 lines
1.7 KiB
Python
53 lines
1.7 KiB
Python
# coding: utf-8
|
|
|
|
from __future__ import unicode_literals
|
|
from threading import local
|
|
|
|
from django.core.cache import caches
|
|
from django.db import connections
|
|
|
|
from .settings import cachalot_settings
|
|
from .transaction import AtomicCache
|
|
from .utils import _get_table_cache_key, _invalidate_table_cache_keys
|
|
|
|
|
|
class CacheHandler(local):
|
|
@property
|
|
def atomic_caches(self):
|
|
if not hasattr(self, '_atomic_caches'):
|
|
self._atomic_caches = []
|
|
return self._atomic_caches
|
|
|
|
def get_atomic_cache(self, cache_alias, level):
|
|
if cache_alias not in self.atomic_caches[level]:
|
|
self.atomic_caches[level][cache_alias] = AtomicCache(
|
|
self.get_cache(cache_alias, level-1))
|
|
return self.atomic_caches[level][cache_alias]
|
|
|
|
def get_cache(self, cache_alias=None, atomic_level=-1):
|
|
if cache_alias is None:
|
|
cache_alias = cachalot_settings.CACHALOT_CACHE
|
|
|
|
min_level = -len(self.atomic_caches)
|
|
if atomic_level < min_level:
|
|
return caches[cache_alias]
|
|
return self.get_atomic_cache(cache_alias, atomic_level)
|
|
|
|
def enter_atomic(self):
|
|
self.atomic_caches.append({})
|
|
|
|
def exit_atomic(self, commit):
|
|
atomic_caches = self.atomic_caches.pop().values()
|
|
if commit:
|
|
for atomic_cache in atomic_caches:
|
|
atomic_cache.commit()
|
|
|
|
def invalidate_all(self, cache_alias, db_alias):
|
|
tables = connections[db_alias].introspection.table_names()
|
|
table_cache_keys = [_get_table_cache_key(db_alias, t) for t in tables]
|
|
_invalidate_table_cache_keys(cachalot_caches.get_cache(cache_alias),
|
|
table_cache_keys)
|
|
|
|
|
|
cachalot_caches = CacheHandler()
|