Merge pull request #3 from vacp2p/Alberto/utils-init

Add status-subscriber

docker-utils/status-subscriber/Dockerfile

@@ -0,0 +1,10 @@
+FROM python:3.13-slim
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install -r requirements.txt
+
+COPY subscribe.py .
+
+CMD ["python", "subscribe.py"]

docker-utils/status-subscriber/README.md

@@ -0,0 +1,8 @@
+# status-subscriber
+
+Simple container that will connect to `status-backend` signal to log information.
+
+## Changelog
+
+- v1.0.0
+  - Working with status-backend `b22c58bd3bdd4a387dc09dccba1d866d5ae09adf`

docker-utils/status-subscriber/requirements.txt

@@ -0,0 +1 @@
+websocket-client==1.8.0

docker-utils/status-subscriber/subscribe.py

@@ -0,0 +1,82 @@
+import os
+import time
+import json
+import logging
+import signal
+import sys
+import websocket
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
+logger = logging.getLogger(__name__)
+
+# Retrieve WebSocket URL from environment variable
+WEBSOCKET_URL = os.getenv("WEBSOCKET_URL", "ws://localhost:3333/signals")
+
+
+ws = None
+should_exit = False
+
+
+def on_message(ws, message):
+    try:
+        data = json.loads(message)
+        logger.info(f"Received valid message: {data}")
+    except json.JSONDecodeError:
+        logger.warning(f"Received invalid JSON message: {message}")
+
+
+def on_error(ws, error):
+    logger.error(f"WebSocket error: {error}")
+
+
+def on_close(ws, close_status_code, close_msg):
+    logger.info(f"WebSocket connection closed. Status code: {close_status_code}, Message: {close_msg}")
+
+
+def on_open(ws):
+    logger.info("Successfully connected to WebSocket server.")
+
+
+def connect_to_websocket():
+    global ws
+    reconnect_delay = 1
+    max_delay = 60  # Cap the reconnect delay to avoid indefinite growth
+
+    while not should_exit:
+        try:
+            logger.info(f"Connecting to WebSocket endpoint: {WEBSOCKET_URL}")
+            ws = websocket.WebSocketApp(
+                WEBSOCKET_URL,
+                on_message=on_message,
+                on_error=on_error,
+                on_close=on_close,
+                on_open=on_open
+            )
+            ws.run_forever()
+            logger.info("WebSocket connection closed. Retrying...")
+        except Exception as e:
+            logger.error(f"Error in WebSocket connection: {e}")
+
+        # Exponential backoff for reconnection
+        reconnect_delay = min(reconnect_delay * 2, max_delay)
+        if not should_exit:
+            logger.info(f"Reconnecting in {reconnect_delay} seconds...")
+            time.sleep(reconnect_delay)
+
+
+def signal_handler(sig, frame):
+    """Handle termination signals (SIGINT, SIGTERM) for graceful shutdown."""
+    global should_exit
+    logger.info("Received termination signal. Shutting down...")
+    should_exit = True
+    if ws:
+        ws.close()
+    sys.exit(0)
+
+
+if __name__ == "__main__":
+    signal.signal(signal.SIGINT, signal_handler)
+    signal.signal(signal.SIGTERM, signal_handler)
+
+    connect_to_websocket()