尝试自己写一个Python缓存装饰器
admin
2023-07-08 04:03:01
0

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

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缓存装饰器

相关内容

热门资讯

我国科学家为细胞信号“导航”开... 新华社济南5月31日电(记者张力元)人体细胞犹如一座精密的通信城市,每天都有大量“指令”穿梭传递,调...
极端大风突袭哈尔滨!过山车停摆... 极目新闻记者 詹钘5月31日,受强对流天气影响,哈尔滨国际会展中心体育场相关设施受到损坏,原计划当晚...
三原电缆取得电缆接头连接用防护... 国家知识产权局信息显示,上海三原电缆附件有限公司取得一项名为“一种电缆接头连接用防护结构”的专利,授...
原创 识... 还是那句话,机圈苦大屏久已…… 虽然大屏有大屏的美,但是小屏也有小屏的俏。在大屏旗舰占据主流的手机市...
玄戒技术取得分频电路专利,实现... 国家知识产权局信息显示,北京玄戒技术有限公司取得一项名为“分频电路、分频器、射频芯片和电子设备”的专...
为什么今年香会基调明显变了 5月29日—31日在新加坡举行的第23届香格里拉对话会(简称“香会”),见证着元首引领下大国关系继续...
成本几毛钱、假驱蚊液香精兑水,... 入夏升温,蚊虫进入活跃期,驱蚊防护成为民生刚需,《财经调查》持续接到消费者投诉,他们买到的多款网红驱...
越来越多80后90后,正在丧失... 六一儿童节到来之际,朋友圈里开始出现一种熟悉的热闹。有人晒出零食礼包,有人半开玩笑地向伴侣讨礼物,还...
洋保电子取得用于低温环境的电气... 国家知识产权局信息显示,洋保电子(太仓)有限公司取得一项名为“一种用于低温环境的电气柜”的专利,授权...
中日韩飞手争霸宁波!2026无... 潮新闻客户端 记者 陈冲 通讯员 朱凝 5月31日,2026小遛·无人机竞速世界杯(中国·宁波鄞州站...