27 lines
699 B
Python
27 lines
699 B
Python
import logging
|
|
|
|
from redis.asyncio import Redis
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class RedisClient:
|
|
def __init__(self, r: Redis):
|
|
self.r = r
|
|
|
|
async def set_search_query(self, query_hash: str, q: str, ttl: int = 3600) -> bool:
|
|
try:
|
|
await self.r.set(f"search:{query_hash}", q, ex=ttl)
|
|
return True
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
return False
|
|
|
|
async def get_search_query(self, query_hash: str) -> str | None:
|
|
try:
|
|
res = await self.r.get(f"search:{query_hash}")
|
|
return res.decode()
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
return None
|