【Python3】装饰器 自动更新缓存

自动更新缓存的需求场景

在某些应用中,我们可能需要定期从外部数据源(如 Redis 或者远程接口)拉取数据,并将其缓存在内存中。当有其他代码需要访问这些数据时,可以立刻从内存获取最新数据,而无需每次都进行耗时的外部操作。

关键思路

  1. 初次加载:程序启动时立即执行目标函数,从数据源获取数据并将结果缓存。
  2. 定时更新 :借助定时任务调度器(如 APScheduler),在指定的时间间隔(如30秒)自动再次执行目标函数,刷新缓存中的数据。
  3. 快速访问:对外暴露的函数调用时直接返回缓存中的数据,不会再次执行耗时的外部操作,从而实现快速访问。

使用装饰器实现

我们使用一个自定义的装饰器 @auto_update(update_time=30) 来封装这一逻辑:

  • 装饰器初始化

    当程序加载被装饰的函数时,装饰器会先执行一次目标函数,将返回值存入缓存。

  • 定时任务调度

    使用 APScheduler 的 BackgroundScheduler 来定期调用该函数更新缓存数据。APScheduler 可以独立运行后台线程,不会阻塞主程序的其他逻辑。

  • 缓存访问

    被装饰的函数在对外调用时,不再直接执行原始函数,而是直接返回缓存中的数据。这样,在任意时刻调用该函数,都可以瞬间获取最新数据。

代码

python 复制代码
import time
from functools import wraps
from apscheduler.schedulers.background import BackgroundScheduler

# 创建并启动全局调度器
scheduler = BackgroundScheduler()
scheduler.start()

def auto_update(update_time=30):
    def decorator(func):
        cache = {"value": None, "initialized": False}

        def update_cache():
            new_value = func()
            cache["value"] = new_value
            cache["initialized"] = True
            print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Cache updated.")

        # 启动时先加载一次
        update_cache()

        # 每隔 update_time 秒自动刷新数据
        scheduler.add_job(update_cache, 'interval', seconds=update_time)

        @wraps(func)
        def wrapper(*args, **kwargs):
            return cache["value"]
        
        return wrapper
    return decorator

@auto_update(update_time=30)
def load_data_from_redis():
    # 模拟从Redis获取数据的函数(实际中可替换为耗时的外部操作)
    return f"data_from_redis_{int(time.time())}"

# 当你调用 load_data_from_redis() 时,能立即获得最新的缓存数据
print("Initial data:", load_data_from_redis())
time.sleep(35)
print("Data after 35s:", load_data_from_redis())

总结

通过上述装饰器和 APScheduler 的组合,你就能轻松实现:

  • 程序启动即从外部数据源加载数据到缓存;
  • 周期性、自动地刷新缓存中的数据;
  • 在任意时间调用时都能快速获取最新数据,而无需阻塞或消耗额外时间。

相关推荐
段帅龙呀几秒前
Redis构建缓存服务器
服务器·redis·缓存
夜斗小神社15 小时前
【黑马点评】(二)缓存
缓存
Hello.Reader1 天前
Redis 延迟监控深度指南
数据库·redis·缓存
Hello.Reader1 天前
Redis 延迟排查与优化全攻略
数据库·redis·缓存
在肯德基吃麻辣烫2 天前
《Redis》缓存与分布式锁
redis·分布式·缓存
先睡2 天前
Redis的缓存击穿和缓存雪崩
redis·spring·缓存
CodeWithMe2 天前
【Note】《深入理解Linux内核》 Chapter 15 :深入理解 Linux 页缓存
linux·spring·缓存
大春儿的试验田3 天前
高并发收藏功能设计:Redis异步同步与定时补偿机制详解
java·数据库·redis·学习·缓存
likeGhee3 天前
python缓存装饰器实现方案
开发语言·python·缓存
C182981825753 天前
OOM电商系统订单缓存泄漏,这是泄漏还是溢出
java·spring·缓存