fix: fix ConnectionResetError

2022-06-07 10:49:22 +08:00 · 2022-06-07 10:49:22 +08:00 · a9d8de964d
commit a9d8de964d
parent 0f666eaafe
1 changed files with 3 additions and 1 deletions
--- a/src/blrec/bili/danmaku_client.py
+++ b/src/blrec/bili/danmaku_client.py
@ -87,7 +87,9 @@ class DanmakuClient(EventEmitter[DanmakuListener], AsyncStoppableMixin):

    @retry(
        wait=wait_exponential(multiplier=0.1, max=10),
-        retry=retry_if_exception_type((asyncio.TimeoutError, aiohttp.ClientError)),
+        retry=retry_if_exception_type(
+            (asyncio.TimeoutError, aiohttp.ClientError, ConnectionError)
+        ),
    )
    async def _connect(self) -> None:
        logger.debug('Connecting to server...')