Question

我想在Celery + RabbitMQ代理上使用清理过程实现任务取消。如何在Celery工作者中获得当前任务的“REVOKED”状态？

# tasks.py -- celery worker
from celery import Celery
app = Celery('tasks', broker='amqp://guest@localhost//')

@app.task
def add(x, y):
  for i in range(0, 10):
    time.sleep(1)
    # I want check here for cleanup.
  return x + y

# caller.py
from tasks import add
result = add.delay(4, 4)
result.revoke()

Celery支持Abortable tasks，但它只适用于数据库后端。

Python 3.4.1 / Celery 3.1.17 / RabbitMQ 3.4.4

Answer 1

看看scheduled_tasks，如果您的任务计划运行，您可以询问芹菜。

EX：

    import celery
    celery_inspect = celery.current_app.control.inspect()
    celery_inspect.registered_tasks()

此方法返回一个包含所有芹菜计划任务的dict。

Answer 2

Felippe Da Motta Raposo的建议适用于我的自定义任务：

from celery import Task
from celery.task.control import inspect

WORKER_NAME = "celery@server"
inspector = inspect([WORKER_NAME])

class CustomTask(Task):
    def _is_revoked(self):
        revoked_list = inspector.revoked()
        return (revoked_list and self.task_id in revoked_list[WORKER_ADDRESS]

    def run(self, *args, **kwargs):
        self.task_id = self.request.id

如何在Celery工作者中获取当前任务的撤销标志？

2 个答案: