103 lines
3.6 KiB
Python
103 lines
3.6 KiB
Python
"""后台管理接口：任务/Agent/日志/手动管控。"""
|
|
from typing import Optional
|
|
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
|
|
from app.constants import TaskStatus
|
|
from app.models.schemas import AgentInfo, LogEntry, TaskInfo
|
|
from app.repository.log_repo import LogRepo
|
|
from app.services.agent_service import AgentService
|
|
from app.services.task_service import TaskService
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
def get_task_service(request: Request) -> TaskService:
|
|
return TaskService(request.app.state.redis)
|
|
|
|
|
|
def get_agent_service(request: Request) -> AgentService:
|
|
return AgentService(request.app.state.redis)
|
|
|
|
|
|
def get_log_repo(request: Request) -> LogRepo:
|
|
return LogRepo(request.app.state.redis)
|
|
|
|
|
|
# ---------- 任务管理 ----------
|
|
@router.get("/tasks", response_model=list[TaskInfo], summary="任务列表")
|
|
async def list_tasks(
|
|
status: Optional[TaskStatus] = None,
|
|
limit: int = 100,
|
|
svc: TaskService = Depends(get_task_service),
|
|
):
|
|
return await svc.list(status=status, limit=limit)
|
|
|
|
|
|
@router.get("/tasks/{request_id}", response_model=TaskInfo, summary="任务详情")
|
|
async def task_detail(request_id: str, svc: TaskService = Depends(get_task_service)):
|
|
task = await svc.get(request_id)
|
|
if not task:
|
|
raise HTTPException(status_code=404, detail="task not found")
|
|
return task
|
|
|
|
|
|
@router.post("/tasks/{request_id}/cancel", summary="取消任务")
|
|
async def admin_cancel(request_id: str, svc: TaskService = Depends(get_task_service)):
|
|
if not await svc.cancel(request_id):
|
|
raise HTTPException(status_code=400, detail="cannot cancel task")
|
|
return {"ok": True}
|
|
|
|
|
|
@router.post("/tasks/{request_id}/reset", summary="重置任务状态")
|
|
async def admin_reset(request_id: str, svc: TaskService = Depends(get_task_service)):
|
|
if not await svc.reset(request_id):
|
|
raise HTTPException(status_code=400, detail="cannot reset task")
|
|
return {"ok": True}
|
|
|
|
|
|
# ---------- Agent 管理 ----------
|
|
@router.get("/agents", response_model=list[AgentInfo], summary="Agent 列表")
|
|
async def list_agents(svc: AgentService = Depends(get_agent_service)):
|
|
return await svc.all()
|
|
|
|
|
|
@router.post("/agents/{agent_id}/offline", summary="下线 Agent")
|
|
async def agent_offline(agent_id: str, svc: AgentService = Depends(get_agent_service)):
|
|
if not await svc.take_offline(agent_id):
|
|
raise HTTPException(status_code=404, detail="agent not found")
|
|
return {"ok": True}
|
|
|
|
|
|
# ---------- 日志审计 ----------
|
|
@router.get("/logs", response_model=list[LogEntry], summary="日志审计")
|
|
async def logs(
|
|
limit: int = 200,
|
|
request_id: Optional[str] = None,
|
|
agent_id: Optional[str] = None,
|
|
repo: LogRepo = Depends(get_log_repo),
|
|
):
|
|
return await repo.list(limit=limit, request_id=request_id, agent_id=agent_id)
|
|
|
|
|
|
# ---------- 概览 ----------
|
|
@router.get("/overview", summary="概览统计")
|
|
async def overview(
|
|
task_svc: TaskService = Depends(get_task_service),
|
|
agent_svc: AgentService = Depends(get_agent_service),
|
|
):
|
|
tasks = await task_svc.list(limit=1000)
|
|
agents = await agent_svc.all()
|
|
from collections import Counter
|
|
|
|
status_count = Counter(t.status.value for t in tasks)
|
|
return {
|
|
"task_total": len(tasks),
|
|
"task_pending": status_count.get(TaskStatus.PENDING.value, 0),
|
|
"task_running": status_count.get(TaskStatus.RUNNING.value, 0),
|
|
"task_success": status_count.get(TaskStatus.SUCCESS.value, 0),
|
|
"task_failed": status_count.get(TaskStatus.FAILED.value, 0),
|
|
"agent_total": len(agents),
|
|
"agent_online": sum(1 for a in agents if a.status.value == "online"),
|
|
"agent_offline": sum(1 for a in agents if a.status.value == "offline"),
|
|
} |