mirror of
https://github.com/long2ice/fastapi-cache.git
synced 2026-03-25 13:07:53 +00:00
28 lines
921 B
Python
28 lines
921 B
Python
from typing import Tuple
|
|
|
|
from aioredis import Redis
|
|
|
|
from fastapi_cache.backends import Backend
|
|
|
|
|
|
class RedisBackend(Backend):
|
|
def __init__(self, redis: Redis):
|
|
self.redis = redis
|
|
|
|
async def get_with_ttl(self, key: str) -> Tuple[int, str]:
|
|
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)
|
|
|
|
async def set(self, key: str, value: str, expire: int = None):
|
|
return await self.redis.set(key, value, ex=expire)
|
|
|
|
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, numkeys=0)
|
|
elif key:
|
|
return await self.redis.delete(key)
|