From 59f3f6bbb772fe4a04f16b245465ef1fb18fb5d8 Mon Sep 17 00:00:00 2001 From: Luna Date: Thu, 30 May 2019 21:58:23 -0300 Subject: [PATCH] gateway.websocket: add ZstdCompressor at start --- litecord/gateway/websocket.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/litecord/gateway/websocket.py b/litecord/gateway/websocket.py index 7283f17..eea81a8 100644 --- a/litecord/gateway/websocket.py +++ b/litecord/gateway/websocket.py @@ -25,6 +25,7 @@ from typing import List, Dict, Any from random import randint import websockets +import zstandard as zstd from logbook import Logger from litecord.auth import raw_token_check @@ -53,7 +54,7 @@ from litecord.storage import int_ log = Logger(__name__) WebsocketProperties = collections.namedtuple( - 'WebsocketProperties', 'v encoding compress zctx tasks' + 'WebsocketProperties', 'v encoding compress zctx zsctx tasks' ) WebsocketObjects = collections.namedtuple( @@ -80,11 +81,14 @@ class GatewayWebsocket: self.presence = self.ext.presence self.ws = ws - self.wsp = WebsocketProperties(kwargs.get('v'), - kwargs.get('encoding', 'json'), - kwargs.get('compress', None), - zlib.compressobj(), - {}) + self.wsp = WebsocketProperties( + kwargs.get('v'), + kwargs.get('encoding', 'json'), + kwargs.get('compress', None), + zlib.compressobj(), + zstd.ZstdCompressor(), + {} + ) log.debug('websocket properties: {!r}', self.wsp)