mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-21 21:58:04 +00:00
30 lines
903 B
Python
30 lines
903 B
Python
|
from typing import Optional
|
||
|
|
|
||
|
|
from gram_core.base_service import BaseService
|
||
|
|
from gram_core.dependence.redisdb import RedisDB
|
||
|
|
|
||
|
|
__all__ = ("ChannelAliasCache",)
|
||
|
|
|
||
|
|
|
||
|
|
class ChannelAliasCache(BaseService.Component):
|
||
|
|
def __init__(self, redis: RedisDB):
|
||
|
|
self.client = redis.client
|
||
|
|
self.qname = "channels:alias"
|
||
|
|
self.ttl = 1 * 60 * 60
|
||
|
|
|
||
|
|
def cache_key(self, key: int) -> str:
|
||
|
|
return f"{self.qname}:{key}"
|
||
|
|
|
||
|
|
async def get_data(self, channel_id: int) -> Optional[int]:
|
||
|
|
data = await self.client.get(self.cache_key(channel_id))
|
||
|
|
if data:
|
||
|
|
return int(data.decode())
|
||
|
|
return None
|
||
|
|
|
||
|
|
async def set_data(self, channel_id: int, user_id: int):
|
||
|
|
ck = self.cache_key(channel_id)
|
||
|
|
await self.client.set(ck, user_id, ex=self.ttl)
|
||
|
|
|
||
|
|
async def delete(self, channel_id: int):
|
||
|
|
await self.client.delete(self.cache_key(channel_id))
|