68 lines
2.1 KiB
Python
68 lines
2.1 KiB
Python
#This file is part of Tryton. The COPYRIGHT file at the top level of
|
|
#this repository contains the full copyright notices and license terms.
|
|
from threading import Lock
|
|
import time
|
|
from trytond.transaction import Transaction
|
|
from trytond.cache import Cache
|
|
|
|
|
|
class Cacheable(object):
|
|
_name = None
|
|
|
|
def __init__(self):
|
|
super(Cacheable, self).__init__()
|
|
self._cache = {}
|
|
self._max_len = 1024
|
|
self._timeout = 3600
|
|
self._lock = Lock()
|
|
self._timestamp = None
|
|
Cache._cache_instance.append(self)
|
|
|
|
def add(self, key, value):
|
|
dbname = Transaction().cursor.dbname
|
|
self._lock.acquire()
|
|
try:
|
|
self._cache.setdefault(dbname, {})
|
|
|
|
lower = None
|
|
if len(self._cache[dbname]) > self._max_len:
|
|
mintime = time.time() - self._timeout
|
|
for key2 in self._cache[dbname].keys():
|
|
last_time = self._cache[dbname][key2][1]
|
|
if mintime > last_time:
|
|
del self._cache[dbname][key2]
|
|
else:
|
|
if not lower or lower[1] > last_time:
|
|
lower = (key2, last_time)
|
|
if len(self._cache[dbname]) > self._max_len and lower:
|
|
del self._cache[dbname][lower[0]]
|
|
|
|
self._cache[dbname][key] = (value, time.time())
|
|
finally:
|
|
self._lock.release()
|
|
|
|
def invalidate(self, key):
|
|
dbname = Transaction().cursor.dbname
|
|
self._lock.acquire()
|
|
try:
|
|
del self._cache[dbname][key]
|
|
finally:
|
|
self._lock.release()
|
|
|
|
def get(self, key):
|
|
dbname = Transaction().cursor.dbname
|
|
try:
|
|
return self._cache[dbname][key][0]
|
|
except KeyError:
|
|
return None
|
|
|
|
def clear(self):
|
|
dbname = Transaction().cursor.dbname
|
|
self._lock.acquire()
|
|
try:
|
|
self._cache.setdefault(dbname, {})
|
|
self._cache[dbname].clear()
|
|
Cache.reset(dbname, self._name)
|
|
finally:
|
|
self._lock.release()
|