diff --git a/src/judas_client/connector.py b/src/judas_client/connector.py index b9b7738..0eb9acf 100644 --- a/src/judas_client/connector.py +++ b/src/judas_client/connector.py @@ -54,6 +54,8 @@ class Connector: self.inbound_buffer: bytes = b"" self.outbound_buffer: bytes = b"" + self.running: bool = True + self.on_message: Callable[[Message], None] = on_message def _send_outbound(self) -> None: @@ -141,7 +143,7 @@ class Connector: """Run the main event loop.""" self.connect() try: - while True: + while self.running: events = self.selector.select(timeout=1) for key, mask in events: if mask & selectors.EVENT_READ: