mirror of
https://github.com/PaiGramTeam/PamGram.git
synced 2024-11-16 12:02:16 +00:00
40 lines
1.3 KiB
Python
40 lines
1.3 KiB
Python
import asyncio
|
||
|
||
import fakeredis.aioredis
|
||
from redis import asyncio as aioredis
|
||
|
||
from logger import Log
|
||
|
||
|
||
class RedisDB:
|
||
|
||
def __init__(self, host="127.0.0.1", port=6379, db=0, loop=None):
|
||
Log.debug(f'获取Redis配置 [host]: {host}')
|
||
Log.debug(f'获取Redis配置 [port]: {port}')
|
||
Log.debug(f'获取Redis配置 [db]: {db}')
|
||
self.client = aioredis.Redis(host=host, port=port, db=db)
|
||
self.ttl = 600
|
||
self.key_prefix = "paimon_bot"
|
||
self._loop = loop
|
||
if self._loop is None:
|
||
self._loop = asyncio.get_event_loop()
|
||
try:
|
||
Log.info("正在尝试建立与Redis连接")
|
||
self._loop.run_until_complete(self.ping())
|
||
except (KeyboardInterrupt, SystemExit):
|
||
pass
|
||
except BaseException as exc:
|
||
Log.warning("尝试连接Redis失败，使用 fakeredis 模拟", exc)
|
||
self.client = fakeredis.aioredis.FakeRedis()
|
||
self._loop.run_until_complete(self.ping())
|
||
|
||
async def ping(self):
|
||
if await self.client.ping():
|
||
Log.info("连接Redis成功")
|
||
else:
|
||
Log.info("连接Redis失败")
|
||
raise RuntimeError("连接Redis失败")
|
||
|
||
async def close(self):
|
||
await self.client.close()
|