[Python]使用Decorator设计Cache

最新推荐文章于 2024-10-21 15:25:40 发布

长剑耿介

最新推荐文章于 2024-10-21 15:25:40 发布

阅读量1.8k

点赞数

分类专栏： Python2.x 文章标签：缓存 python Decorator

本文链接：https://blog.youkuaiyun.com/jiezhi2013/article/details/51203895

版权

Python2.x 专栏收录该内容

4 篇文章

订阅专栏

本文通过Python的Decorator设计了一个缓存功能，旨在提高代码效率，避免重复计算。通过运行示例代码，展示了装饰器如何存储并返回之前计算过的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接上源码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
Created on 4/20/16

@author: Jiezhi.G@gmail.com

My Blog: jiezhi.github.io

Reference: <Expert Python Programming> Chapter 2--Decorators_Proxy Page.53
"""
import time
import hashlib
import pickle

cache = {}


def is_obsolete(entry, duration):
    return time.time() - entry['time'] > duration


def compute_key(function, args, kw):
    key = pickle.dumps((function.func_name, args, kw))
    return hashlib.sha1(key).hexdigest()


def memoize(duration=10):
    def _memoize(function):
        def __memoize(*args, **kw):
            key = compute_key(function, args, kw)

            # do we have it already?
            if key in cache and not is_obsolete(cache[key], duration):
                print 'we got a winner'
                return cache[key]['value']

            # computing
            result = function(*args, **kw)

            # storing the result
            cache[key] = {'value': result, 'time': time.time()}
            return result

        return __memoize
    return _memoize


@memoize()
def very_very_very_complex_stuff(a, b):
    return a + b


if __name__ == '__main__':
    print very_very_very_complex_stuff(2, 2)
    print cache
    print very_very_very_complex_stuff(2, 2)

运行后可以看到结果：

4
{'dedfca39c250ca2047c5d66a13c5df2e9ac90181': {'value': 4, 'time': 1461155366.249486}}
we got a winner
4