Выделение БД и кэша как отдельные модули
This commit is contained in:
42
App/cache/__init__.py
vendored
Normal file
42
App/cache/__init__.py
vendored
Normal file
@@ -0,0 +1,42 @@
|
||||
import json
|
||||
from redis import Redis, ConnectionError, DataError
|
||||
|
||||
class RedisCache:
|
||||
def __init__(self, config: dict):
|
||||
self.config = config
|
||||
self.conn = self._connect()
|
||||
|
||||
def _connect(self):
|
||||
try:
|
||||
conn = Redis(**self.config)
|
||||
return conn
|
||||
except DataError as err:
|
||||
print(err)
|
||||
return None
|
||||
|
||||
def set_value(self, name: str, value_dict: dict, ttl: int):
|
||||
try:
|
||||
value_js = json.dumps(value_dict)
|
||||
self.conn.set(name=name, value=value_js)
|
||||
if ttl > 0:
|
||||
self.conn.expire(name, ttl)
|
||||
return True
|
||||
except ConnectionError as err:
|
||||
print(err)
|
||||
return False
|
||||
|
||||
def get_value(self, name: str):
|
||||
try:
|
||||
value_js = self.conn.get(name)
|
||||
except:
|
||||
return None
|
||||
value_js = self.conn.get(name)
|
||||
if value_js:
|
||||
value_dict = json.loads(value_js)
|
||||
return value_dict
|
||||
else:
|
||||
return None
|
||||
|
||||
def __exit__(self, exc_type, exc_val, exc_tb):
|
||||
self.conn.close()
|
||||
return True
|
||||
26
App/cache/wrapper.py
vendored
Normal file
26
App/cache/wrapper.py
vendored
Normal file
@@ -0,0 +1,26 @@
|
||||
from functools import wraps
|
||||
from . import RedisCache
|
||||
|
||||
def fetch_from_cache(cache_name: str, cache_config: dict):
|
||||
cache_conn = RedisCache(cache_config['redis'])
|
||||
ttl = cache_config['ttl']
|
||||
def decorator(f):
|
||||
# как оно работает
|
||||
# лезем в кэш и смотрим, есть ли там что-то
|
||||
# если есть, то возвращаем кэшированную информацию
|
||||
# если нет, то запускаем декорируемую функцию
|
||||
# достаём оттуда информацию
|
||||
# заносим её в кэш
|
||||
# возвращаем кэшированную информацию
|
||||
@wraps(f)
|
||||
def wrapper(*args, **kwargs):
|
||||
cached_value = cache_conn.get_value(cache_name)
|
||||
print("cached_value=", cached_value)
|
||||
if cached_value:
|
||||
return cached_value
|
||||
response = f(*args, **kwargs)
|
||||
print("response=", response)
|
||||
cache_conn.set_value(cache_name,response,ttl)
|
||||
return response
|
||||
return wrapper
|
||||
return decorator
|
||||
Reference in New Issue
Block a user