python的cache修饰器

简单的memory cache。可以用来内存缓存任意函数方法。

#!/usr/bin/python

import functools
from threading import RLock
import logging

LOGGER = logging.getLogger(__name__)

class CacheNullValue(object):
    pass

_Null = CacheNullValue()

class Cache(object):

    def __init__(self, cache_limit=1000):
        self._cache = {}
        self._queue = []
        self.cache_limit = cache_limit
        self._lock = RLock()

    def __getitem__(self, name):
        with self._lock:
            return self._cache.get(name, _Null)
    get = __getitem__

    def __setitem__(self, name, value):
        with self._lock:
            if len(self._queue) >= self.cache_limit:
                del self._cache[self._queue.pop(0)]
            self._queue.append(name)
            self._cache[name] = value
    set = __setitem__

def cache(cache_limit):
    """A function decorator for method cache store
    """

    def cached(f):
        @functools.wraps(f)
        def _cached(*args, **kwargs):
            cid = repr((args, kwargs))
            data = _cached.cache.get(cid)
            if data != _Null:
                LOGGER.debug('hit cached `' + _cached.__name__ + '` for ' + cid)
                return data
            data = _cached.f(*args, **kwargs)
            _cached.cache.set(cid, data)
            return data

        _cached.f = f
        _cached.cache = Cache(cache_limit)

        return _cached

    return cached

def setup(cache_limit):
    """ A global thread safe key-value store
    """
    global _cache_object
    _cache_object = Cache(cache_limit)

def set(cid, data):

    _cache_object.set(cid, data)

def get(cid):
    return _cache_object.get(cid)

if __name__ == '__main__':

    logging.basicConfig(level=logging.DEBUG,
                        format='%(asctime)s %(levelname)-8s %(message)s',
                        datefmt='%Y-%m-%d %H:%M:%S', filemode='a+')

    class T(object):

        @cache(100)
        def c(self, t):
            return t

    @cache(100)
    def cc(t):
        return t

    t = T()
    print t.c('hello')
    print t.c('hello')
    print t.c('h')
    print t.c('hello')
    print cc('he')
    print cc('he')