judas/judas_client
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 3 Wiki Activity

feat(connector.py): ensure sent data is ACK'd in send()

Browse Source
This commit is contained in:
Artur Borecki
2025-08-30 18:33:21 +02:00
parent 6efc7754f9
commit 5d5dbd371f
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/judas_client/connector.py
View File

@@ -104,14 +104,21 @@ class Connector:
try: try:
self.socket.sendall(data) self.socket.sendall(data)
self.logger.info("[>] Data sent") self.logger.info("[>] Data sent")
self._send_ack() acknowledged: bool = self._check_ack()
break if acknowledged:
self.logger.debug("[.] Data acknowledged")
break
else:
self.logger.warning(
"[!] Data not acknowledged, retrying..."
)
except BrokenPipeError as e: except BrokenPipeError as e:
self.logger.error(f"[!] Broken pipe: {e}") self.logger.error(f"[!] Broken pipe: {e}")
self.logger.info("[.] Reconnecting...") self.logger.info("[.] Reconnecting...")
self.connect() self.connect()
except (socket.error, ValueError) as e: except (socket.error, ValueError) as e:
self.logger.error(f"[!] Failed to send data: {e}") self.logger.error(f"[!] Failed to send data: {e}")
time.sleep(1)
def receive(self) -> bytes: def receive(self) -> bytes:
self.logger.debug("[.] Waiting to receive data...") self.logger.debug("[.] Waiting to receive data...")