函数记忆:python装饰器实现短时缓存方法执行结果

# 【函数记忆】方法结果缓存装饰器
cache = {}


def is_obsolete(entry, duration):
    d = time.time() - entry['time']
    return d > duration


def compute_key(function, args, kwargs):
    key = pickle.dumps((function.__name__, args, kwargs))
    return hashlib.sha1(key).hexdigest()


def memoize(duration=10):
    def _memorize(function):
        @wraps(function)  # update_wrapper维持被修饰函数属性不变
        def __memorize(*args, **kwargs):
            key = compute_key(function, args, kwargs)  # 方法名和传入的参数不同key会不同

            if key in cache:
                if not is_obsolete(cache[key], duration):
                    print('%s 使用缓存' % function.__name__)
                    return cache[key]['value']
                else:
                    print('%s 删除缓存' % function.__name__)
                    del cache[key]  # 删除过时缓存

            result = function(*args, **kwargs)
            cache[key] = {'value': result, 'time': time.time()}
            return result

        return __memorize

    return _memorize

使用方法:

@memoize(10)    
def get_data_list(num, limit, skip):
    # data将会被缓存10秒,十秒内调用get_data_list方法将会使用缓存,10秒后将重新执行方法并再缓存10秒,以此类推...
  data = num + limit + skip
  return data     

 

 

 

posted @ 2021-03-24 15:28  橘个栗子  阅读(57)  评论(0编辑  收藏  举报