如何在芹菜中将任务从一个队列移动到另一个队列

时间:2017-02-08 14:07:04

标签: python redis rabbitmq celery

我需要能够做两件事,这两件事取决于我能够将任务从一个队列移动到另一个队列。:

  1. 当任务失败一定数量的重试时,将其移至另一个队列
  2. 当任务满足某些条件时,将其移动到不同的队列(将由不同的工作人员处理)
  3. 请注意,我的意思是将内容放在另一个队列中,而不是来自主应用程序中的一个任务。

    这里引用了我能找到的唯一一段代码 - https://stackoverflow.com/a/27144119/112050 有人能指出正确的api吗?

1 个答案:

答案 0 :(得分:2)

移动任务的想法实际上归结为使用相同的参数运行相同的任务,但将其发送到不同的队列。

apply_async有一个queue参数

from celery.exceptions import MaxRetriesExceededError

@shared_task(default_retry_delay = 1 * 60, max_retries = 10)
def some_task(arg1, arg2):
    try:
        # task logic

        if some_condition:
            some_task.apply_async([arg1, arg2], queue='different_queue')
            return

    except MaxRetriesExceededError:
        some_task.apply_async([arg1, arg2], queue='different_queue')

    except Exception, exc:
        raise some_task.retry(exc=exc) 

如果您确实想在队列之间移动任务,假设您使用的是RabbitMQ,则可以使用Shovel Plugin。例如,将消息从队列q1移动到本地代理上的队列q2

rabbitmqctl set_parameter shovel my-shovel \
'{"src-uri": "amqp://", "src-queue": "q1", \
 "dest-uri": "amqp://", "dest-queue": "q2"}'