feat: Redis 싱글턴 클라이언트 추가 + kis_realtime_stream SETEX 연동
- backend/app/core/redis.py 신규 생성
- TTL 상수: TTL_TICK/QUOTE=10s, RECOMMENDATIONS=1800s, ANALYSIS=3600s
- init_redis() / close_redis() → FastAPI lifespan 훅
- setex_json() / get_json() / delete_key() 헬퍼
- Redis 미연결 시 경고 로그 후 non-blocking 계속 동작
- kis_realtime_stream.py: tick/quote 수신 즉시 Redis SETEX 캐시
- _cache_realtime_data() 모듈 함수 추가
- tick:{code} / quote:{code} 키 저장 (API 없이 현재가 조회 가능)
- main.py lifespan: init_redis() / close_redis() 호출 추가
This commit is contained in:
127
backend/app/core/redis.py
Normal file
127
backend/app/core/redis.py
Normal file
@@ -0,0 +1,127 @@
|
||||
"""
|
||||
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
|
||||
|
||||
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
|
||||
logger.info(f"✅ Redis 연결 완료: {settings.REDIS_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, json.dumps(data, ensure_ascii=False))
|
||||
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
|
||||
Reference in New Issue
Block a user