繁体   English   中英

如何按名称从芹菜队列中删除任务

[英]How to remove a task by name from a Celery Queue

我正在尝试找到一种方法来从 Celery 队列中删除所有具有特定给定名称的当前排队任务。 从官方文档中,我知道我可以检查工人并通过查找他们的姓名然后获取他们的 ID 来撤销任务,例如:

def drop_celery_task(options):
    def _get_tasks_id(workers: list, tasks_ids: list, task_name: str):
        """
        Get task ids with the given name included inside the given `workers` tasks.
        {'worker1.example.com': [
             {'name': 'tasks.sleeptask', 'id': '32666e9b-809c-41fa-8e93-5ae0c80afbbf',
              'args': '(8,)', 'kwargs': '{}'}]
        }
        """

        for worker in workers:
            if not workers[worker]:
                continue
            for _task in workers[worker]:
                if _task["name"].split(".")[-1] == task_name:
                    tasks_ids.append(_task["id"])

    task_name = options.drop_celery_task["name"]
    i = Inspect(app=celery_app)  # Inspect all nodes.
    registered = i.registered()
    if not registered:
        raise Exception("No registered tasks found")

    if not any(task_name == worker.split(".")[-1] for worker in chain(*list(registered.values()))):
        raise Exception(f"Task not registered: {task_name}")

    tasks_ids = []
    _get_tasks_id(i.active(), tasks_ids, task_name)
    _get_tasks_id(i.scheduled(), tasks_ids, task_name)
    _get_tasks_id(i.reserved(), tasks_ids, task_name)

    if tasks_ids:
        for task_id in tasks_ids:
            Control(app=celery_app).revoke(task_id)
    else:
        logging.info(f"No active/scheduled/registered task found with the name {task_name}")

但是这段代码只撤销 celery 工作人员获取或预获取的任务,而不是仍在队列中的任务(使用 Redis 作为后端)。 关于如何使用 celery 命令删除 Redis 中的那些,或阻止工作人员接受具有给定名称的任务的任何建议?

我最终在 Redis 中使用我想要的名称识别任务的 ID(使用 redis 客户端,而不是 celery 命令),然后通过Control(app=celery_app).revoke(task_id)命令撤销这些 ID。 在 Redis 中,队列是具有队列名称的键下的列表对象。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM