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

Implement zlib streaming for the gateway.

Browse Source
This commit is contained in:
Rapptz
2017-10-12 22:53:20 -04:00
parent 9b4a2dc7cb
commit 462191a08b
2 changed files with 26 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
discord/gateway.py
View File

@ -186,6 +186,8 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
# ws related stuff
self.session_id = None
self.sequence = None
self._zlib = zlib.decompressobj()
self._buffer = bytearray()
@classmethod
@asyncio.coroutine
@ -312,8 +314,18 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol):
self._dispatch('socket_raw_receive', msg)
if isinstance(msg, bytes):
msg = zlib.decompress(msg, 15, 10490000) # This is 10 MiB
msg = msg.decode('utf-8')
self._buffer.extend(msg)
if len(msg) >= 4:
suffix = int.from_bytes(msg[-4:], byteorder='big')
if suffix == 0xFFFF:
msg = self._zlib.decompress(self._buffer)
msg = msg.decode('utf-8')
self._buffer = bytearray()
else:
return
else:
return
msg = json.loads(msg)