github-repos/blivedm
1
0
Fork 0
mirror of https://github.com/xfgryujk/blivedm.git synced 2024-12-26 12:50:13 +08:00
Code Issues Projects Releases Wiki Activity

发送心跳包添加错误处理

Browse Source
This commit is contained in:
John Smith 2021-12-18 01:18:05 +08:00
parent afa04d557b
commit 60c60e69c5
1 changed files with 19 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
blivedm/client.py
View File

@ -410,9 +410,26 @@ class BLiveClient:
""" """
定时发送心跳包的回调 定时发送心跳包的回调
""" """
coro = self._websocket.send_bytes(self._make_packet({}, Operation.HEARTBEAT)) if self._websocket is None or self._websocket.closed:
asyncio.ensure_future(coro, loop=self._loop) self._heartbeat_timer_handle = None
return
self._heartbeat_timer_handle = self._loop.call_later(self._heartbeat_interval, self._on_send_heartbeat) self._heartbeat_timer_handle = self._loop.call_later(self._heartbeat_interval, self._on_send_heartbeat)
asyncio.ensure_future(self._send_heartbeat(), loop=self._loop)
async def _send_heartbeat(self):
"""
发送心跳包
"""
if self._websocket is None or self._websocket.closed:
return
try:
await self._websocket.send_bytes(self._make_packet({}, Operation.HEARTBEAT))
except ConnectionResetError as e:
logger.warning('room=%d _send_heartbeat() failed: %r', self.room_id, e)
except Exception: # noqa
logger.exception('room=%d _send_heartbeat() failed:', self.room_id)
async def _on_ws_message(self, message: aiohttp.WSMessage): async def _on_ws_message(self, message: aiohttp.WSMessage):
""" """