python 装饰器实现缓存_尝试自己写一个Python缓存装饰器

本文记录尝试编写一个简单的缓存装饰器,以学习为目的,实际生产环境建议大家用标准库。

from hashlib import md5

from pickle import dump, load

# 用pickle进行数据的读取、写入

def _dkL(f):

with open(f,'rb') as file: return load(file)

def _dkD(o,f):

with open(f, 'wb') as file: return dump(o,file)

def cache(ex_time=10,

start=0,

have_args=True,

have_kw=True,

cache_path="/tmp/pyCache"):

# 判断缓存目录是否存在

if not path.exists(cache_path): makedirs(cache_path)

def _func(f):

def _dec(*args, **kw):

# args参数合并成字符串

_func_args = ''.join([ str(_) for _ in args][start:]) if have_args else ''

# kw参数合并成字符串

_func_kw = ''.join([ "%s-%s" % (i,kw[i]) for i in kw if kw]) if have_kw else ''

# 用于识别方法名的字符串

_func_str = f.__qualname__ + _func_args + _func_kw

# 进行md5加密

_md5 = md5()

_md5.update(_func_str.encode('utf-8'))

_func_md5 = _md5.hexdigest()

# 方法执行的缓存位置

file_path = path.join(cache_path, _func_md5)

# 判断方法执行结果是否过期

if path.exists(file_path):

file_mtime = int(path.getmtime(file_path))

if int(time()) - file_mtime <= int(ex_time):

return _dkL(file_path)

func_result = f(*args, **kw)

_dkD(func_result, file_path)

return func_result

return _dec

return _func

使用方法

@cache(60, have_args=False)

def search_all(self):

....

实测结果

测试两个相同的请求,第一次没有缓存,第二次读取了缓存

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值