From 7077652c8eb25958b95aeb0812175dbefea1c421 Mon Sep 17 00:00:00 2001 From: Alex Cheema Date: Sat, 13 Jul 2024 20:43:37 -0700 Subject: [PATCH] graceful node shutdown --- main_dynamic.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/main_dynamic.py b/main_dynamic.py index e2ee94cc..c5bf7bf0 100644 --- a/main_dynamic.py +++ b/main_dynamic.py @@ -39,7 +39,7 @@ async def shutdown(signal, loop): [task.cancel() for task in server_tasks] print(f"Cancelling {len(server_tasks)} outstanding tasks") await asyncio.gather(*server_tasks, return_exceptions=True) - await server.shutdown() + await server.stop() loop.stop() async def main(): @@ -61,6 +61,8 @@ if __name__ == "__main__": asyncio.set_event_loop(loop) try: loop.run_until_complete(main()) + except KeyboardInterrupt: + print("Received keyboard interrupt. Shutting down...") finally: + loop.run_until_complete(shutdown(signal.SIGTERM, loop)) loop.close() -