如何告诉芹菜工人停止接受任务?如何检查是否有任何芹菜工人任务正在运行

Question

场景：

• 在由 Python/Flask Web 应用程序和使用 Celery 的后台任务组成的服务器上运行的系统
• Web应用程序和芹菜工人都作为新贵工作运行（Nginx背后的Web应用程序）

• 部署到生产环境是使用脚本完成的，该脚本：

  • 停止新贵作业
  • 将代码推送到服务器
  • 运行任何数据库迁移
  • 启动新贵作业

如何增强部署脚本以使其执行以下操作？

• 告诉芹菜工人停止接受任务
• 等到任何当前正在运行的芹菜任务完成
• 停止新贵作业
• 将代码推送到服务器
• 运行任何数据库迁移
• 启动新贵作业

Answer 1

以下脚本作为部署的一部分运行解决了问题：

import time
from celery.app.control import Control
from myapp.tasks import celery # my application's Celery app
if __name__ == "__main__":
    control = Control(celery)
    control.cancel_consumer("celery") # queue name, must probably be specified once per queue, but my app uses a single queue
    inspect = control.inspect()
    while True:
        active = inspect.active()
        running_jobs = []
        for key, value in active.items():
            running_jobs.extend(value)
        if len(running_jobs) > 0:
            print("{} jobs running: {}".format(len(running_jobs), ", ".join(job["name"] for job in running_jobs)))
            time.sleep(10)
        else:
            print("No running jobs")
            break