有没有办法使用生存时间装饰器来缓存Python 3.5定义?

时间:2016-11-18 16:21:19

标签: python python-3.x caching

目前,我使用functools' lru_cache来处理我对函数的缓存。问题是缓存的大小永远不会增长到足以使用LRU(因为函数从不接受参数)。相反,该函数在被调用时会打开一个特定的URL并返回其内容。

有没有一种方法可以指定生活时间'缓存,在经过一定时间/一定数量的调用后,它会刷新缓存吗?

2 个答案:

答案 0 :(得分:2)

我不知道装饰器,但您可以跟踪上次获取页面并根据需要进行更新。如果这是一个单线程应用程序,它可以很简单

_cached_page = ''
_cached_page_time = 0

def get_page():
    global _cached_page, _cached_page_time
    now = time.time()
    # invalidate in 1 hour
    if not _cached_page or now - _cached_page_time > 60 * 60:
        _cached_page = get_the_page_here()
        _cached_page_time = time.time()
    return _cached_page

您还可以在后台使用计时器对页面进行老化。您需要使用锁来控制访问,但这也使缓存在多线程程序中可用。

_cached_page = ''
_cached_page_lock = threading.Lock()

def _invalidate_page():
    global _cached_page
    with _cached_page_lock:
        _cached_page = ''

def get_page():
    global _cached_page
    with _cached_page_lock:
        if not _cached_page:
            _cached_page = get_the_page_here()
            # invalidate in 1 hour
            threading.Timer(60*60, _invalidate_page)
        return _cached_page

最后,服务器可能在http标头中包含Expires: ...字段。根据服务的编写程度,这可以很好地反映页面的缓存时间。

答案 1 :(得分:1)

functools.lru_cache函数接受maxsize参数,该结果会将结果保存到maxsize最近的调用。

您可以通过调用已修饰函数的cache_info属性来检查此内容。

如果要完全刷新缓存,则应通过计算缓存调用次数并在达到最大大小时重置缓存来手动实现现金对象。

from functools import wraps


class Mycache(object):
    def __init__(self, maxcount):
        self.count = 0
        self.maxcount = maxcount
        self.cache = {}

    def __call__(self, func):

        @wraps(func)
        def wrapped(*args):
            self.count += 1
            if self.count > self.maxcount:
                self.cache = {}
                self.count = 0
                result = self.cache[args] = func(*args)
            else:
                try:
                    result = self.cache[args]
                except KeyError:
                    result = self.cache[args] = func(*args)
            return result
        return wrapped

演示:

@Mycache(3)
def a(arg):
    print("arg is : {}".format(arg))
    return arg ** 2

print(a(4))
print(a(4))
print(a(4))
print(a(4))
print(a(3))
print(a(4))

输出:

arg is : 4
16
16
16
arg is : 4
16
arg is : 3
9
16