Redis
 sql >> Base de Dados >  >> NoSQL >> Redis

Django Celery obtém contagem de tarefas


Aqui está como você pode obter o número de mensagens em uma fila usando o aipo que é independente do agente.

Usando connection_or_acquire , você pode minimizar o número de conexões abertas com seu agente utilizando o pool de conexões internas do aipo.
celery = Celery(app)

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

Você também pode estender o Celery para fornecer esta funcionalidade:
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')