update:优化声纹识别,添加服务器健康检查机制避免运行时超时等待

This commit is contained in:
3030332422
2025-08-14 17:57:46 +08:00
parent 7a4431d80c
commit dcf8c134a2
@@ -1,7 +1,7 @@
import asyncio
import json
import time
import aiohttp
import requests
from urllib.parse import urlparse, parse_qs
from typing import Optional, Dict
from config.logger import setup_logging
@@ -57,8 +57,13 @@ class VoiceprintProvider:
logger.bind(tag=TAG).warning("未配置有效的说话人,声纹识别将被禁用")
self.enabled = False
else:
self.enabled = True
logger.bind(tag=TAG).info(f"声纹识别已配置: API={self.api_url}, 说话人={len(self.speaker_ids)}")
# 进行健康检查,验证服务器是否可用
if self._check_server_health():
self.enabled = True
logger.bind(tag=TAG).info(f"声纹识别已启用: API={self.api_url}, 说话人={len(self.speaker_ids)}")
else:
self.enabled = False
logger.bind(tag=TAG).warning(f"声纹识别服务器不可用,声纹识别已禁用: {self.api_url}")
def _parse_speakers(self) -> Dict[str, Dict[str, str]]:
"""解析说话人配置"""
@@ -76,6 +81,41 @@ class VoiceprintProvider:
logger.bind(tag=TAG).warning(f"解析说话人配置失败: {speaker_str}, 错误: {e}")
return speaker_map
def _check_server_health(self) -> bool:
"""检查声纹识别服务器健康状态"""
if not self.api_url or not self.api_key:
return False
try:
# 健康检查URL
parsed_url = urlparse(self.api_url)
health_url = f"{parsed_url.scheme}://{parsed_url.netloc}/voiceprint/health?key={self.api_key}"
# 发送健康检查请求
response = requests.get(health_url, timeout=3)
if response.status_code == 200:
result = response.json()
if result.get("status") == "healthy":
logger.bind(tag=TAG).info("声纹识别服务器健康检查通过")
return True
else:
logger.bind(tag=TAG).warning(f"声纹识别服务器状态异常: {result}")
return False
else:
logger.bind(tag=TAG).warning(f"声纹识别服务器健康检查失败: HTTP {response.status_code}")
return False
except requests.exceptions.ConnectTimeout:
logger.bind(tag=TAG).warning("声纹识别服务器连接超时")
return False
except requests.exceptions.ConnectionError:
logger.bind(tag=TAG).warning("声纹识别服务器连接被拒绝")
return False
except Exception as e:
logger.bind(tag=TAG).warning(f"声纹识别服务器健康检查异常: {e}")
return False
async def identify_speaker(self, audio_data: bytes, session_id: str) -> Optional[str]:
"""识别说话人"""
if not self.enabled or not self.api_url or not self.api_key:
@@ -132,3 +172,4 @@ class VoiceprintProvider:
elapsed = time.monotonic() - api_start_time
logger.bind(tag=TAG).error(f"声纹识别失败: {e}")
return None