diff --git a/swh/scheduler/celery_backend/config.py b/swh/scheduler/celery_backend/config.py --- a/swh/scheduler/celery_backend/config.py +++ b/swh/scheduler/celery_backend/config.py @@ -8,7 +8,7 @@ import urllib.parse from celery import Celery -from celery.signals import setup_logging +from celery.signals import setup_logging, celeryd_after_setup from celery.utils.log import ColorFormatter from celery.worker.control import Panel @@ -76,6 +76,13 @@ celery_task_logger.setLevel(loglevel) +@celeryd_after_setup.connect +def setup_queues_and_tasks(sender, instance, **kwargs): + for task_name in instance.app.tasks: + if task_name.startswith('swh.'): + instance.app.amqp.queues.select_add(task_name) + + @Panel.register def monotonic(state): """Get the current value for the monotonic clock"""