重试丢失或失败的任务(Celery,Django和RabbitMQ)

时间:2011-03-17 08:32:09

标签: rabbitmq celery django-celery

有没有办法确定是否有任何任务丢失并重试?


我认为丢失的原因可能是调度程序错误或工作线程崩溃。

我打算重试它们,但我不确定如何确定哪些任务需要退役?

如何自动完成此过程?我可以使用自己的自定义调度程序来创建新任务吗?

编辑:我从文档中发现RabbitMQ从未松散任务,但是当工作线程在任务执行过程中崩溃时会发生什么?

1 个答案:

答案 0 :(得分:29)

您需要的是设置

CELERY_ACKS_LATE = True

延迟确认意味着在执行任务后将确认任务消息, 不仅仅是之前,这是默认行为。 通过这种方式,如果工作人员崩溃,那么MQ仍然会收到消息。

显然总崩溃(Rabbit + worker)同时无法恢复任务,除非您实现了任务启动和任务结束的日志记录。 我个人每次在任务开始时用mongodb写一行,在任务完成时用另一个写入(独立形成结果),这样我就可以通过分析mongo日志知道哪个任务被中断了。

您可以通过覆盖芹菜基础任务类的方法__call__after_return来轻松完成。

下面你看到我的一段代码使用了一个taskLogger类作为上下文管理器(带有入口和出口点)。 taskLogger类只是在mongodb实例中写入包含任务信息的行。

def __call__(self, *args, **kwargs):
    """In celery task this function call the run method, here you can
    set some environment variable before the run of the task"""

    #Inizialize context managers    

    self.taskLogger = TaskLogger(args, kwargs)
    self.taskLogger.__enter__()

    return self.run(*args, **kwargs)

def after_return(self, status, retval, task_id, args, kwargs, einfo):
    #exit point for context managers
    self.taskLogger.__exit__(status, retval, task_id, args, kwargs, einfo)

我希望这可以帮助