未来
2bd8b61e91
新功能: - 大纲/章节生成改为服务端后台任务，支持断线续传 - 后台任务队列排队执行，按用户排队（同用户串行不同用户并发） - 章节管理页面添加后台任务列表弹窗和进度面板 - 章节状态添加 pending（待处理）选项 - 集成json5容错解析器 + 上下文感知JSON修复 - SSE流式生成添加心跳保活，防止连接超时 - SSEPostClient添加credentials:include修复network error - 每章最大伏笔数从2调整为5 - 添加大纲读区伏笔的功能 Bug修复: - 修复AI生成JSON中未转义引号/中文标点/多对象属性值未合并 - 修复JSON非法转义字符清洗和中文引号处理 - 修复MCP插件TimeoutError/连接失败上下文清理 - MCP插件后台注册添加重试机制 - 续写模式添加缺失的mcp_references参数 - 修复Alembic迁移链分叉 - 使用torch CPU版本加速Docker构建
122 lines
4.1 KiB
Python
122 lines
4.1 KiB
Python
"""后台任务API - 查询状态、取消任务"""
|
|
from fastapi import APIRouter, Depends, HTTPException, Request
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
from typing import Optional
|
|
|
|
from app.database import get_db
|
|
from app.models.background_task import BackgroundTask
|
|
from app.services.background_task_service import background_task_service
|
|
from app.logger import get_logger
|
|
|
|
router = APIRouter(prefix="/tasks", tags=["后台任务"])
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
@router.get("/{task_id}", summary="获取任务状态")
|
|
async def get_task_status(
|
|
task_id: str,
|
|
request: Request,
|
|
db: AsyncSession = Depends(get_db)
|
|
):
|
|
"""获取后台任务的状态和进度"""
|
|
user_id = getattr(request.state, 'user_id', None)
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="未登录")
|
|
|
|
task = await background_task_service.get_task(task_id, user_id, db)
|
|
if not task:
|
|
raise HTTPException(status_code=404, detail="任务不存在")
|
|
|
|
return {
|
|
"id": task.id,
|
|
"task_type": task.task_type,
|
|
"project_id": task.project_id,
|
|
"status": task.status,
|
|
"progress": task.progress,
|
|
"status_message": task.status_message,
|
|
"progress_details": task.progress_details,
|
|
"error_message": task.error_message,
|
|
"task_result": task.task_result,
|
|
"retry_count": task.retry_count,
|
|
"cancel_requested": task.cancel_requested,
|
|
"created_at": task.created_at.isoformat() if task.created_at else None,
|
|
"started_at": task.started_at.isoformat() if task.started_at else None,
|
|
"completed_at": task.completed_at.isoformat() if task.completed_at else None,
|
|
"updated_at": task.updated_at.isoformat() if task.updated_at else None,
|
|
}
|
|
|
|
|
|
@router.get("", summary="获取任务列表")
|
|
async def get_tasks(
|
|
project_id: str,
|
|
request: Request,
|
|
task_type: Optional[str] = None,
|
|
limit: int = 20,
|
|
db: AsyncSession = Depends(get_db)
|
|
):
|
|
"""获取项目的后台任务列表"""
|
|
user_id = getattr(request.state, 'user_id', None)
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="未登录")
|
|
|
|
tasks = await background_task_service.get_project_tasks(
|
|
project_id, user_id, db, task_type=task_type, limit=limit
|
|
)
|
|
|
|
return {
|
|
"items": [
|
|
{
|
|
"id": t.id,
|
|
"task_type": t.task_type,
|
|
"status": t.status,
|
|
"progress": t.progress,
|
|
"status_message": t.status_message,
|
|
"progress_details": t.progress_details,
|
|
"error_message": t.error_message,
|
|
"created_at": t.created_at.isoformat() if t.created_at else None,
|
|
"completed_at": t.completed_at.isoformat() if t.completed_at else None,
|
|
}
|
|
for t in tasks
|
|
]
|
|
}
|
|
|
|
|
|
@router.post("/{task_id}/cancel", summary="取消任务")
|
|
async def cancel_task(
|
|
task_id: str,
|
|
request: Request,
|
|
db: AsyncSession = Depends(get_db)
|
|
):
|
|
"""请求取消后台任务"""
|
|
user_id = getattr(request.state, 'user_id', None)
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="未登录")
|
|
|
|
success = await background_task_service.cancel_task(task_id, user_id, db)
|
|
if not success:
|
|
raise HTTPException(status_code=400, detail="无法取消任务（不存在或已完成）")
|
|
|
|
return {"message": "任务已取消", "task_id": task_id}
|
|
|
|
|
|
@router.delete("/{task_id}", summary="删除任务记录")
|
|
async def delete_task(
|
|
task_id: str,
|
|
request: Request,
|
|
db: AsyncSession = Depends(get_db)
|
|
):
|
|
"""删除已完成/失败的任务记录"""
|
|
user_id = getattr(request.state, 'user_id', None)
|
|
if not user_id:
|
|
raise HTTPException(status_code=401, detail="未登录")
|
|
|
|
task = await background_task_service.get_task(task_id, user_id, db)
|
|
if not task:
|
|
raise HTTPException(status_code=404, detail="任务不存在")
|
|
|
|
if task.status in ("pending", "running"):
|
|
raise HTTPException(status_code=400, detail="无法删除进行中的任务，请先取消")
|
|
|
|
await db.delete(task)
|
|
await db.commit()
|
|
return {"message": "任务记录已删除"} |