Redis
 sql >> Base de données >  >> NoSQL >> Redis

Django Celery obtient le nombre de tâches

Voici comment vous pouvez obtenir le nombre de messages dans une file d'attente à l'aide de céleri indépendant du courtier.

En utilisant connection_or_acquire , vous pouvez minimiser le nombre de connexions ouvertes à votre courtier en utilisant la mise en commun des connexions internes de celery.

celery = Celery(app)

with celery.connection_or_acquire() as conn:
    conn.default_channel.queue_declare(
        queue='my-queue', passive=True).message_count

Vous pouvez également étendre Celery pour fournir cette fonctionnalité :

from celery import Celery as _Celery


class Celery(_Celery)

    def get_message_count(self, queue):
        '''
        Raises: amqp.exceptions.NotFound: if queue does not exist
        '''
        with self.connection_or_acquire() as conn:
            return conn.default_channel.queue_declare(
                queue=queue, passive=True).message_count


celery = Celery(app)
num_messages = celery.get_message_count('my-queue')