auto-stock/backend/app/core/redis.py

"""
Redis 클라이언트 모듈

- redis.asyncio 기반 싱글턴 비동기 클라이언트
- init_redis() / close_redis() → FastAPI lifespan에서 호출
- setex_json() / get_json() 헬퍼 함수 제공
- 연결 실패 시 경고 로그 후 None 반환 (서비스 무중단)

키 스키마:
  tick:{code}              실시간 체결가 (TTL: TTL_TICK)
  quote:{code}             실시간 호가   (TTL: TTL_QUOTE)
  recommendations:latest   배치 추천 목록 (TTL: TTL_RECOMMENDATIONS)
  analysis:{code}          종목별 분석   (TTL: TTL_ANALYSIS)
"""
import json
import logging
import time
from typing import Any, Optional

import redis.asyncio as aioredis
from redis.asyncio import Redis

from app.core.config import settings
from app.core.json_helpers import _dumps

logger = logging.getLogger(__name__)

# ── TTL 상수 ─────────────────────────────────────────────────────────────────
TTL_TICK            = 10      # tick:{code}              실시간 체결가 캐시 (초)
TTL_QUOTE           = 10      # quote:{code}             실시간 호가 캐시 (초)
TTL_RECOMMENDATIONS = 1800    # recommendations:latest   배치 추천 캐시 (30분)
TTL_ANALYSIS        = 3600    # analysis:{code}          종목별 분석 결과 (1시간)

# ── 싱글턴 클라이언트 ──────────────────────────────────────────────────────────
_redis_client: Optional[Redis] = None

# [라운드4] 초기 연결 실패 후 지연 재연결 쓰로틀(초). get_redis() 가 None 일 때만 시도한다.
REDIS_RECONNECT_INTERVAL_SEC = 30.0
_last_reconnect_attempt: float = 0.0
_shutting_down: bool = False


def _build_client() -> Redis:
    """클라이언트 생성(from_url 은 지연 연결이라 네트워크 I/O 없음)."""
    return aioredis.from_url(
        settings.REDIS_URL,
        encoding="utf-8",
        decode_responses=True,
        socket_timeout=settings.REDIS_SOCKET_TIMEOUT,
        socket_connect_timeout=settings.REDIS_SOCKET_CONNECT_TIMEOUT,
    )


async def init_redis() -> None:
    """FastAPI lifespan startup에서 호출.

    REDIS_URL 은 settings.REDIS_URL 사용 (기본: redis://localhost:6379/0).
    연결 실패 시 _redis_client = None 유지 → 캐시 함수들이 조용히 skip.
    """
    global _redis_client, _last_reconnect_attempt, _shutting_down
    _shutting_down = False
    _last_reconnect_attempt = time.monotonic()
    try:
        # [2026-10-02] 소켓 타임아웃 — 이전에는 무제한이라 Redis 가 행(응답 없음)이면 DT 청산
        # 마무리·분산락 획득이 무기한 멈췄다. 타임아웃 시 redis.TimeoutError 가 나고, 기존 호출부의
        # 예외 처리(캐시 skip·락 실패 경로)로 흘러간다.
        client = _build_client()
        await client.ping()
        _redis_client = client
        _masked_url = settings.REDIS_URL.split("@")[-1] if "@" in settings.REDIS_URL else settings.REDIS_URL
        logger.info(f"✅ Redis 연결 완료: {_masked_url}")
    except Exception as e:
        logger.warning(
            f"⚠️ Redis 연결 실패 (캐시 비활성화, {REDIS_RECONNECT_INTERVAL_SEC:.0f}초 간격 지연 재연결): {e}"
        )
        _redis_client = None


async def close_redis() -> None:
    """FastAPI lifespan shutdown에서 호출."""
    global _redis_client, _shutting_down
    _shutting_down = True
    if _redis_client:
        await _redis_client.aclose()
        _redis_client = None
        logger.info("🔴 Redis 연결 종료")


def get_redis() -> Optional[Redis]:
    """현재 Redis 클라이언트 반환.

    초기 연결 실패로 None 이면 REDIS_RECONNECT_INTERVAL_SEC 쓰로틀로 클라이언트를 재생성한다.
    재생성 클라이언트는 연결 미검증(지연 연결)이므로 Redis 가 아직 죽어 있으면 호출부의
    기존 예외 처리(경고 후 skip)로 흘러가고, 복구되면 자동으로 정상 동작한다.
    """
    global _redis_client, _last_reconnect_attempt
    if _redis_client is not None or _shutting_down:
        return _redis_client
    now = time.monotonic()
    if now - _last_reconnect_attempt < REDIS_RECONNECT_INTERVAL_SEC:
        return None
    _last_reconnect_attempt = now
    try:
        _redis_client = _build_client()
        logger.warning("⚠️ Redis 지연 재연결: 클라이언트 재생성(연결은 첫 사용 시 확인)")
    except Exception as e:
        logger.warning(f"⚠️ Redis 지연 재연결 실패: {e}")
        _redis_client = None
    return _redis_client


# ── 헬퍼 함수 ─────────────────────────────────────────────────────────────────

async def setex_json(key: str, ttl: int, data: Any) -> bool:
    """JSON 직렬화 후 SETEX. 실패 시 False 반환 (non-blocking).

    Args:
        key:  Redis 키 (예: "tick:005930")
        ttl:  만료 시간 (초)
        data: JSON 직렬화 가능한 객체

    Returns:
        True: 저장 성공 / False: Redis 미연결 또는 직렬화·네트워크 오류
    """
    client = get_redis()
    if not client:
        return False
    try:
        await client.setex(key, ttl, _dumps(data))
        return True
    except Exception as e:
        logger.warning(f"⚠️ Redis SETEX 실패 [{key}]: {e}")
        return False


async def get_json(key: str) -> Optional[Any]:
    """Redis GET 후 JSON 역직렬화. 없거나 실패 시 None 반환.

    Args:
        key: Redis 키 (예: "tick:005930")

    Returns:
        파싱된 데이터 또는 None (캐시 미스 / Redis 미연결)
    """
    client = get_redis()
    if not client:
        return None
    try:
        value = await client.get(key)
        if value is None:
            return None
        return json.loads(value)
    except Exception as e:
        logger.warning(f"⚠️ Redis GET 실패 [{key}]: {e}")
        return None


async def delete_key(key: str) -> bool:
    """Redis 키 삭제. 실패 시 False 반환."""
    client = get_redis()
    if not client:
        return False
    try:
        await client.delete(key)
        return True
    except Exception as e:
        logger.warning(f"⚠️ Redis DELETE 실패 [{key}]: {e}")
        return False