Spaces:
Runtime error
Runtime error
File size: 2,560 Bytes
e887d1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
import os
from pickle import dumps, loads
REDIS_HOST = os.getenv('REDIS_HOST')
REDIS_PASS = os.getenv('REDIS_PASS')
from redis import StrictRedis
redis = StrictRedis(host=REDIS_HOST, port=int(os.getenv("REDIS_PORT", "6379")),
max_connections=256, password=REDIS_PASS)
class PythonNativeRedisClient(object):
"""A simple redis client for storing and retrieving native python datatypes."""
def __init__(self, redis_host=redis):
"""Initialize client."""
self.client = redis_host
def set(self, key, value, **kwargs):
"""Store a value in Redis."""
return self.client.set(key, dumps(value), **kwargs)
def get(self, key, default=None):
"""Retrieve a value from Redis."""
val = self.client.get(key)
if val:
return loads(val)
return default
def expire(self, key, expire_secs):
self.client.expire(key, expire_secs)
def ttl(self, key):
self.client.ttl(key)
def delete(self, key):
redis.delete(key)
redis_client = PythonNativeRedisClient(redis)
# Reids 缓存，结果保存到 redis 中, 根据函数的第一个参数缓存结果, 可以指定缓存有效期
class RedisCache:
def __init__(self, expire=60, cache_prefix="redis_cache_"):
self.expire = expire
self.cache_prefix = cache_prefix
def format_key(self, input_key):
return self.cache_prefix + "_" + input_key
def get(self, input_key):
return redis_client.get(self.format_key(input_key))
def __call__(self, func):
def wrapper(*args, **kwargs):
# 使用函数的第一个参数作为 limit_key
key = args[1]
if key:
try:
cached_value = self.get(key)
if cached_value:
print("Cached value from redis for method:%s with param:%s" % (func.__name__, str(key)) )
return cached_value
except Exception as ex:
print("Error get data from cache, key:%s error message:%s" % (key, repr(ex)))
result = func(*args, **kwargs)
try:
redis_key = self.format_key(key)
redis_client.set(redis_key, result)
redis_client.expire(redis_key, self.expire)
except Exception as ex:
print("Error put data to cache, key:%s error message:%s" % (key, repr(ex)) )
return result
return wrapper # to be used in method within a class |