Files
auto-trade/features/trade/hooks/useKisTradeWebSocket.ts

78 lines
2.6 KiB
TypeScript

import { useEffect, useMemo } from "react";
import { type KisRuntimeCredentials } from "@/features/settings/store/use-kis-runtime-store";
import type {
DashboardRealtimeTradeTick,
DashboardStockOrderBookResponse,
} from "@/features/trade/types/trade.types";
import { resolveTradeTrIds } from "@/features/trade/utils/kisRealtimeUtils";
import { useKisWebSocketStore } from "@/features/kis-realtime/stores/kisWebSocketStore";
import { useDomesticSession } from "@/features/trade/hooks/useDomesticSession";
import { useTradeTickSubscription } from "@/features/trade/hooks/useTradeTickSubscription";
import { useOrderbookSubscription } from "@/features/trade/hooks/useOrderbookSubscription";
/**
* @description KIS 실시간 매매/호가 데이터를 통합 구독하는 커스텀 훅입니다.
* @summary
* - `useDomesticSession`: 장 운영 시간(정규장, 시간외 등) 세션 관리
* - `useTradeTickSubscription`: 실시간 체결가(Tick) 데이터 구독 및 상태 관리
* - `useOrderbookSubscription`: 실시간 호가(Orderbook) 데이터 구독
* - `useKisWebSocketStore`: 전역 웹소켓 연결 상태 및 에러 관리
*
* 위 훅들을 조합(Composition)하여 트레이딩 화면에 필요한 모든 실시간 데이터를 제공합니다.
*/
export function useKisTradeWebSocket(
symbol: string | undefined,
credentials: KisRuntimeCredentials | null,
isVerified: boolean,
onTick?: (tick: DashboardRealtimeTradeTick) => void,
options?: {
orderBookSymbol?: string;
onOrderBookMessage?: (data: DashboardStockOrderBookResponse) => void;
},
) {
const marketSession = useDomesticSession();
const { isConnected, error } = useKisWebSocketStore();
const { latestTick, recentTradeTicks, lastTickAt } = useTradeTickSubscription(
{
symbol,
isVerified,
credentials,
marketSession,
onTick,
},
);
useOrderbookSubscription({
symbol: options?.orderBookSymbol,
isVerified,
credentials,
marketSession,
onOrderBookMessage: options?.onOrderBookMessage,
});
// Connection/Data warning
useEffect(() => {
if (!isConnected || lastTickAt || !symbol) return;
const timer = window.setTimeout(() => {
// Just a warning, not blocking
}, 8000);
return () => window.clearTimeout(timer);
}, [isConnected, lastTickAt, symbol]);
const realtimeTrId = useMemo(() => {
if (!credentials) return "H0STCNT0";
const ids = resolveTradeTrIds(credentials.tradingEnv, marketSession);
return ids[0] ?? "H0STCNT0";
}, [credentials, marketSession]);
return {
latestTick,
recentTradeTicks,
isConnected,
error,
lastTickAt,
realtimeTrId,
};
}