1 files changed, 5 insertions, 1 deletions
diff --git a/services/data-collector/src/data_collector/main.py b/services/data-collector/src/data_collector/main.py
index 608d6cd..171db52 100644
--- a/services/data-collector/src/data_collector/main.py
+++ b/services/data-collector/src/data_collector/main.py
@@ -11,6 +11,7 @@ from shared.logging import setup_logging
 from shared.metrics import ServiceMetrics
 from shared.models import Candle
 from shared.notifier import TelegramNotifier
+from shared.shutdown import GracefulShutdown
 
 from data_collector.config import CollectorConfig
 
@@ -83,10 +84,13 @@ async def run() -> None:
     symbols = config.symbols
     timeframe = config.timeframes[0] if config.timeframes else "1Day"
 
+    shutdown = GracefulShutdown()
+    shutdown.install_handlers()
+
     log.info("starting", symbols=symbols, timeframe=timeframe, poll_interval=poll_interval)
 
     try:
-        while True:
+        while not shutdown.is_shutting_down:
             # Check if market is open
             try:
                 is_open = await alpaca.is_market_open()