ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-07-17 03:20:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

⚗️ add option for ws reconnect

Browse Source
This commit is contained in:
yanyongyu
2021-11-30 12:24:09 +08:00
parent a6112fdf64
commit b92c1a0b33
3 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
nonebot/drivers/fastapi.py
View File

@ -555,7 +555,7 @@ class FullDriver(ForwardDriver, Driver):
asyncio.create_task(bot.handle_message(msg))
except ConnectionClosed:
logger.opt(colors=True).error(
"<r><bg #f8bbd0>WebSocket connection closed by peer. "
"<r><bg #f8bbd0>WebSocket connection closed. "
"Try to reconnect...</bg #f8bbd0></r>"
)
break
@ -568,6 +568,12 @@ class FullDriver(ForwardDriver, Driver):
if bot:
self._bot_disconnect(bot)
bot = None
if not setup_.reconnect:
logger.info(
f"WebSocket reconnect disabled for bot {setup_.self_id}"
)
break
await asyncio.sleep(setup_.reconnect_interval)
except asyncio.CancelledError: