238 lines
8.4 KiB
Python
238 lines
8.4 KiB
Python
"""
|
||
Redis 客户端与会话/缓存能力：供应用生命周期、会话历史、任务追踪等使用。
|
||
配置从 app.core.config.settings 读取，禁止业务层散落 os.getenv。
|
||
"""
|
||
|
||
import json
|
||
from datetime import datetime, timezone
|
||
from typing import Any, Dict, List, Optional
|
||
|
||
import redis.asyncio as aioredis
|
||
|
||
from app.core.config import settings
|
||
from app.core.logging import get_logger
|
||
|
||
logger = get_logger(__name__)
|
||
|
||
|
||
class RedisService:
|
||
"""Redis 服务：连接管理、对话历史、通用缓存。"""
|
||
|
||
def __init__(self) -> None:
|
||
self.redis_url = settings.redis_url
|
||
self._client: Optional[aioredis.Redis] = None
|
||
self.session_ttl = settings.redis_session_ttl
|
||
|
||
async def get_client(self) -> aioredis.Redis:
|
||
"""获取 Redis 客户端（延迟初始化）。"""
|
||
if self._client is None:
|
||
try:
|
||
self._client = await aioredis.from_url(
|
||
self.redis_url,
|
||
encoding="utf-8",
|
||
decode_responses=True,
|
||
)
|
||
await self._client.ping()
|
||
logger.info("Redis 连接成功")
|
||
logger.debug("Redis 连接 URL: {}", self.redis_url)
|
||
except Exception as e:
|
||
logger.error("Redis 连接失败: {}", e)
|
||
raise
|
||
return self._client
|
||
|
||
async def close(self) -> None:
|
||
"""关闭 Redis 连接。"""
|
||
if self._client:
|
||
await self._client.close()
|
||
self._client = None
|
||
|
||
def _conversation_key(self, conversation_id: str) -> str:
|
||
return f"conversation:history:{conversation_id}"
|
||
|
||
async def get_conversation_history(
|
||
self, conversation_id: str
|
||
) -> List[Dict[str, Any]]:
|
||
try:
|
||
client = await self.get_client()
|
||
key = self._conversation_key(conversation_id)
|
||
data = await client.get(key)
|
||
if data:
|
||
return json.loads(data)
|
||
return []
|
||
except Exception as e:
|
||
logger.error("获取对话历史失败: {}", e)
|
||
return []
|
||
|
||
async def set_conversation_history(
|
||
self, conversation_id: str, history: List[Dict[str, Any]]
|
||
) -> bool:
|
||
"""整表覆盖会话历史（用于从 DB 回填），应用 session_ttl。"""
|
||
try:
|
||
client = await self.get_client()
|
||
key = self._conversation_key(conversation_id)
|
||
await client.setex(
|
||
key, self.session_ttl, json.dumps(history, ensure_ascii=False)
|
||
)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("写入对话历史失败: {}", e)
|
||
return False
|
||
|
||
async def add_message(
|
||
self,
|
||
conversation_id: str,
|
||
role: str,
|
||
content: str,
|
||
message_type: str = "text",
|
||
voice_session_id: str | None = None,
|
||
timestamp: str | int | None = None,
|
||
audio_duration_seconds: int | None = None,
|
||
) -> bool:
|
||
try:
|
||
client = await self.get_client()
|
||
key = self._conversation_key(conversation_id)
|
||
history = await self.get_conversation_history(conversation_id)
|
||
item = {
|
||
"role": role,
|
||
"content": content,
|
||
"messageType": message_type,
|
||
"timestamp": timestamp or datetime.now(timezone.utc).isoformat(),
|
||
}
|
||
if voice_session_id:
|
||
item["voiceSessionId"] = voice_session_id
|
||
if (
|
||
audio_duration_seconds is not None
|
||
and audio_duration_seconds > 0
|
||
and message_type == "audio"
|
||
):
|
||
item["durationSeconds"] = int(audio_duration_seconds)
|
||
history.append(item)
|
||
await client.setex(
|
||
key, self.session_ttl, json.dumps(history, ensure_ascii=False)
|
||
)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("添加消息失败: {}", e)
|
||
return False
|
||
|
||
async def append_tts_audio_url_to_last_ai_message(
|
||
self, conversation_id: str, url: str
|
||
) -> bool:
|
||
"""向最近一条 AI 消息的 ttsAudioUrls 追加 upload 返回的 canonical URL（非预签名）。客户端通过 GET /messages 等出口收到预签名 URL。"""
|
||
if not url:
|
||
return False
|
||
try:
|
||
client = await self.get_client()
|
||
key = self._conversation_key(conversation_id)
|
||
history = await self.get_conversation_history(conversation_id)
|
||
for i in range(len(history) - 1, -1, -1):
|
||
if history[i].get("role") == "ai":
|
||
existing = history[i].get("ttsAudioUrls")
|
||
urls: List[str] = (
|
||
[x for x in existing if isinstance(x, str)]
|
||
if isinstance(existing, list)
|
||
else []
|
||
)
|
||
urls.append(url)
|
||
history[i]["ttsAudioUrls"] = urls
|
||
break
|
||
else:
|
||
logger.warning(
|
||
"append_tts_audio_url: no ai message in history conversation_id={}",
|
||
conversation_id,
|
||
)
|
||
return False
|
||
await client.setex(
|
||
key, self.session_ttl, json.dumps(history, ensure_ascii=False)
|
||
)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("append_tts_audio_url 失败: {}", e)
|
||
return False
|
||
|
||
async def clear_conversation_history(self, conversation_id: str) -> bool:
|
||
try:
|
||
client = await self.get_client()
|
||
key = self._conversation_key(conversation_id)
|
||
await client.delete(key)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("清除对话历史失败: {}", e)
|
||
return False
|
||
|
||
async def delete_keys_matching_pattern(self, pattern: str) -> int:
|
||
"""按 SCAN 批量删除 key，避免阻塞式 KEYS *。"""
|
||
try:
|
||
client = await self.get_client()
|
||
batch: list[str] = []
|
||
deleted = 0
|
||
|
||
async for key in client.scan_iter(match=pattern):
|
||
batch.append(key)
|
||
if len(batch) >= 200:
|
||
deleted += int(await client.delete(*batch))
|
||
batch.clear()
|
||
if batch:
|
||
deleted += int(await client.delete(*batch))
|
||
return deleted
|
||
except Exception as e:
|
||
logger.error("按 pattern 删除 Redis key 失败: {}", e)
|
||
return 0
|
||
|
||
async def extend_session_ttl(self, conversation_id: str) -> bool:
|
||
try:
|
||
client = await self.get_client()
|
||
key = self._conversation_key(conversation_id)
|
||
await client.expire(key, self.session_ttl)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("延长会话TTL失败: {}", e)
|
||
return False
|
||
|
||
async def set_cache(self, key: str, value: Any, ttl: Optional[int] = None) -> bool:
|
||
try:
|
||
client = await self.get_client()
|
||
data = (
|
||
json.dumps(value, ensure_ascii=False)
|
||
if not isinstance(value, str)
|
||
else value
|
||
)
|
||
if ttl:
|
||
await client.setex(key, ttl, data)
|
||
else:
|
||
await client.set(key, data)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("设置缓存失败: {}", e)
|
||
return False
|
||
|
||
async def get_cache(self, key: str) -> Optional[Any]:
|
||
try:
|
||
client = await self.get_client()
|
||
data = await client.get(key)
|
||
if data:
|
||
try:
|
||
return json.loads(data)
|
||
except json.JSONDecodeError:
|
||
return data
|
||
return None
|
||
except Exception as e:
|
||
logger.error("获取缓存失败: {}", e)
|
||
return None
|
||
|
||
async def delete_cache(self, key: str) -> bool:
|
||
try:
|
||
client = await self.get_client()
|
||
await client.delete(key)
|
||
return True
|
||
except Exception as e:
|
||
logger.error("删除缓存失败: {}", e)
|
||
return False
|
||
|
||
def is_available(self) -> bool:
|
||
return self._client is not None
|
||
|
||
|
||
# 全局单例，供 main 生命周期与各 feature 通过 get_redis_service 或直接引用使用
|
||
redis_service = RedisService()
|