python - 从 celery 任务中获取 celery worker 的名字？

Question

我希望 celery 任务能够获取执行它的工作人员的姓名，以便进行日志记录。我需要在任务中处理这个问题，而不是直接查询代理。有没有办法做到这一点？如果重要的话，我正在将 celery 与 RabbitMQ 一起使用。

Answer 1

使用 celeryd_after_setup 信号像这样捕获 worker 名称:

from celery.signals import celeryd_after_setup

@celeryd_after_setup.connect
def capture_worker_name(sender, instance, **kwargs):
    os.environ["WORKER_NAME"] = '{0}'.format(sender)