feat: run UserDataStream in parallel with candle stream

Made-with: Cursor
This commit is contained in:
21in7
2026-03-02 16:25:13 +09:00
parent 7e7f0f4f22
commit 4ffee0ae8b

View File

@@ -9,6 +9,7 @@ from src.notifier import DiscordNotifier
from src.risk_manager import RiskManager from src.risk_manager import RiskManager
from src.ml_filter import MLFilter from src.ml_filter import MLFilter
from src.ml_features import build_features from src.ml_features import build_features
from src.user_data_stream import UserDataStream
class TradingBot: class TradingBot:
@@ -272,7 +273,19 @@ class TradingBot:
balance = await self.exchange.get_balance() balance = await self.exchange.get_balance()
self.risk.set_base_balance(balance) self.risk.set_base_balance(balance)
logger.info(f"기준 잔고 설정: {balance:.2f} USDT (동적 증거금 비율 기준점)") logger.info(f"기준 잔고 설정: {balance:.2f} USDT (동적 증거금 비율 기준점)")
await self.stream.start(
api_key=self.config.api_key, user_stream = UserDataStream(
api_secret=self.config.api_secret, exchange=self.exchange,
on_order_filled=self._on_position_closed,
)
await asyncio.gather(
self.stream.start(
api_key=self.config.api_key,
api_secret=self.config.api_secret,
),
user_stream.start(
api_key=self.config.api_key,
api_secret=self.config.api_secret,
),
) )