mirror of
https://github.com/fastapi-practices/fastapi_best_architecture.git
synced 2026-03-13 09:31:31 +08:00
52 lines
2.0 KiB
Python
52 lines
2.0 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
from typing import Annotated
|
|
|
|
from fastapi import APIRouter, Depends, Path
|
|
from starlette.concurrency import run_in_threadpool
|
|
|
|
from backend.app.task import celery_app
|
|
from backend.app.task.schema.control import TaskRegisteredDetail
|
|
from backend.common.exception import errors
|
|
from backend.common.response.response_schema import ResponseModel, ResponseSchemaModel, response_base
|
|
from backend.common.security.jwt import DependsJwtAuth
|
|
from backend.common.security.permission import RequestPermission
|
|
from backend.common.security.rbac import DependsRBAC
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get('/registered', summary='获取已注册的任务', dependencies=[DependsJwtAuth])
|
|
async def get_task_registered() -> ResponseSchemaModel[list[TaskRegisteredDetail]]:
|
|
inspector = celery_app.control.inspect(timeout=0.5)
|
|
registered = await run_in_threadpool(inspector.registered)
|
|
if not registered:
|
|
raise errors.ServerError(msg='Celery Worker 暂不可用，请稍后重试')
|
|
task_registered = []
|
|
celery_app_tasks = celery_app.tasks
|
|
for _, tasks in registered.items():
|
|
for task in tasks:
|
|
task_ins = celery_app_tasks.get(task)
|
|
if task_ins:
|
|
task_doc = task_ins.__doc__
|
|
task_registered.append({'name': task_doc or task_ins, 'task': task_ins})
|
|
else:
|
|
task_registered.append({'name': task, 'task': task})
|
|
return response_base.success(data=task_registered)
|
|
|
|
|
|
@router.delete(
|
|
'/{task_id}/cancel',
|
|
summary='撤销任务',
|
|
dependencies=[
|
|
Depends(RequestPermission('sys:task:revoke')),
|
|
DependsRBAC,
|
|
],
|
|
)
|
|
async def revoke_task(task_id: Annotated[str, Path(description='任务 UUID')]) -> ResponseModel:
|
|
workers = await run_in_threadpool(celery_app.control.ping, timeout=0.5)
|
|
if not workers:
|
|
raise errors.ServerError(msg='Celery Worker 暂不可用，请稍后重试')
|
|
celery_app.control.revoke(task_id)
|
|
return response_base.success()
|