mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-11-23 00:01:45 +00:00
37 lines
1.1 KiB
Python
37 lines
1.1 KiB
Python
|
import asyncio
|
||
|
|
|
||
|
|
from logger import Log
|
||
|
|
from redis import asyncio as aioredis
|
||
|
|
|
||
|
|
|
||
|
|
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配置 [host]: {port}')
|
||
|
|
Log.debug(f'获取Redis配置 [host]: {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 Exception as exc:
|
||
|
|
Log.error("尝试连接Redis失败 \n")
|
||
|
|
raise exc
|
||
|
|
|
||
|
|
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()
|