2022-12-27 13:56:12 +00:00
|
|
|
import json
|
|
|
|
|
from typing import Dict, Any, Optional
|
|
|
|
|
|
|
|
|
|
from enkanetwork import Cache
|
|
|
|
|
from redis import asyncio as aioredis
|
|
|
|
|
|
|
|
|
|
__all__ = ("RedisCache",)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class RedisCache(Cache):
|
|
|
|
|
def __init__(self, redis: aioredis.Redis, key: Optional[str] = None, ttl: int = 60 * 3) -> None:
|
|
|
|
|
self.redis = redis
|
|
|
|
|
self.ttl = ttl
|
|
|
|
|
self.key = key
|
|
|
|
|
|
|
|
|
|
def get_qname(self, key):
|
2023-02-19 06:20:34 +00:00
|
|
|
return f"{self.key}:{key}" if self.key else f"enka_network:{key}"
|
2022-12-27 13:56:12 +00:00
|
|
|
|
|
|
|
|
async def get(self, key) -> Optional[Dict[str, Any]]:
|
|
|
|
|
qname = self.get_qname(key)
|
|
|
|
|
data = await self.redis.get(qname)
|
|
|
|
|
if data:
|
|
|
|
|
json_data = str(data, encoding="utf-8")
|
|
|
|
|
return json.loads(json_data)
|
|
|
|
|
return None
|
|
|
|
|
|
|
|
|
|
async def set(self, key, value) -> None:
|
|
|
|
|
qname = self.get_qname(key)
|
|
|
|
|
data = json.dumps(value)
|
|
|
|
|
await self.redis.set(qname, data, ex=self.ttl)
|
