flask-cache demo, get_dict的实现(测试代码)

代码如下:

# -*- coding: utf-8 -*-
import string
import random
from flask import Flask
from flask.ext.cache import Cache

SECRET_KEY = '\xfb\x12\xdf\xa1@i\xd6>V\xc0\xbb\x8fp\x16#Z\x0b\x81\xeb\x16'
DEBUG = True
CACHE_TYPE = 'memcached'
CACHE_DEFAULT_TIMEOUT = 3600  # 默认缓存1个小时
CACHE_KEY_PREFIX = 'test_'  # 所有key之前添加前缀
CACHE_MEMCACHED_SERVERS = ['127.0.0.1:11211']


app = Flask(__name__)
app.config.from_object(__name__)
cache = Cache(app)

strings = string.ascii_lowercase


#: This is an example of a memoized function
@cache.memoize(60)
def hello(a):
    print 'not from cache {0}'.format(a)
    return strings[a] + random.choice(strings)


def get_dict_by_ids(func, ids):
    """
    :param func: 有cache.memoize装饰的方法
    :params ids: id列表
    """
    if not ids:
        return

    # cache_key列表
    cache_key_list = []
    for item_id in ids:
        cache_key = func.make_cache_key(func.uncached, item_id)
        cache_key_list.append(cache_key)

    # 批量获取缓存
    items = cache.get_many(*cache_key_list)

    uncached_dict = {}
    # 单独获取缓存中没有的数据, 应该使用set_many设置缓存
    for (index, item) in enumerate(items):
        if item is None:
            item_id = ids[index]
            cache_key = cache_key_list[index]

            value = hello.uncached(item_id)
            items[index] = value
            uncached_dict[cache_key] = value
    if uncached_dict:
        cache.set_many(uncached_dict, timeout=func.cache_timeout)

    return dict(zip(ids, items))


@app.route('/many')
def delete_cache():
    cache.delete_memoized(hello)
    print hello(1)
    print hello(3)
    print get_dict_by_ids(hello, [1, 2, 3])
    return 'OK'

if __name__ == '__main__':
    app.run(host='0.0.0.0')