From 5ae9ea26b176b4368b7887b70e3e0416405b4ace Mon Sep 17 00:00:00 2001 From: Rapptz Date: Sun, 10 Jun 2018 18:37:51 -0400 Subject: [PATCH] Disable compression for websockets. Increase of RAM and CPU doesn't give me much benefit I feel. --- discord/gateway.py | 4 ++-- discord/shard.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/discord/gateway.py b/discord/gateway.py index 0b54c1d24..db5809b60 100644 --- a/discord/gateway.py +++ b/discord/gateway.py @@ -196,7 +196,7 @@ class DiscordWebSocket(websockets.client.WebSocketClientProtocol): This is for internal use only. """ gateway = await client.http.get_gateway() - ws = await websockets.connect(gateway, loop=client.loop, klass=cls) + ws = await websockets.connect(gateway, loop=client.loop, klass=cls, compression=None) # dynamically add attributes needed ws.token = client.http.token @@ -587,7 +587,7 @@ class DiscordVoiceWebSocket(websockets.client.WebSocketClientProtocol): async def from_client(cls, client, *, resume=False): """Creates a voice websocket for the :class:`VoiceClient`.""" gateway = 'wss://' + client.endpoint + '/?v=3' - ws = await websockets.connect(gateway, loop=client.loop, klass=cls) + ws = await websockets.connect(gateway, loop=client.loop, klass=cls, compression=None) ws.gateway = gateway ws._connection = client ws._max_heartbeat_timeout = 60.0 diff --git a/discord/shard.py b/discord/shard.py index 7a736bfe8..8e04b9c5b 100644 --- a/discord/shard.py +++ b/discord/shard.py @@ -210,7 +210,7 @@ class AutoShardedClient(Client): async def launch_shard(self, gateway, shard_id): try: - coro = websockets.connect(gateway, loop=self.loop, klass=DiscordWebSocket) + coro = websockets.connect(gateway, loop=self.loop, klass=DiscordWebSocket, compression=None) ws = await asyncio.wait_for(coro, loop=self.loop, timeout=180.0) except Exception as e: log.info('Failed to connect for shard_id: %s. Retrying...', shard_id)