velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

gateway: don't set FIN bit on zlib-stream chunks 

close #139
This commit is contained in:
Luna 2022-01-25 23:46:53 -03:00
parent d77a84be97
commit f792769656
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
litecord/gateway/websocket.py
View File

@ -268,16 +268,13 @@ class GatewayWebsocket:
"""Split data in chunk_size-big chunks and send them """Split data in chunk_size-big chunks and send them
over the websocket.""" over the websocket."""
log.debug( log.debug(
"zlib-stream: chunking {} bytes into {}-byte chunks", len(data), chunk_size "zlib-stream: sending {} bytes into {}-byte chunks", len(data), chunk_size
) )
total_chunks = 0 # we send the entire iterator as per websockets documentation
for chunk in yield_chunks(data, chunk_size): # to pretent setting FIN when we don't want to
total_chunks += 1 # see https://gitlab.com/litecord/litecord/-/issues/139
log.debug("zlib-stream: chunk {}", total_chunks) await self.ws.send(yield_chunks(data, chunk_size))
await self.ws.send(chunk)
log.debug("zlib-stream: sent {} chunks", total_chunks)
async def _zlib_stream_send(self, encoded): async def _zlib_stream_send(self, encoded):
"""Sending a single payload across multiple compressed """Sending a single payload across multiple compressed
@ -314,7 +311,7 @@ class GatewayWebsocket:
# clients should handle chunked sends (via detection # clients should handle chunked sends (via detection
# of the ZLIB_SUFFIX suffix appended to data2), so # of the ZLIB_SUFFIX suffix appended to data2), so
# this shouldn't being problems. # this shouldn't cause problems.
# TODO: the chunks are 1024 bytes, 1KB, is this good enough? # TODO: the chunks are 1024 bytes, 1KB, is this good enough?
await self._chunked_send(data1, 1024) await self._chunked_send(data1, 1024)