“Python实用秘技09”更好用的函数运算缓存

本文完整示例代码及文件已上传至我的Github仓库//github.com/CNFeffery/PythonPracticalSkills

  这是我的系列文章“Python实用秘技”的第9期,本系列立足于笔者日常工作中使用Python积累的心得体会,每一期为大家带来一个几分钟内就可学会的简单小技巧。

  作为系列第9期,我们即将学习的是:在Python中实现简单好用的函数运算缓存

  函数运算缓存,顾名思义就是我们可以针对指定的函数,让其记住过往参数输入和返回结果,使得后续接收到相同的参数时跳过函数运算,直接返回已缓存的结果值。

  很多朋友应该知道Python标准库里functools.lru_cache可以做函数运算缓存,但是它的功能实在是太简陋了,像过期时间设置之类的功能都没有。

  而我们可以使用第三方库cachier来代替,它的基本使用方式非常简单,使用pip install cachier完成安装后,我们来看一个简单的示例:

  这里我们定义一个具有一定运算耗时的函数,利用cachier.cachier()装饰,并利用参数stale_after设置缓存到期时间为10秒:

import time
from cachier import cachier
from datetime import timedelta

@cachier(stale_after=timedelta(seconds=10))
def demo(x: int, y: int):
    
    time.sleep(2)
    
    return x * y

for i in range(10):
    print('-'*50)
    print(f'第{i+1}次执行')
    start = time.time()
    demo(1, 1)
    print(f'耗时{round(time.time() - start, 2)}秒')

    time.sleep(2)

  过程打印记录如下:

  可以观察到,我们的函数按照设定好的过期时间进行着缓存。除此之外,cachier还有很多其他实用特性,譬如基于mongodb实现分布式存储等,你可以前往其官方仓库查看详细文档说明://github.com/shaypal5/cachier


  本期分享结束,咱们下回见~👋

Tags: