bec/ophyd_devices
0
0
Fork 0
mirror of https://github.com/bec-project/ophyd_devices.git synced 2026-01-22 17:32:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix(socketio): receive until buffer is empty

Browse Source
This commit is contained in:
wakonig_k
2024-07-29 21:00:12 +02:00
parent 5ce19e9c5e
commit b76b5d21ce
1 changed files with 11 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ophyd_devices/utils/socket.py
View File

@@ -202,9 +202,17 @@ class SocketIO:
return self.sock.send(msg_bytes) return self.sock.send(msg_bytes)
def _recv(self, buffer_length=1024): def _recv(self, buffer_length=1024):
full_msg = b""
while True:
try:
msg = self.sock.recv(buffer_length) msg = self.sock.recv(buffer_length)
logger.debug(f"recv message: {msg}") 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): def _initialize_socket(self):
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)