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