mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 20:10:36 +00:00
8f424bf0d4
♻️ 重构插件系统 ⚙️ 重写插件 🎨 改进代码结构 📝 完善文档 Co-authored-by: zhxy-CN <admin@owo.cab> Co-authored-by: 洛水居室 <luoshuijs@outlook.com> Co-authored-by: xtaodada <xtao@xtaolink.cn> Co-authored-by: Li Chuangbo <im@chuangbo.li>
45 lines
1.4 KiB
Python
45 lines
1.4 KiB
Python
import asyncio
|
|
|
|
import fakeredis.aioredis
|
|
from redis import asyncio as aioredis
|
|
from typing_extensions import Self
|
|
|
|
from core.config import BotConfig
|
|
from core.service import Service
|
|
from utils.log import logger
|
|
|
|
|
|
class RedisDB(Service):
|
|
@classmethod
|
|
def from_config(cls, config: BotConfig) -> Self:
|
|
return cls(**config.redis.dict())
|
|
|
|
def __init__(self, host="127.0.0.1", port=6379, database=0, loop=None):
|
|
self.client = aioredis.Redis(host=host, port=port, db=database)
|
|
self.ttl = 600
|
|
self.key_prefix = "paimon_bot"
|
|
self._loop = loop
|
|
|
|
async def ping(self):
|
|
if await self.client.ping():
|
|
logger.info("连接 [red]Redis[/] 成功")
|
|
else:
|
|
logger.info("连接 [red]Redis[/] 失败")
|
|
raise RuntimeError("连接 [red]Redis[/] 失败")
|
|
|
|
async def start(self): # pylint: disable=W0221
|
|
if self._loop is None:
|
|
self._loop = asyncio.get_running_loop()
|
|
logger.info("正在尝试建立与 [red]Redis[/] 连接")
|
|
try:
|
|
await self.ping()
|
|
except (KeyboardInterrupt, SystemExit):
|
|
pass
|
|
except BaseException as exc:
|
|
logger.warning("尝试连接 [red]Redis[/] 失败，使用 [red]fakeredis[/] 模拟", exc)
|
|
self.client = fakeredis.aioredis.FakeRedis()
|
|
await self.ping()
|
|
|
|
async def stop(self): # pylint: disable=W0221
|
|
await self.client.close()
|