gateway: add impl for zstd_stream_send

- gateway: add utils pkg for a basic wrapper for zstandard's own
   stream writer

litecord/gateway/utils.py

@@ -0,0 +1,30 @@
+"""
+
+Litecord
+Copyright (C) 2018-2019  Luna Mendes
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, version 3 of the License.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>.
+
+"""
+
+import asyncio
+
+class WebsocketFileHandler:
+    """A handler around a websocket that wraps normal I/O calls into
+    the websocket's respective asyncio calls via asyncio.ensure_future."""
+    def __init__(self, ws):
+        self.ws = ws
+
+    def write(self, data):
+        """Write data into the websocket"""
+        asyncio.ensure_future(self.ws.send(data))

litecord/gateway/websocket.py

@@ -49,6 +49,8 @@ from litecord.gateway.encoding import (
     encode_json, decode_json, encode_etf, decode_etf
 )
 
+from litecord.gateway.utils import WebsocketFileHandler
+
 from litecord.storage import int_
 
 log = Logger(__name__)
@@ -155,7 +157,11 @@ class GatewayWebsocket:
         await self._chunked_send(data2, 1024)
 
     async def _zstd_stream_send(self, encoded):
-        pass
+        compressor = self.wsp.zsctx.stream_writer(
+            WebsocketFileHandler(self.ws))
+
+        compressor.write(encoded)
+        compressor.flush(zstd.FLUSH_FRAME)
 
     async def send(self, payload: Dict[str, Any]):
         """Send a payload to the websocket.