mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-05-10 15:59:50 +00:00
Code Issues Projects Releases Wiki Activity

Add traceback to debug blocking issues

Browse Source
This commit is contained in:
Rapptz 2020-04-14 04:21:20 -04:00
parent 0cef8c1f60
commit d6be6adf8b
1 changed files with 12 additions and 1 deletions

13
discord/gateway.py

@ -33,6 +33,7 @@ import struct
import sys
import time
import threading
import traceback
import zlib
import websockets
@ -66,6 +67,7 @@ class KeepAliveHandler(threading.Thread):
shard_id = kwargs.pop('shard_id', None)
threading.Thread.__init__(self, *args, **kwargs)
self.ws = ws
self._main_thread_id = ws.thread_id
self.interval = interval
self.daemon = True
self.shard_id = shard_id
@ -106,7 +108,14 @@ class KeepAliveHandler(threading.Thread):
break
except concurrent.futures.TimeoutError:
total += 5
log.warning(self.block_msg, total)
try:
frame = sys._current_frames()[self._main_thread_id]
except KeyError:
msg = self.block_msg
else:
stack = traceback.format_stack(frame)
msg = '%s\nLoop thread stacktrace:\n%s' % (self.block_msg, ''.join(stack))
log.warning(msg, total)
except Exception:
self.stop()
@ -215,6 +224,7 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
self._dispatch_listeners = []
# the keep alive
self._keep_alive = None
self.thread_id = threading.get_ident()
# ws related stuff
self.session_id = None
@ -648,6 +658,7 @@ class DiscordVoiceWebSocket(websockets.client.WebSocketClientProtocol):
ws.gateway = gateway
ws._connection = client
ws._max_heartbeat_timeout = 60.0
ws.thread_id = threading.get_ident()
if resume:
await ws.resume()