diff options
| author | TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com> | 2026-04-01 17:46:56 +0900 |
|---|---|---|
| committer | TheSiahxyz <164138827+TheSiahxyz@users.noreply.github.com> | 2026-04-01 17:46:56 +0900 |
| commit | c294e9df6207973306f58186d278e12399f162a3 (patch) | |
| tree | 9f847439eb29c82a77da0dcf7fe426cfa0cbe65b /services/data-collector | |
| parent | 69e88b3b353f1a2ab7a78259b480e8afbd87669c (diff) | |
fix: WS factory, backtester config, CI docker builds, health port docs
Diffstat (limited to 'services/data-collector')
| -rw-r--r-- | services/data-collector/src/data_collector/main.py | 13 | ||||
| -rw-r--r-- | services/data-collector/src/data_collector/ws_factory.py | 33 | ||||
| -rw-r--r-- | services/data-collector/tests/test_ws_factory.py | 18 |
3 files changed, 61 insertions, 3 deletions
diff --git a/services/data-collector/src/data_collector/main.py b/services/data-collector/src/data_collector/main.py index 4384985..b393cc2 100644 --- a/services/data-collector/src/data_collector/main.py +++ b/services/data-collector/src/data_collector/main.py @@ -9,9 +9,15 @@ from shared.logging import setup_logging from shared.metrics import ServiceMetrics from shared.notifier import TelegramNotifier -from data_collector.binance_ws import BinanceWebSocket from data_collector.config import CollectorConfig from data_collector.storage import CandleStorage +from data_collector.ws_factory import create_websocket + + +# Health check port: base (HEALTH_PORT, default 8080) + offset +# data-collector: +0 (8080), strategy-engine: +1 (8081) +# order-executor: +2 (8082), portfolio-manager: +3 (8083) +HEALTH_PORT_OFFSET = 0 async def run() -> None: @@ -43,14 +49,15 @@ async def run() -> None: # Use the first configured timeframe for the WebSocket subscription. timeframe = config.timeframes[0] if config.timeframes else "1m" - ws = BinanceWebSocket( + ws = create_websocket( + exchange_id=config.exchange_id, symbols=config.symbols, timeframe=timeframe, on_candle=on_candle, ) health = HealthCheckServer( - "data-collector", port=config.health_port, auth_token=config.metrics_auth_token + "data-collector", port=config.health_port + HEALTH_PORT_OFFSET, auth_token=config.metrics_auth_token ) health.register_check("redis", broker.ping) await health.start() diff --git a/services/data-collector/src/data_collector/ws_factory.py b/services/data-collector/src/data_collector/ws_factory.py new file mode 100644 index 0000000..b8e2719 --- /dev/null +++ b/services/data-collector/src/data_collector/ws_factory.py @@ -0,0 +1,33 @@ +"""WebSocket factory for exchange-specific connections.""" +import logging + +from data_collector.binance_ws import BinanceWebSocket + +logger = logging.getLogger(__name__) + +# Supported exchanges for WebSocket streaming +SUPPORTED_WS = {"binance": BinanceWebSocket} + + +def create_websocket(exchange_id: str, **kwargs): + """Create an exchange-specific WebSocket handler. + + Args: + exchange_id: Exchange identifier (e.g. 'binance') + **kwargs: Passed to the WebSocket constructor (symbols, timeframe, on_candle) + + Returns: + WebSocket handler instance + + Raises: + ValueError: If exchange is not supported for WebSocket streaming + """ + ws_cls = SUPPORTED_WS.get(exchange_id) + if ws_cls is None: + supported = ", ".join(sorted(SUPPORTED_WS.keys())) + raise ValueError( + f"WebSocket streaming not supported for '{exchange_id}'. " + f"Supported: {supported}. " + f"Use REST polling as fallback for unsupported exchanges." + ) + return ws_cls(**kwargs) diff --git a/services/data-collector/tests/test_ws_factory.py b/services/data-collector/tests/test_ws_factory.py new file mode 100644 index 0000000..ef0449c --- /dev/null +++ b/services/data-collector/tests/test_ws_factory.py @@ -0,0 +1,18 @@ +"""Tests for WebSocket factory.""" +import pytest +from data_collector.ws_factory import create_websocket, SUPPORTED_WS +from data_collector.binance_ws import BinanceWebSocket + + +def test_create_binance_ws(): + ws = create_websocket("binance", symbols=["BTCUSDT"], timeframe="1m", on_candle=lambda c: None) + assert isinstance(ws, BinanceWebSocket) + + +def test_create_unsupported_exchange(): + with pytest.raises(ValueError, match="not supported"): + create_websocket("unsupported_exchange", symbols=["BTCUSDT"], timeframe="1m", on_candle=lambda c: None) + + +def test_supported_exchanges(): + assert "binance" in SUPPORTED_WS |
