本文記錄嘗試編寫(xiě)一個(gè)簡(jiǎn)單的緩存裝飾器，以學(xué)習(xí)為目的，實(shí)際生產(chǎn)環(huán)境建議大家用標(biāo)準(zhǔn)庫(kù)。

創(chuàng)新互聯(lián)主要從事成都網(wǎng)站設(shè)計(jì)、成都做網(wǎng)站、網(wǎng)頁(yè)設(shè)計(jì)、企業(yè)做網(wǎng)站、公司建網(wǎng)站等業(yè)務(wù)。立足成都服務(wù)臨夏州,十載網(wǎng)站建設(shè)經(jīng)驗(yàn),價(jià)格優(yōu)惠、服務(wù)專(zhuān)業(yè),歡迎來(lái)電咨詢建站服務(wù):18982081108

from hashlib import md5
from pickle import dump, load

# 用pickle進(jìn)行數(shù)據(jù)的讀取、寫(xiě)入
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參數(shù)合并成字符串
            _func_args = ''.join([ str(_) for _ in args][start:]) if have_args else ''

            # kw參數(shù)合并成字符串
            _func_kw   = ''.join([ "%s-%s" % (i,kw[i]) for i in kw if kw]) if have_kw else ''

            # 用于識(shí)別方法名的字符串
            _func_str  = f.__qualname__ + _func_args + _func_kw 

            # 進(jìn)行md5加密
            _md5 = md5()
            _md5.update(_func_str.encode('utf-8'))
            _func_md5 = _md5.hexdigest()

            # 方法執(zhí)行的緩存位置
            file_path = path.join(cache_path, _func_md5)

                                # 判斷方法執(zhí)行結(jié)果是否過(guò)期
            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):
        ....

實(shí)測(cè)結(jié)果

測(cè)試兩個(gè)相同的請(qǐng)求，第一次沒(méi)有緩存，第二次讀取了緩存