异常教程

使用任务参数自定义 Celery

更新时间:

💡一则或许对你有用的小广告

欢迎加入小哈的星球 ，你将获得：专属的项目实战 / 1v1 提问 / Java 学习路线 / 学习打卡 / 每月赠书 / 社群讨论

  • 新项目:《从零手撸：仿小红书（微服务架构）》 正在持续爆肝中，基于 Spring Cloud Alibaba + Spring Boot 3.x + JDK 17...，点击查看项目介绍 ;
  • 《从零手撸：前后端分离博客项目（全栈开发）》 2 期已完结，演示链接： http://116.62.199.48/ ;

截止目前， 星球 内专栏累计输出 82w+ 字，讲解图 3441+ 张，还在持续爆肝中.. 后续还会上新更多项目，目标是将 Java 领域典型的项目都整一波，如秒杀系统, 在线商城, IM 即时通讯，权限管理，Spring Cloud Alibaba 微服务等等，已有 2800+ 小伙伴加入学习 ，欢迎点击围观

Celery 是一个很棒的 Python 分布式异步任务系统。它开箱即用，但有几次我需要对其进行自定义。具体来说，我希望能够根据新的 apply_sync 参数定义行为。此外，能够将状态传递给工作任务会很好。

首先，您可以子类化主 Celery 类以定义自定义 Task 类。 


 import socket

from celery import Celery, Task
from kombu.exceptions import InconsistencyError


class MyCelery(Celery):
""" Subclass of a Celery application class that uses a custom Task type """
task_cls = 'myapp.mymodule:MyTask'

在您的 Task 类中，您可以覆盖 apply_async （也从 delay 调用）以及 __call__ ，它环绕实际的任务主体。 


 import socket

from celery import Celery, Task
from kombu.exceptions import InconsistencyError


class MyCelery(Celery):
""" Subclass of a Celery application class that uses a custom Task type """
task_cls = 'myapp.mymodule:MyTask'

在此示例中，我向 apply_async 引入了一个可选的 safe 参数，它会捕获并忽略试图派生任务的特定异常。它还搭载 celery 任务标头以将自身传递给工作进程，在工作进程中它忽略任务本身抛出的任何异常。