函数记忆: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     
 
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值