Python中如何实现多级缓存

import cachetools

import redis

from functools import wraps

# 初始化Redis客户端

r = redis.Redis(host='localhost', port=6379, db=0)

# 使用LRU策略的内存缓存

memory_cache = cachetools.LRUCache(maxsize=1024)

def multi_level_cache(func):

    @wraps(func)

    def wrapper(key):

        # 尝试从内存缓存中获取数据

        if key in memory_cache:

            print("Hit in-memory cache")

            return memory_cache[key]

        # 如果不在内存，则尝试从Redis缓存中获取

        value = r.get(key)

        if value is not None:

            print("Hit Redis cache")

            # 更新内存缓存

            memory_cache[key] = value

            return value

        # 当两个缓存都未命中时，调用原始函数计算结果

        result = func(key)

        # 存储结果到两个层级的缓存中

        r.set(key, result)

        memory_cache[key] = result

        return result

    return wrapper

@multi_level_cache

def fetch_music_data(music_id):

    """模拟从数据库获取音乐信息的过程"""

    # 这里仅作演示，实际上应该是查询数据库的操作

    return f"Music data for {music_id}"

# 测试多级缓存功能

print(fetch_music_data('123'))

print(fetch_music_data('123'))  # 第二次调用应直接命中缓存