feat(ws): 키움 WS 시세 마이그레이션 검증 인프라

KIS WS(41 한도) → 키움 WS(100 한도) 본격 전환 전, 두 소스를 동시 운영해
가격 일치성을 데이터로 검증하기 위한 인프라.

신규
- kiwoom_ws.py: 키움 WS 클라이언트 (KIS WS 와 동일 get_price 인터페이스, 메모리 dict 캐시)
- kis_trader/network/ws_validator.py: 5초마다 KIS↔키움 가격 비교, ws_price_validation 테이블에 1행 INSERT, |diff|≥WARN_PCT 시 WARN 로그
- test_kiwoom_ws.py: 키움 WS 단독 동작 확인 스크립트 (토큰/LOGIN/REG/시세)

수정
- database.py: ws_price_validation 테이블 + ENV 키 (WS_PROVIDER, WS_VALIDATION_INTERVAL_SEC, WS_VALIDATION_DIFF_WARN_PCT) + insert_ws_price_validation/get_ws_validation_stats 헬퍼
- kis_trader/main.py: WS_PROVIDER=kis_with_validation 시 키움 WS + Validator 백그라운드 기동, 종료 시 정리

운영 영향: 0. 매매·시세 의사결정은 항상 KIS WS만 사용. 키움 WS는 백그라운드 비교 기록만.
적용: DB env_config 에 WS_PROVIDER=kis_with_validation INSERT 후 봇 재시작.

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
2026-05-05 21:28:22 +09:00
parent f61c471aac
commit 3fa9eb9bf7
5 changed files with 1087 additions and 1 deletions

View File

@@ -111,6 +111,9 @@ class TradingOrchestrator:
# 시장 급락 서킷브레이커 (KOSPI/KOSDAQ 지수 폭락 시 신규 매수 전면 차단).
# 지수 조회는 모의 도메인 미지원 가능성 → 실키 우선, 폴백 self.client.
self.market_guard: MarketGuard | None = None
# 시세 마이그레이션 검증 (WS_PROVIDER=kis_with_validation 시만 기동)
self.kiwoom_ws = None
self.ws_validator = None
self.strategies: List[BaseStrategy] = []
self._stop = False
@@ -221,6 +224,9 @@ class TradingOrchestrator:
# 시장 급락 서킷브레이커 매니저 기동 (활성/비활성 무관, 토글로 즉시 전환)
self._start_market_guard()
# 시세 WS 마이그레이션 검증 (WS_PROVIDER 토글에 따라 키움 WS 동시 기동)
self._start_ws_validator()
# 전략 등록 후 쓰레드 시작 (각자 독립 쓰레드)
self._register_strategies()
for strat in self.strategies:
@@ -400,6 +406,77 @@ class TradingOrchestrator:
logger.error("MarketGuard 기동 실패 (전략은 가드 없이 동작): %s", e)
self.market_guard = None
# ------------------------------------------------------------------
def _start_ws_validator(self) -> None:
"""시세 WS 마이그레이션 검증 — 키움 WS + Validator 백그라운드 기동.
``WS_PROVIDER`` env 토글:
kis_only : 미기동 (기본). 운영 100% 동일.
kis_with_validation : KIS WS 운영 + 키움 WS 검증 동시 (매매 영향 없음).
kiwoom_only : 추후 단계 (검증 통과 후 시세 키움 전환).
매매·시세 의사결정은 항상 KIS WS 만 사용. 키움 WS 는 ``ws_price_validation``
테이블에 기록만 함. 1~2주 누적 후 통계 분석으로 마이그레이션 결정.
"""
from .utils.env import get_env_from_db
provider = (get_env_from_db("WS_PROVIDER", "kis_only") or "kis_only").strip().lower()
if provider == "kis_only":
logger.info(" WS_PROVIDER=kis_only → 키움 WS 검증 비활성")
return
if provider not in ("kis_with_validation",):
logger.warning(
"⚠️ WS_PROVIDER='%s' 알 수 없음 → kis_only 로 폴백 (검증 비활성)",
provider,
)
return
# 키움 키 로드 (kis_ws._get_kiwoom_creds 재사용)
try:
from kis_ws import _get_kiwoom_creds # type: ignore
app_key, app_secret, is_mock = _get_kiwoom_creds(self.db)
except Exception as e:
logger.warning("키움 자격증명 로드 실패 → 검증 비활성: %s", e)
return
if not app_key or not app_secret:
logger.warning("키움 키 미설정 → WS 검증 비활성")
return
# 키움 WS 인스턴스
try:
from kiwoom_ws import KiwoomWebSocketPriceCache # type: ignore
self.kiwoom_ws = KiwoomWebSocketPriceCache(
app_key, app_secret, is_mock=is_mock,
)
if not self.kiwoom_ws.start():
logger.warning("키움 WS 시작 실패 → 검증 비활성")
self.kiwoom_ws = None
return
except Exception as e:
logger.warning("키움 WS 인스턴스 생성 실패: %s", e)
self.kiwoom_ws = None
return
# KIS WS 핸들 가져오기 (WSManager.ws_cache)
kis_ws_handle = getattr(self.ws, "ws_cache", None)
if kis_ws_handle is None:
logger.warning("KIS WS 핸들 미발견 → Validator 비활성 (검증 불가)")
return
# Validator 기동
try:
from .network.ws_validator import WSPriceValidator
self.ws_validator = WSPriceValidator(
kis_ws=kis_ws_handle,
kiwoom_ws=self.kiwoom_ws,
db=self.db,
)
self.ws_validator.start()
logger.info("🔬 [시세 검증 모드] WS_PROVIDER=%s — KIS↔키움 동시 운영", provider)
except Exception as e:
logger.warning("Validator 기동 실패: %s", e)
self.ws_validator = None
def _build_market_client(self) -> KISClient:
"""
시세/조회 전용 실전 KISClient 생성 (init 단계).
@@ -816,6 +893,16 @@ class TradingOrchestrator:
self.ranking_mgr.stop()
except Exception:
pass
try:
if self.ws_validator:
self.ws_validator.stop()
except Exception:
pass
try:
if self.kiwoom_ws:
self.kiwoom_ws.stop()
except Exception:
pass
try:
self.ws.stop()
except Exception:

View File

@@ -0,0 +1,213 @@
"""
kis_trader/network/ws_validator.py — KIS↔키움 WS 가격 검증
============================================================
주기적으로 같은 종목의 KIS WS 캐시와 키움 WS 캐시를 비교해
``ws_price_validation`` 테이블에 1행 INSERT.
운영(매매)에는 영향 없음 — **읽기만** 함.
검증 모드(``WS_PROVIDER=kis_with_validation``)에서만 기동.
데이터 흐름::
[KIS WS] ─┐
│ 각각 메모리 dict 캐시
[키움 WS] ─┘
[Validator] ── 5s 주기 ──► ws_price_validation 테이블
│ │
│ └─→ |diff| ≥ WARN_PCT 면 WARN 로그
└─→ 24h/1주 통계 분석 → 마이그레이션 전환 결정 근거
비교 대상 종목
--------------
KIS WS 가 구독 중인 종목 = 봇이 매매에 쓰는 가격이 실제로 들어오고 있는 종목.
키움 WS 도 같은 종목을 구독하도록 동기화 (subscribe).
env_config 토글
---------------
``WS_VALIDATION_INTERVAL_SEC`` (기본 5)
``WS_VALIDATION_DIFF_WARN_PCT`` (기본 0.10 — 0.1%p)
"""
from __future__ import annotations
import threading
import time
from typing import Optional, Set
from ..utils.env import get_env_float, get_env_int
from ..utils.logger import get_logger
logger = get_logger("kis_trader.ws_validator")
class WSPriceValidator:
"""KIS↔키움 WS 가격 비교 백그라운드 워커.
매 N초마다:
1) KIS WS 의 구독 중인 종목 목록을 읽음
2) 키움 WS 가 같은 종목을 구독하도록 동기화
3) 두 캐시에서 가격 조회 → DB INSERT
4) |diff_pct| ≥ warn_pct 시 WARN 로그
"""
def __init__(
self,
*,
kis_ws, # kis_ws.KISWebSocketPriceCache 인스턴스
kiwoom_ws, # kiwoom_ws.KiwoomWebSocketPriceCache 인스턴스
db, # database.TradeDB
):
self.kis_ws = kis_ws
self.kiwoom_ws = kiwoom_ws
self.db = db
self._thread: Optional[threading.Thread] = None
self._running = False
self._last_warn_ts: dict = {} # code → 최근 WARN 로그 시각 (스팸 방지)
# ------------------------------------------------------------------
def start(self) -> bool:
if self._thread and self._thread.is_alive():
return True
self._running = True
self._thread = threading.Thread(
target=self._loop, daemon=True, name="WSPriceValidator",
)
self._thread.start()
logger.info(
"✅ WS 가격 검증기 시작 — KIS↔키움 비교 (interval=%ds, warn≥%.2f%%)",
self._interval_sec(), self._warn_pct(),
)
return True
def stop(self) -> None:
self._running = False
# ------------------------------------------------------------------
def _interval_sec(self) -> int:
return max(1, get_env_int("WS_VALIDATION_INTERVAL_SEC", 5))
def _warn_pct(self) -> float:
return max(0.0, get_env_float("WS_VALIDATION_DIFF_WARN_PCT", 0.10))
# ------------------------------------------------------------------
def _kis_subscribed(self) -> Set[str]:
"""KIS WS 가 현재 구독 중인 종목 set."""
# KISWebSocketPriceCache 의 _subscribed 직접 참조 (kis_ws.py 정의)
try:
with self.kis_ws._sub_lock: # type: ignore[attr-defined]
return set(self.kis_ws._subscribed) # type: ignore[attr-defined]
except AttributeError:
return set()
def _sync_kiwoom_subscriptions(self, target: Set[str]) -> None:
"""키움 WS 구독 = KIS WS 구독 으로 맞춤."""
try:
current = set()
with self.kiwoom_ws._sub_lock: # type: ignore[attr-defined]
current = set(self.kiwoom_ws._subscribed) # type: ignore[attr-defined]
to_add = target - current
to_remove = current - target
for code in to_add:
self.kiwoom_ws.subscribe(code)
for code in to_remove:
self.kiwoom_ws.unsubscribe(code)
except Exception as e:
logger.debug("키움 구독 동기화 실패: %s", e)
# ------------------------------------------------------------------
def _loop(self) -> None:
# 시작 직후 KIS/키움 둘 다 캐시 채워질 시간 약간 줌
time.sleep(15)
while self._running:
try:
self._tick()
except Exception as e:
logger.warning("WS 검증기 tick 예외: %s", e)
time.sleep(self._interval_sec())
def _tick(self) -> None:
"""1회 비교."""
codes = self._kis_subscribed()
if not codes:
return
# 키움 WS 구독 동기화 (구독 안 된 종목은 데이터 없으니 의미 없음)
self._sync_kiwoom_subscriptions(codes)
# 키움 WS 연결되어 있어야 의미 있음. 미연결이면 한쪽만 기록.
kiwoom_ready = self.kiwoom_ws.is_connected()
warn_pct = self._warn_pct()
warn_count = 0
sample_count = 0
for code in codes:
kis_data = self.kis_ws.get_price(code, max_age_sec=10.0)
kw_data = self.kiwoom_ws.get_price(code, max_age_sec=10.0) if kiwoom_ready else None
kis_price = self._parse_price(kis_data, "stck_prpr")
kw_price = self._parse_price(kw_data, "stck_prpr")
kis_age = self._parse_age_ms(kis_data)
kw_age = self._parse_age_ms(kw_data)
if kis_price is None and kw_price is None:
continue
# DB 1행 INSERT
self.db.insert_ws_price_validation(
code=code,
kis_price=kis_price,
kiwoom_price=kw_price,
kis_age_ms=kis_age,
kiwoom_age_ms=kw_age,
)
sample_count += 1
# 차이 경고
if kis_price not in (None, 0) and kw_price is not None:
diff_pct = (kw_price - kis_price) / kis_price * 100.0
if abs(diff_pct) >= warn_pct:
# 종목당 60초 1회만 로그 (스팸 방지)
now = time.time()
last = self._last_warn_ts.get(code, 0)
if now - last >= 60:
self._last_warn_ts[code] = now
logger.warning(
"⚠️ [WS 검증] %s 가격 차이 %.3f%% (KIS=%.0f, 키움=%.0f)",
code, diff_pct, kis_price, kw_price,
)
warn_count += 1
if sample_count > 0:
logger.debug(
"📊 [WS 검증] tick: %d종목 비교 (warn=%d, 키움연결=%s)",
sample_count, warn_count, kiwoom_ready,
)
# ------------------------------------------------------------------
@staticmethod
def _parse_price(data: Optional[dict], key: str) -> Optional[float]:
if not data:
return None
try:
v = float(str(data.get(key, "0")).replace(",", ""))
return abs(v) if v else None
except (ValueError, TypeError):
return None
@staticmethod
def _parse_age_ms(data: Optional[dict]) -> Optional[int]:
if not data:
return None
v = data.get("_age_ms")
if v is None:
return None
try:
return int(v)
except (ValueError, TypeError):
return None