Changing to a connections property.
This commit is contained in:
@@ -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]:
|
||||||
|
|||||||
Reference in New Issue
Block a user