update:healthy接口增加访问密钥

Author: openrz

app/api/v1/health.py

@@ -1,6 +1,7 @@
-from fastapi import APIRouter
-from ...models.voiceprint import HealthResponse
+from fastapi import APIRouter, HTTPException, Query
+from ...services.voiceprint_service import voiceprint_service
 from ...core.logging import get_logger
+from ...core.config import settings
 
 logger = get_logger(__name__)
 
@@ -10,16 +11,32 @@
 @router.get(
     "/health",
     summary="健康检查",
-    response_model=HealthResponse,
-    description="检查服务运行状态",
+    response_model=dict,
+    description="检查服务运行状态，需要提供正确的密钥",
 )
-async def health_check():
+async def health_check(
+    key: str = Query(..., description="访问密钥", example="your-secret-key")
+):
     """
     健康检查接口
 
+    Args:
+        key: 访问密钥，必须与配置中的authorization密钥匹配
+
     Returns:
-        HealthResponse: 服务状态信息
+        dict: 服务状态信息
+
+    Raises:
+        HTTPException: 当密钥不正确时返回401错误
     """
-    return HealthResponse(
-        status="healthy", message="3D-Speaker voiceprint API service running."
-    )
+    # 验证密钥
+    if key != settings.api_token:
+        logger.warning(f"健康检查接口收到无效密钥: {key}")
+        raise HTTPException(status_code=401, detail="密钥验证失败")
+
+    try:
+        count = voiceprint_service.get_voiceprint_count()
+        return {"total_voiceprints": count, "status": "healthy"}
+    except Exception as e:
+        logger.error(f"获取统计信息异常: {e}")
+        raise HTTPException(status_code=500, detail=f"获取统计信息失败: {str(e)}")

app/api/v1/voiceprint.py

@@ -143,29 +143,3 @@ async def delete_voiceprint(
     except Exception as e:
         logger.error(f"删除声纹异常 {speaker_id}: {e}")
         raise HTTPException(status_code=500, detail=f"删除声纹失败: {str(e)}")
-
-
-@router.get(
-    "/stats",
-    summary="获取统计信息",
-    description="获取声纹数据库统计信息",
-    dependencies=[Depends(security)],
-)
-async def get_stats(
-    token: AuthorizationToken,
-):
-    """
-    获取统计信息接口
-
-    Args:
-        token: 接口令牌（Header）
-
-    Returns:
-        dict: 统计信息
-    """
-    try:
-        count = voiceprint_service.get_voiceprint_count()
-        return {"total_voiceprints": count, "status": "healthy"}
-    except Exception as e:
-        logger.error(f"获取统计信息异常: {e}")
-        raise HTTPException(status_code=500, detail=f"获取统计信息失败: {str(e)}")

app/models/voiceprint.py

@@ -38,18 +38,3 @@ class VoiceprintIdentifyResponse(BaseModel):
 
     class Config:
         schema_extra = {"example": {"speaker_id": "user_001", "score": 0.85}}
-
-
-class HealthResponse(BaseModel):
-    """健康检查响应模型"""
-
-    status: str
-    message: str
-
-    class Config:
-        schema_extra = {
-            "example": {
-                "status": "healthy",
-                "message": "3D-Speaker voiceprint API service running.",
-            }
-        }