mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 21:15:13 +00:00
Wu Clan
5762834744
* Update the ruff rules and format the code * Update the per-file-ignores * Update the ci * Update rules * Fix codes * Fix pagination * Update rules
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import asyncio
|
|
|
|
from typing import Any
|
|
|
|
from celery import Task
|
|
from sqlalchemy.exc import SQLAlchemyError
|
|
|
|
from backend.common.socketio.actions import task_notification
|
|
from backend.core.conf import settings
|
|
|
|
|
|
class TaskBase(Task):
|
|
"""Celery 任务基类"""
|
|
|
|
autoretry_for = (SQLAlchemyError,)
|
|
max_retries = settings.CELERY_TASK_MAX_RETRIES
|
|
|
|
async def before_start(self, task_id: str, args, kwargs) -> None: # noqa: ANN001
|
|
"""
|
|
任务开始前执行钩子
|
|
|
|
:param task_id: 任务 ID
|
|
:return:
|
|
"""
|
|
await task_notification(msg=f'任务 {task_id} 开始执行')
|
|
|
|
async def on_success(self, retval: Any, task_id: str, args, kwargs) -> None: # noqa: ANN001
|
|
"""
|
|
任务成功后执行钩子
|
|
|
|
:param retval: 任务返回值
|
|
:param task_id: 任务 ID
|
|
:return:
|
|
"""
|
|
await task_notification(msg=f'任务 {task_id} 执行成功')
|
|
|
|
def on_failure(self, exc: Exception, task_id: str, args, kwargs, einfo) -> None: # noqa: ANN001
|
|
"""
|
|
任务失败后执行钩子
|
|
|
|
:param exc: 异常对象
|
|
:param task_id: 任务 ID
|
|
:param einfo: 异常信息
|
|
:return:
|
|
"""
|
|
asyncio.create_task(task_notification(msg=f'任务 {task_id} 执行失败'))
|