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
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


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

    REDIS_URL 은 settings.REDIS_URL 사용 (기본: redis://localhost:6379/0).
    연결 실패 시 _redis_client = None 유지 → 캐시 함수들이 조용히 skip.
    """
    global _redis_client
    try:
        client = aioredis.from_url(
            settings.REDIS_URL,
            encoding="utf-8",
            decode_responses=True,
        )
        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 연결 실패 (캐시 비활성화): {e}")
        _redis_client = None


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


def get_redis() -> Optional[Redis]:
    """현재 Redis 클라이언트 반환. 연결 안 됐으면 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