add graceful shutdown

2024-05-09 16:49:59 +02:00
parent 5fe455e42f
commit 537d81fa19
3 changed files with 19 additions and 4 deletions
--- a/app/src/async_stream.py
+++ b/app/src/async_stream.py
@@ -90,11 +90,15 @@ class AsyncStream():
                    f"{traceback.format_exc()}")

    async def disc(self) -> None:
+        if self.writer.is_closing():
+            return
        logger.debug(f'AsyncStream.disc() l{self.l_addr} | r{self.r_addr}')
        self.writer.close()
        await self.writer.wait_closed()

    def close(self):
+        if self.writer.is_closing():
+            return
        logger.debug(f'AsyncStream.close() l{self.l_addr} | r{self.r_addr}')
        self.writer.close()