mirrors/discord.py
1
0
Fork 0
mirror of https://github.com/Rapptz/discord.py.git synced 2025-09-06 09:56:09 +00:00
Code Issues Projects Releases Wiki Activity

Add enable_debug_events parameter to enable expensive debug events

Browse Source 
This should allow less dispatching during heavy gateway event streams.
This commit is contained in:
Rapptz
2021-07-31 20:35:16 -04:00
parent 2de0398d66
commit b640493300
3 changed files with 30 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
discord/gateway.py
View File

@ -293,6 +293,12 @@ class DiscordWebSocket:
def is_ratelimited(self):
return self._rate_limiter.is_ratelimited()
def log_receive(self, data, /):
self._dispatch('socket_raw_receive', data)
def empty_log_receive(self, _, /):
pass
@classmethod
async def from_client(cls, client, *, initial=False, gateway=None, shard_id=None, session=None, sequence=None, resume=False):
"""Creates a main websocket for Discord from a :class:`Client`.
@ -318,6 +324,10 @@ class DiscordWebSocket:
ws.sequence = sequence
ws._max_heartbeat_timeout = client._connection.heartbeat_timeout
if client._enable_debug_events:
ws.send = ws.debug_send
ws.log_receive = ws.empty_log_receive
client._connection._update_references(ws)
log.debug('Created websocket connected to %s', gateway)
@ -409,8 +419,8 @@ class DiscordWebSocket:
await self.send_as_json(payload)
log.info('Shard ID %s has sent the RESUME payload.', self.shard_id)
async def received_message(self, msg):
self._dispatch('socket_raw_receive', msg)
async def received_message(self, msg, /):
self.log_receive(msg)
if type(msg) is bytes:
self._buffer.extend(msg)
@ -574,11 +584,15 @@ class DiscordWebSocket:
log.info('Websocket closed with %s, cannot reconnect.', code)
raise ConnectionClosed(self.socket, shard_id=self.shard_id, code=code) from None
async def send(self, data):
async def debug_send(self, data, /):
await self._rate_limiter.block()
self._dispatch('socket_raw_send', data)
await self.socket.send_str(data)
async def send(self, data, /):
await self._rate_limiter.block()
await self.socket.send_str(data)
async def send_as_json(self, data):
try:
await self.send(utils.to_json(data))