Changing to a connections property.

This commit is contained in:
Michael Woods
2025-02-15 15:57:58 -05:00
parent be3e76acfa
commit c90052fdbe

View File

@@ -25,11 +25,17 @@ class Client:
self.started = False self.started = False
signal.signal(signal.SIGINT, self.exit_gracefully) signal.signal(signal.SIGINT, self.exit_gracefully)
signal.signal(signal.SIGTERM, self.exit_gracefully) signal.signal(signal.SIGTERM, self.exit_gracefully)
self.connection_map = None
def exit_gracefully(self, signum, frame): def exit_gracefully(self, signum, frame):
self.stop() self.stop()
@property
def connections(self) -> dict:
if not self.started:
return {}
else:
return self.app._engine._active_handler._handlers[1]._connection_map._connections
def stop(self): def stop(self):
self.started = False self.started = False
self.clear_connections() self.clear_connections()
@@ -58,8 +64,8 @@ class Client:
raise RuntimeError("Connection disconnected unexpectedly.") raise RuntimeError("Connection disconnected unexpectedly.")
time.sleep(.1) time.sleep(.1)
logging.debug(f"Connection to {dest} ready.") logging.debug(f"Connection to {dest} ready.")
logging.debug("Allowing connection to stabilize for 5 seconds") logging.debug("Allowing connection to stabilize for 8 seconds")
time.sleep(5) time.sleep(8)
return conn return conn
def send_and_receive(self, req: Request, conn: PacketServerConnection, timeout: int = 300) -> Optional[Response]: def send_and_receive(self, req: Request, conn: PacketServerConnection, timeout: int = 300) -> Optional[Response]: