从超过 max_retries 失败的任务中恢复

Question

我尝试异步使用 Web 服务，因为它最多需要 45 秒才能返回。不幸的是，这个 Web 服务也有些不可靠，可能会引发错误。我已经设置了 django-celery 并执行了我的任务，它可以正常工作，直到任务失败超过 max_retries 为止。

到目前为止，这是我所得到的：

@task(default_retry_delay=5, max_retries=10)
def request(xml):
    try:
        server = Client('https://www.whatever.net/RealTimeService.asmx?wsdl')
        xml = server.service.RunRealTimeXML(
            username=settings.WS_USERNAME,
            password=settings.WS_PASSWORD,
            xml=xml
        )
    except Exception, e:
        result = Result(celery_id=request.request.id, details=e.reason, status="i")
        result.save()
        try:
            return request.retry(exc=e)
        except MaxRetriesExceededError, e:
            result = Result(celery_id=request.request.id, details="Max Retries Exceeded", status="f")
            result.save()
            raise
    result = Result(celery_id=request.request.id, details=xml, status="s")
    result.save()
    return result

不幸的是，Retry() 没有抛出 MaxRetriesExceededError ，所以我不确定如何处理此任务的失败。 Django 已经将 HTML 返回给客户端，并且我正在通过 AJAX 检查 Result 的内容，这永远不会达到完全失败 f 状态。

所以问题是：当 Celery 任务超过 max_retries 时，如何更新数据库？

Answer 1

问题是，当 celery 达到重试限制时，它会尝试重新引发您传入的异常。执行此重新提升的代码位于： https://github.com/celery/celery/blob/v3.1.20/celery/app/task.py#L673-L681

解决这个问题的最简单方法就是根本没有芹菜管理你的异常：

@task(max_retries=10)
def mytask():
    try:
        do_the_thing()
    except Exception as e:
        try:
            mytask.retry()
        except MaxRetriesExceededError:
            do_something_to_handle_the_error()
            logger.exception(e)

Answer 2

您可以重写 celery 任务类的 after_return 方法，该方法在任务执行后调用，无论 ret 状态如何（SUCCESS、FAILED、RETRY）

class MyTask(celery.task.Task)

    def run(self, xml, **kwargs)
        #Your stuffs here

    def after_return(self, status, retval, task_id, args, kwargs, einfo=None):
        if self.max_retries == int(kwargs['task_retries']):
            #If max retries are equals to task retries do something
        if status == "FAILURE":
            #You can do also something if the tasks fail instead of check the retries

http://readthedocs.org/docs/celery/en/latest/reference/celery.task.base.html#celery.task.base.BaseTask.after_return

< a href="http://celery.readthedocs.org/en/latest/reference/celery.app.task.html?highlight=after_return#celery.app.task.Task.after_return" rel="noreferrer">http://celery.readthedocs.org/en/latest/reference/celery.app.task.html?highlight=after_return#celery.app.task.Task.after_return

Answer 3

对于 Celery 2.3.2 版本，这种方法对我来说效果很好：

class MyTask(celery.task.Task):
    abstract = True

    def after_return(self, status, retval, task_id, args, kwargs, einfo):
        if self.max_retries == self.request.retries:
            #If max retries is equal to task retries do something

@task(base=MyTask, default_retry_delay=5, max_retries=10)
def request(xml):
    #Your stuff here

Answer 4

我现在只考虑这个，这样可以省去子类化 Task 的工作，并且很容易理解。

# auto-retry with delay as defined below. After that, hook is disabled.
@celery.shared_task(bind=True, max_retries=5, default_retry_delay=300)
def post_data(self, hook_object_id, url, event, payload):
    headers = {'Content-type': 'application/json'}
    try:
        r = requests.post(url, data=payload, headers=headers)
        r.raise_for_status()
    except requests.exceptions.RequestException as e:
        if self.request.retries >= self.max_retries:
            log.warning("Auto-deactivating webhook %s for event %s", hook_object_id, event)
            Webhook.objects.filter(object_id=hook_object_id).update(active=False)
            return False
        raise self.retry(exc=e)
    return True