mirror of
https://github.com/bec-project/ophyd_devices.git
synced 2026-01-22 09:32:19 +01:00
fix(socketio): receive until buffer is empty
This commit is contained in:
@@ -202,9 +202,17 @@ class SocketIO:
|
||||
return self.sock.send(msg_bytes)
|
||||
|
||||
def _recv(self, buffer_length=1024):
|
||||
msg = self.sock.recv(buffer_length)
|
||||
logger.debug(f"recv message: {msg}")
|
||||
return msg
|
||||
full_msg = b""
|
||||
while True:
|
||||
try:
|
||||
msg = self.sock.recv(buffer_length)
|
||||
logger.debug(f"recv message: {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)
|
||||
|
||||
Reference in New Issue
Block a user