You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
233 lines
7.2 KiB
233 lines
7.2 KiB
"""管理后台API路由 - 对应Go的api/admin_router.go"""
|
|
from fastapi import APIRouter, Depends, HTTPException, Header, Query
|
|
from typing import Optional
|
|
|
|
from app.models import (
|
|
Response, ConfigListRequest, ConfigUpdateRequest,
|
|
ReloadRequest, AdapterToggleRequest, AdapterConfigUpdateRequest,
|
|
APITestRequest, WSTestRequest, TestHistoryRequest
|
|
)
|
|
from app.services import ConfigService, AdapterService, TestService
|
|
from app.core.config import get_config
|
|
|
|
admin_router = APIRouter()
|
|
|
|
# 服务实例
|
|
config_service = ConfigService()
|
|
adapter_service = AdapterService()
|
|
test_service = TestService()
|
|
|
|
|
|
def verify_admin_token(x_admin_token: Optional[str] = Header(None)):
|
|
"""验证Admin Token"""
|
|
# TODO: 实现Token验证
|
|
return x_admin_token
|
|
|
|
|
|
# ============================================
|
|
# 系统管理接口
|
|
# ============================================
|
|
|
|
@admin_router.get("/admin/system/status", response_model=Response)
|
|
def get_system_status(
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""获取系统状态"""
|
|
try:
|
|
data = config_service.get_system_status()
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.post("/admin/system/reload", response_model=Response)
|
|
def reload_config(
|
|
req: Optional[ReloadRequest] = None,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""热加载配置"""
|
|
try:
|
|
if req is None:
|
|
req = ReloadRequest()
|
|
data = config_service.reload_config(req)
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.post("/admin/system/restart", response_model=Response)
|
|
def restart_service(
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""重启服务"""
|
|
# TODO: 实现服务重启逻辑
|
|
return Response(
|
|
code=0,
|
|
message="重启命令已发送",
|
|
data={"status": "restarting"}
|
|
)
|
|
|
|
|
|
# ============================================
|
|
# 配置管理接口
|
|
# ============================================
|
|
|
|
@admin_router.get("/admin/config", response_model=Response)
|
|
def get_config_list(
|
|
type: Optional[str] = Query(None, description="配置类型筛选"),
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""获取配置列表"""
|
|
try:
|
|
from app.models import ConfigType
|
|
req = ConfigListRequest()
|
|
if type:
|
|
req.type = ConfigType(type)
|
|
|
|
data = config_service.get_config_list(req)
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.put("/admin/config", response_model=Response)
|
|
def update_config(
|
|
req: ConfigUpdateRequest,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""更新配置"""
|
|
try:
|
|
data = config_service.update_config(req)
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.post("/admin/config/reload", response_model=Response)
|
|
def reload_config_endpoint(
|
|
req: Optional[ReloadRequest] = None,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""热加载配置"""
|
|
return reload_config(req, token)
|
|
|
|
|
|
# ============================================
|
|
# 适配器管理接口
|
|
# ============================================
|
|
|
|
@admin_router.get("/admin/adapters", response_model=Response)
|
|
def get_adapter_list(
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""获取适配器列表"""
|
|
try:
|
|
data = adapter_service.get_adapter_list()
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.post("/admin/adapters/toggle", response_model=Response)
|
|
def toggle_adapter(
|
|
req: AdapterToggleRequest,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""切换适配器状态"""
|
|
try:
|
|
adapter_service.toggle_adapter(req)
|
|
return Response(code=0, message="success")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.put("/admin/adapters/config", response_model=Response)
|
|
def update_adapter_config(
|
|
req: AdapterConfigUpdateRequest,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""更新适配器配置"""
|
|
try:
|
|
adapter_service.update_adapter_config(req)
|
|
return Response(code=0, message="success")
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
# ============================================
|
|
# 测试管理接口
|
|
# ============================================
|
|
|
|
@admin_router.get("/admin/tests/api", response_model=Response)
|
|
def get_api_test_list(
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""获取API测试列表"""
|
|
try:
|
|
data = test_service.get_api_test_list()
|
|
# 设置基础URL
|
|
config = get_config()
|
|
data.base_url = f"http://localhost:{config.server.port}"
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.post("/admin/tests/api/run", response_model=Response)
|
|
async def run_api_test(
|
|
req: APITestRequest,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""执行API测试"""
|
|
try:
|
|
config = get_config()
|
|
base_url = f"http://localhost:{config.server.port}"
|
|
data = await test_service.run_api_test(base_url, req)
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.get("/admin/tests/ws", response_model=Response)
|
|
def get_ws_test_list(
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""获取WebSocket测试列表"""
|
|
try:
|
|
data = test_service.get_ws_test_list()
|
|
config = get_config()
|
|
data.ws_url = f"ws://localhost:{config.server.port}/v1/stream"
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.post("/admin/tests/ws/run", response_model=Response)
|
|
async def run_ws_test(
|
|
req: WSTestRequest,
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""执行WebSocket测试"""
|
|
try:
|
|
config = get_config()
|
|
ws_url = f"ws://localhost:{config.server.port}/v1/stream"
|
|
data = await test_service.run_ws_test(ws_url, req)
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|
|
|
|
|
|
@admin_router.get("/admin/tests/history", response_model=Response)
|
|
def get_test_history(
|
|
type: Optional[str] = Query(None, description="测试类型"),
|
|
limit: int = Query(default=20, ge=1, le=100, description="数量限制"),
|
|
token: str = Depends(verify_admin_token)
|
|
):
|
|
"""获取测试历史"""
|
|
try:
|
|
req = TestHistoryRequest(type=type, limit=limit)
|
|
data = test_service.get_test_history(req)
|
|
return Response(code=0, message="success", data=data)
|
|
except Exception as e:
|
|
raise HTTPException(status_code=500, detail=str(e))
|