使用任务参数自定义 Celery

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

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

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

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

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 任务标头以将自身传递给工作进程,在工作进程中它忽略任务本身抛出的任何异常。


相关文章