- Fix redis cache.

- Encode key builder.
This commit is contained in:
long2ice
2021-07-26 16:33:22 +08:00
parent de1bde39fd
commit 767241be41
6 changed files with 16 additions and 10 deletions

View File

@@ -10,10 +10,8 @@ class RedisBackend(Backend):
self.redis = redis
async def get_with_ttl(self, key: str) -> Tuple[int, str]:
p = self.redis.pipeline()
p.ttl(key)
p.get(key)
return await p.execute()
async with self.redis.pipeline(transaction=True) as pipe:
return await (pipe.ttl(key).get(key).execute())
async def get(self, key) -> str:
return await self.redis.get(key)
@@ -24,6 +22,6 @@ class RedisBackend(Backend):
async def clear(self, namespace: str = None, key: str = None) -> int:
if namespace:
lua = f"for i, name in ipairs(redis.call('KEYS', '{namespace}:*')) do redis.call('DEL', name); end"
return await self.redis.eval(lua)
return await self.redis.eval(lua, numkeys=0)
elif key:
return await self.redis.delete(key)

View File

@@ -3,9 +3,9 @@ import json
import pickle # nosec:B403
from decimal import Decimal
from typing import Any
from fastapi.encoders import jsonable_encoder
import dateutil.parser
from fastapi.encoders import jsonable_encoder
CONVERTERS = {
"date": dateutil.parser.parse,

View File

@@ -19,7 +19,7 @@ def default_key_builder(
cache_key = (
prefix
+ hashlib.md5( # nosec:B303
f"{func.__module__}:{func.__name__}:{args}:{kwargs}"
f"{func.__module__}:{func.__name__}:{args}:{kwargs}".encode()
).hexdigest()
)
return cache_key