fix(socketio): receive until buffer is empty

This commit is contained in:
2024-07-29 21:00:12 +02:00
parent 5ce19e9c5e
commit b76b5d21ce

View File

@@ -202,9 +202,17 @@ class SocketIO:
return self.sock.send(msg_bytes)
def _recv(self, buffer_length=1024):
full_msg = b""
while True:
try:
msg = self.sock.recv(buffer_length)
logger.debug(f"recv message: {msg}")
return msg
except TimeoutError:
msg = b""
full_msg += msg
if len(msg) < buffer_length:
break
return full_msg
def _initialize_socket(self):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)