From 0e177eafbed026445e50da6a5992177521fb8212 Mon Sep 17 00:00:00 2001
From: TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com>
Date: Thu, 2 Apr 2026 15:42:12 +0900
Subject: feat: add graceful shutdown with SIGTERM/SIGINT handlers to all
 services

---
 services/strategy-engine/src/strategy_engine/main.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'services/strategy-engine')

diff --git a/services/strategy-engine/src/strategy_engine/main.py b/services/strategy-engine/src/strategy_engine/main.py
index 2635b7c..411c54b 100644
--- a/services/strategy-engine/src/strategy_engine/main.py
+++ b/services/strategy-engine/src/strategy_engine/main.py
@@ -13,6 +13,7 @@ from shared.logging import setup_logging
 from shared.metrics import ServiceMetrics
 from shared.notifier import TelegramNotifier
 from shared.sentiment_models import MarketSentiment
+from shared.shutdown import GracefulShutdown
 
 from strategy_engine.config import StrategyConfig
 from strategy_engine.engine import StrategyEngine
@@ -97,6 +98,9 @@ async def run() -> None:
         params = config.strategy_params.get(strategy.name, {})
         strategy.configure(params)
 
+    shutdown = GracefulShutdown()
+    shutdown.install_handlers()
+
     log.info("loaded_strategies", count=len(strategies), names=[s.name for s in strategies])
 
     engine = StrategyEngine(broker=broker, strategies=strategies)
@@ -131,7 +135,7 @@ async def run() -> None:
             )
             log.info("stock_selector_enabled", time=config.selector_final_time)
 
-        await asyncio.gather(*tasks)
+        await shutdown.wait()
     except Exception as exc:
         log.error("fatal_error", error=str(exc))
         await notifier.send_error(str(exc), "strategy-engine")
-- 
cgit v1.2.3