5.15.9 - Add stop sig bypass

Author: vkottler

runtimepy/net/arbiter/base.py

@@ -237,9 +237,10 @@ async def _init_connections(self) -> None:
             self._register_connection(value, key)
 
         # Ensure connections are all initialized.
-        await _asyncio.gather(
-            *(x.initialized.wait() for x in self._connections.values())
-        )
+        with self.log_time("Connection initialization", reminder=True):
+            await _asyncio.gather(
+                *(x.initialized.wait() for x in self._connections.values())
+            )
 
     async def _build_structs(self, info: AppInfo) -> None:
         """Build structs."""
@@ -372,6 +373,8 @@ async def _entry(
         info: Optional[AppInfo] = None
 
         try:
+            self.stop_sig.clear()
+
             async with _AsyncExitStack() as stack:
                 result, info = await self._main(
                     stack,

runtimepy/net/connection.py

@@ -43,6 +43,7 @@ class Connection(
     byte_order: ByteOrder = DEFAULT_BYTE_ORDER
 
     default_auto_restart = False
+    ignore_stop_sig = False
 
     def __init__(
         self,
@@ -300,7 +301,7 @@ async def process(
             )
 
         # Allow a stop signal to also disable the connection.
-        if stop_sig is not None:
+        if stop_sig is not None and not self.ignore_stop_sig:
             self._tasks.append(_asyncio.create_task(self._wait_sig(stop_sig)))
 
         self.exited.clear()

runtimepy/net/manager.py

@@ -122,6 +122,9 @@ async def manage(self, stop_sig: _asyncio.Event) -> None:
             tasks = next_tasks
 
         with self.log_time("Shutting down", reminder=True):
+            for conn in self._conns:
+                conn.disable("connection manager shutdown")
+
             # Allow existing tasks to clean up.
             if new_conn_task is not None:
                 new_conn_task.cancel()