From bc62dbed66b08990a9aca790a831b07d8b5afcdb Mon Sep 17 00:00:00 2001 From: Artur Borecki Date: Wed, 19 Nov 2025 21:56:38 +0100 Subject: [PATCH] feat(connector.py): handle `self.selector.unregister` exceptions in `Connector.close()` --- src/judas_client/connector.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/judas_client/connector.py b/src/judas_client/connector.py index bab4346..8f47e31 100644 --- a/src/judas_client/connector.py +++ b/src/judas_client/connector.py @@ -106,14 +106,16 @@ class Connector: def send_hello(self) -> None: """Send a HELLO message to the server.""" self.logger.debug("[*] Sending HELLO message...") - hello_message: bytes = Message.hello(self.mac_address).to_bytes() - self.outbound_buffer += hello_message - self._send_outbound() + hello_message: Message = Message.hello(self.mac_address) + self.send(hello_message) def close(self) -> None: """Close the connection and clean up resources.""" self.logger.debug("[*] Closing connection...") - self.selector.unregister(self.socket) + try: + self.selector.unregister(self.socket) + except Exception as e: + self.logger.error(f"[!] Error unregistering socket: {e}") self.socket.close() self.logger.debug("[.] Connection closed.")