尝试自己写一个Python缓存装饰器-创新互联

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

成都创新互联公司-专业网站定制、快速模板网站建设、高性价比沐川网站开发、企业建站全套包干低至880元,成熟完善的模板库,直接使用。一站式沐川网站制作公司更省心,省钱,快速模板网站建设找我们,业务覆盖沐川地区。费用合理售后完善,10多年实体公司更值得信赖。
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):
        ....

实测结果

测试两个相同的请求,第一次没有缓存,第二次读取了缓存
尝试自己写一个Python缓存装饰器

另外有需要云服务器可以了解下创新互联cdcxhl.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。


网站名称:尝试自己写一个Python缓存装饰器-创新互联
网页路径:http://ybzwz.com/article/cochgg.html