auto-stock/backend/app/services/portfolio_cache_bridge.py

"""portfolio_cache_bridge — 매매 엔진(데이트레이딩/스윙)에서 포트폴리오 화면 상태를 즉시 갱신.

역할:
  - 매수 직후: Portfolio 행 upsert(db_worker 큐) + 인메모리 캐시에 즉시 추가 (strategy_type 태깅)
  - 매도 직후: Portfolio 수량 차감/0 (db_worker 큐) + 캐시에서 즉시 제거/차감 + 최근매도 기록
    (KIS 잔고 반영 지연으로 동기화가 되살리는 것 방지)

DB 쓰기는 전부 db_worker 큐 경유(CLAUDE.md §3.2). 캐시는 인메모리라 실시간 계층에서 호출 가능.
portfolio.py 는 지연 import (순환 import·기동 순서 회피).
"""
import logging
from decimal import Decimal

logger = logging.getLogger(__name__)


def _cache():
    from app.api.endpoints.portfolio import portfolio_cache
    return portfolio_cache


async def _db_quantity(stock_code: str) -> int | None:
    """Portfolio 테이블의 현재 수량 (읽기 전용). 조회 실패 시 None."""
    try:
        from sqlalchemy import select
        from app.core.database import AsyncSessionLocal
        from app.models.models import Portfolio

        async with AsyncSessionLocal() as session:
            row = (await session.execute(
                select(Portfolio.quantity).where(Portfolio.stock_code == stock_code)
            )).scalars().first()
        return int(row) if row is not None else None
    except Exception as e:
        logger.debug(f"[cache_bridge] Portfolio 수량 조회 실패 ({stock_code}): {e}")
        return None


async def reflect_buy(
    stock_code: str,
    stock_name: str,
    quantity: int,
    price: float,
    strategy_type: str = "swing",
    market: str = "KRX",
) -> None:
    """매수 체결 즉시 Portfolio(큐) + 캐시 반영."""
    try:
        from app.services.db_worker import enqueue_portfolio_upsert
        from app.models.models import Portfolio

        cache = _cache()
        existing = next((h for h in cache.holdings if h.stock_code == stock_code), None)
        if existing is not None:
            old_qty = int(existing.quantity or 0)
            old_avg = float(existing.average_price or price)
            new_qty = old_qty + quantity
            new_avg = (old_avg * old_qty + price * quantity) / new_qty if new_qty > 0 else price
            existing.quantity = new_qty
            existing.average_price = Decimal(str(round(new_avg, 2)))
            existing.current_price = Decimal(str(price))
            # [2026-09-23] 기존 행의 strategy_type 은 변경하지 않는다.
            # 스윙 8주 보유 중 DT 4주를 사면 행(12주)이 daytrading 이 되어
            # get_swing_holdings 에서 빠지고 스윙 8주의 손절·익절이 완전히 멈춘다.
            # 화면의 DT 그룹은 엔진 포지션(/daytrading/positions)으로 별도 표시한다.
            #
            # [2026-09-28] 예외 하나: 유입 격리('external') 행에 **스윙** 시스템 매수가
            # 들어오면 swing 으로 승격한다. 승격하지 않으면 새로 산 스윙 수량이
            # get_swing_holdings 에서 빠져 다음 sync 까지 손절·익절이 붙지 않는다.
            # DT 매수는 승격하지 않는다 — 엔진이 인메모리로 자기 수량을 관리하고,
            # 'daytrading' 으로 바꾸면 격리 수량이 external 표시에서 사라진다.
            if (existing.strategy_type or "swing") == "external" and strategy_type == "swing":
                existing.strategy_type = "swing"
                _promote_to_swing = True
                logger.warning(
                    f"[cache_bridge] ⬆️ 유입 격리 해제(승격): {stock_name}({stock_code}) "
                    f"external → swing (스윙 시스템 매수 {quantity}주)"
                )
            else:
                _promote_to_swing = False
            avg_for_db, qty_for_db = new_avg, new_qty
        else:
            cache.holdings.append(Portfolio(
                stock_code=stock_code, stock_name=stock_name, quantity=quantity,
                average_price=Decimal(str(price)), current_price=Decimal(str(price)),
                profit_loss=Decimal("0"), profit_loss_rate=Decimal("0"),
                market=market, strategy_type=strategy_type,
                currency_code="KRW" if market == "KRX" else "USD",
            ))
            avg_for_db, qty_for_db = price, quantity

        _upsert = {
            "stock_code": stock_code,
            "stock_name": stock_name,
            "quantity": qty_for_db,
            "average_price": round(avg_for_db, 2),
            "current_price": price,
            "market": market,
        }
        if existing is None:
            # 신규 행일 때만 전략 태깅 (기존 행은 다른 전략 보유분이 섞여 있을 수 있음)
            _upsert["strategy_type"] = strategy_type
        elif _promote_to_swing:
            # external → swing 승격만 DB 에도 반영 (강등·DT 전환은 하지 않는다)
            _upsert["strategy_type"] = "swing"
        await enqueue_portfolio_upsert(_upsert)
        await cache.apply_buy(stock_code, stock_name, quantity, price)
    except Exception as e:
        logger.warning(f"[cache_bridge] reflect_buy 실패 ({stock_code}): {e}")


async def reflect_sell(
    stock_code: str,
    quantity: int,
    price: float,
    remaining_qty: int | None = None,
) -> None:
    """매도 체결 즉시 Portfolio(큐) 차감 + 캐시 제거/차감 + 최근매도 기록.

    remaining_qty: 호출측(전략 엔진)이 아는 '그 전략 포지션'의 잔량. Portfolio 행은
    같은 종목의 스윙+데이트레이딩 수량이 합쳐진 값이므로, 행 잔량은 항상
    `행 수량 - 매도 수량`으로 계산한다. remaining_qty는 캐시에 행이 없을 때의 폴백으로만 쓴다.
    (2026-09-23: DT 전량청산이 스윙 보유분까지 0으로 만들던 버그)
    """
    try:
        from app.services.db_worker import enqueue_portfolio_upsert

        cache = _cache()
        existing = next((h for h in cache.holdings if h.stock_code == stock_code), None)
        avg = float(existing.average_price or price) if existing is not None else price

        # 잔량 산출 기준은 DB(Portfolio) — 캐시는 sync 실패 시 최대 5분까지 낡을 수 있어
        # 캐시 수량으로 절대값을 쓰면 실제와 다른 잔량이 기록된다(2026-09-23 감사 #14).
        db_qty = await _db_quantity(stock_code)
        if db_qty is not None:
            remaining = max(0, db_qty - quantity)
        elif existing is not None:
            remaining = max(0, int(existing.quantity or 0) - quantity)
        elif remaining_qty is not None:
            remaining = max(0, int(remaining_qty))
        else:
            remaining = None  # 알 수 없음 → DB 수량은 건드리지 않음
        if existing is not None and remaining is not None:
            if remaining <= 0:
                cache.holdings = [h for h in cache.holdings if h.stock_code != stock_code]
            else:
                existing.quantity = remaining
                existing.current_price = Decimal(str(price))

        if remaining is not None:
            await enqueue_portfolio_upsert({
                "stock_code": stock_code,
                "quantity": remaining,   # 0 → 목록에서 제외, 다음 동기화에서 삭제
                "current_price": price,
            })
        cache.note_sell(stock_code, quantity)
        await cache.apply_sell(stock_code, quantity, price, avg)
    except Exception as e:
        logger.warning(f"[cache_bridge] reflect_sell 실패 ({stock_code}): {e}")