47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
|
"""任务队列 API"""
|
||
|
|
|
||
|
|
from fastapi import APIRouter, HTTPException
|
||
|
|
|
||
|
|
from app import database as db
|
||
|
|
from app.models import TaskCreate, TaskOut
|
||
|
|
|
||
|
|
router = APIRouter()
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("", response_model=list[TaskOut])
|
||
|
|
async def list_tasks(status: str | None = None, limit: int = 50):
|
||
|
|
return await db.list_tasks(status=status, limit=limit)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("", response_model=dict)
|
||
|
|
async def create_task(body: TaskCreate):
|
||
|
|
return await db.create_task(
|
||
|
|
task_type=body.task_type,
|
||
|
|
request_payload=body.request_payload,
|
||
|
|
plan_id=body.plan_id,
|
||
|
|
priority=body.priority,
|
||
|
|
max_retries=body.max_retries,
|
||
|
|
callback_url=body.callback_url,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.get("/{task_id}", response_model=TaskOut)
|
||
|
|
async def get_task(task_id: str):
|
||
|
|
d = await db.get_db()
|
||
|
|
cur = await d.execute("SELECT * FROM tasks WHERE id=?", (task_id,))
|
||
|
|
row = await cur.fetchone()
|
||
|
|
if not row:
|
||
|
|
raise HTTPException(404, "Task not found")
|
||
|
|
t = db.row_to_dict(row)
|
||
|
|
t["request_payload"] = db._parse_json(t["request_payload"], {})
|
||
|
|
t["response_payload"] = db._parse_json(t.get("response_payload"))
|
||
|
|
return t
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/{task_id}/cancel")
|
||
|
|
async def cancel_task(task_id: str):
|
||
|
|
ok = await db.update_task(task_id, status="cancelled")
|
||
|
|
if not ok:
|
||
|
|
raise HTTPException(404, "Task not found")
|
||
|
|
return {"ok": True}
|