284 lines
8.9 KiB
Python
284 lines
8.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""执行与队列 API
|
||
|
||
提供 8 个端点：
|
||
- POST /api/execution/run — 直接执行任务
|
||
- GET /api/execution/queue — 获取当前队列（按 site_id 过滤）
|
||
- POST /api/execution/queue — 添加到队列
|
||
- PUT /api/execution/queue/reorder — 重排队列
|
||
- DELETE /api/execution/queue/{id} — 删除队列任务
|
||
- POST /api/execution/{id}/cancel — 取消执行中的任务
|
||
- GET /api/execution/history — 执行历史（按 site_id 过滤）
|
||
- GET /api/execution/{id}/logs — 获取历史日志
|
||
|
||
所有端点需要 JWT 认证，site_id 从 JWT 提取。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import json
|
||
import logging
|
||
import uuid
|
||
|
||
from fastapi import APIRouter, Depends, HTTPException, Query, status
|
||
|
||
from app.auth.dependencies import CurrentUser, get_current_user
|
||
from app.database import get_connection
|
||
from app.schemas.execution import (
|
||
ExecutionHistoryItem,
|
||
ExecutionLogsResponse,
|
||
ExecutionRunResponse,
|
||
QueueTaskResponse,
|
||
ReorderRequest,
|
||
)
|
||
from app.schemas.tasks import TaskConfigSchema
|
||
from app.services.task_executor import task_executor
|
||
from app.services.task_queue import task_queue
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
router = APIRouter(prefix="/api/execution", tags=["任务执行"])
|
||
|
||
|
||
# ── POST /api/execution/run — 直接执行任务 ────────────────────
|
||
|
||
@router.post("/run", response_model=ExecutionRunResponse)
|
||
async def run_task(
|
||
config: TaskConfigSchema,
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> ExecutionRunResponse:
|
||
"""直接执行任务（不经过队列）。
|
||
|
||
从 JWT 注入 store_id，创建 execution_id 后异步启动子进程。
|
||
"""
|
||
config = config.model_copy(update={"store_id": user.site_id})
|
||
execution_id = str(uuid.uuid4())
|
||
|
||
# 异步启动执行，不阻塞响应
|
||
asyncio.create_task(
|
||
task_executor.execute(
|
||
config=config,
|
||
execution_id=execution_id,
|
||
site_id=user.site_id,
|
||
)
|
||
)
|
||
|
||
return ExecutionRunResponse(
|
||
execution_id=execution_id,
|
||
message="任务已提交执行",
|
||
)
|
||
|
||
|
||
# ── GET /api/execution/queue — 获取当前队列 ───────────────────
|
||
|
||
@router.get("/queue", response_model=list[QueueTaskResponse])
|
||
async def get_queue(
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> list[QueueTaskResponse]:
|
||
"""获取当前门店的待执行队列。"""
|
||
tasks = task_queue.list_pending(user.site_id)
|
||
return [
|
||
QueueTaskResponse(
|
||
id=t.id,
|
||
site_id=t.site_id,
|
||
config=t.config,
|
||
status=t.status,
|
||
position=t.position,
|
||
created_at=t.created_at,
|
||
started_at=t.started_at,
|
||
finished_at=t.finished_at,
|
||
exit_code=t.exit_code,
|
||
error_message=t.error_message,
|
||
)
|
||
for t in tasks
|
||
]
|
||
|
||
|
||
# ── POST /api/execution/queue — 添加到队列 ───────────────────
|
||
|
||
@router.post("/queue", response_model=QueueTaskResponse, status_code=status.HTTP_201_CREATED)
|
||
async def enqueue_task(
|
||
config: TaskConfigSchema,
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> QueueTaskResponse:
|
||
"""将任务配置添加到执行队列。"""
|
||
config = config.model_copy(update={"store_id": user.site_id})
|
||
task_id = task_queue.enqueue(config, user.site_id)
|
||
|
||
# 查询刚创建的任务返回完整信息
|
||
conn = get_connection()
|
||
try:
|
||
with conn.cursor() as cur:
|
||
cur.execute(
|
||
"""
|
||
SELECT id, site_id, config, status, position,
|
||
created_at, started_at, finished_at,
|
||
exit_code, error_message
|
||
FROM task_queue WHERE id = %s
|
||
""",
|
||
(task_id,),
|
||
)
|
||
row = cur.fetchone()
|
||
conn.commit()
|
||
finally:
|
||
conn.close()
|
||
|
||
if row is None:
|
||
raise HTTPException(status_code=500, detail="入队后查询失败")
|
||
|
||
config_data = row[2] if isinstance(row[2], dict) else json.loads(row[2])
|
||
return QueueTaskResponse(
|
||
id=str(row[0]),
|
||
site_id=row[1],
|
||
config=config_data,
|
||
status=row[3],
|
||
position=row[4],
|
||
created_at=row[5],
|
||
started_at=row[6],
|
||
finished_at=row[7],
|
||
exit_code=row[8],
|
||
error_message=row[9],
|
||
)
|
||
|
||
|
||
# ── PUT /api/execution/queue/reorder — 重排队列 ──────────────
|
||
|
||
@router.put("/queue/reorder")
|
||
async def reorder_queue(
|
||
body: ReorderRequest,
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> dict:
|
||
"""调整队列中任务的执行顺序。"""
|
||
task_queue.reorder(body.task_id, body.new_position, user.site_id)
|
||
return {"message": "队列已重排"}
|
||
|
||
|
||
# ── DELETE /api/execution/queue/{id} — 删除队列任务 ──────────
|
||
|
||
@router.delete("/queue/{task_id}")
|
||
async def delete_queue_task(
|
||
task_id: str,
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> dict:
|
||
"""从队列中删除待执行任务。仅允许删除 pending 状态的任务。"""
|
||
deleted = task_queue.delete(task_id, user.site_id)
|
||
if not deleted:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_409_CONFLICT,
|
||
detail="任务不存在或非待执行状态，无法删除",
|
||
)
|
||
return {"message": "任务已从队列中删除"}
|
||
|
||
|
||
# ── POST /api/execution/{id}/cancel — 取消执行 ──────────────
|
||
|
||
@router.post("/{execution_id}/cancel")
|
||
async def cancel_execution(
|
||
execution_id: str,
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> dict:
|
||
"""取消正在执行的任务。"""
|
||
cancelled = await task_executor.cancel(execution_id)
|
||
if not cancelled:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_404_NOT_FOUND,
|
||
detail="执行任务不存在或已完成",
|
||
)
|
||
return {"message": "已发送取消信号"}
|
||
|
||
|
||
# ── GET /api/execution/history — 执行历史 ────────────────────
|
||
|
||
@router.get("/history", response_model=list[ExecutionHistoryItem])
|
||
async def get_execution_history(
|
||
limit: int = Query(default=50, ge=1, le=200),
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> list[ExecutionHistoryItem]:
|
||
"""获取执行历史记录，按 started_at 降序排列。"""
|
||
conn = get_connection()
|
||
try:
|
||
with conn.cursor() as cur:
|
||
cur.execute(
|
||
"""
|
||
SELECT id, site_id, task_codes, status, started_at,
|
||
finished_at, exit_code, duration_ms, command, summary,
|
||
schedule_id
|
||
FROM task_execution_log
|
||
WHERE site_id = %s
|
||
ORDER BY started_at DESC
|
||
LIMIT %s
|
||
""",
|
||
(user.site_id, limit),
|
||
)
|
||
rows = cur.fetchall()
|
||
conn.commit()
|
||
finally:
|
||
conn.close()
|
||
|
||
return [
|
||
ExecutionHistoryItem(
|
||
id=str(row[0]),
|
||
site_id=row[1],
|
||
task_codes=row[2] or [],
|
||
status=row[3],
|
||
started_at=row[4],
|
||
finished_at=row[5],
|
||
exit_code=row[6],
|
||
duration_ms=row[7],
|
||
command=row[8],
|
||
summary=row[9],
|
||
schedule_id=str(row[10]) if row[10] else None,
|
||
)
|
||
for row in rows
|
||
]
|
||
|
||
|
||
# ── GET /api/execution/{id}/logs — 获取历史日志 ──────────────
|
||
|
||
@router.get("/{execution_id}/logs", response_model=ExecutionLogsResponse)
|
||
async def get_execution_logs(
|
||
execution_id: str,
|
||
user: CurrentUser = Depends(get_current_user),
|
||
) -> ExecutionLogsResponse:
|
||
"""获取指定执行的完整日志。
|
||
|
||
优先从内存缓冲区读取（执行中），否则从数据库读取（已完成）。
|
||
"""
|
||
# 先尝试内存缓冲区（执行中的任务）
|
||
if task_executor.is_running(execution_id):
|
||
lines = task_executor.get_logs(execution_id)
|
||
return ExecutionLogsResponse(
|
||
execution_id=execution_id,
|
||
output_log="\n".join(lines) if lines else None,
|
||
)
|
||
|
||
# 从数据库读取已完成任务的日志
|
||
conn = get_connection()
|
||
try:
|
||
with conn.cursor() as cur:
|
||
cur.execute(
|
||
"""
|
||
SELECT output_log, error_log
|
||
FROM task_execution_log
|
||
WHERE id = %s AND site_id = %s
|
||
""",
|
||
(execution_id, user.site_id),
|
||
)
|
||
row = cur.fetchone()
|
||
conn.commit()
|
||
finally:
|
||
conn.close()
|
||
|
||
if row is None:
|
||
raise HTTPException(
|
||
status_code=status.HTTP_404_NOT_FOUND,
|
||
detail="执行记录不存在",
|
||
)
|
||
|
||
return ExecutionLogsResponse(
|
||
execution_id=execution_id,
|
||
output_log=row[0],
|
||
error_log=row[1],
|
||
)
|