Question

简单的问题，但谷歌或Pika开源代码没有帮助。有没有办法在Pika中查询当前队列大小（项目计数器）？

Answer 1

我知道这个问题有点陈旧，但这里有一个用pika做这个的例子。

关于AMQP和RabbitMQ，如果您已经声明了队列，则可以在passive flag开启的情况下重新声明队列，并保持所有其他队列参数相同。对此声明declare-ok的响应将包括队列中的消息数。

以下是pika 0.9.5的一个例子：

import pika

def on_callback(msg):
    print msg

params = pika.ConnectionParameters(
        host='localhost',
        port=5672,
        credentials=pika.credentials.PlainCredentials('guest', 'guest'),
    )

# Open a connection to RabbitMQ on localhost using all default parameters
connection = pika.BlockingConnection(parameters=params)

# Open the channel
channel = connection.channel()

# Declare the queue
channel.queue_declare(
        callback=on_callback,
        queue="test",
        durable=True,
        exclusive=False,
        auto_delete=False
    )

# ...

# Re-declare the queue with passive flag
res = channel.queue_declare(
        callback=on_callback,
        queue="test",
        durable=True,
        exclusive=False,
        auto_delete=False,
        passive=True
    )
print 'Messages in queue %d' % res.method.message_count

这将打印以下内容：

<Method(['frame_type=1', 'channel_number=1', "method=<Queue.DeclareOk(['queue=test', 'message_count=0', 'consumer_count=0'])>"])>
<Method(['frame_type=1', 'channel_number=1', "method=<Queue.DeclareOk(['queue=test', 'message_count=0', 'consumer_count=0'])>"])>
Messages in queue 0

您可以获得message_count成员发送的邮件数量。

Answer 2

以下是使用pika获取队列长度的方法（考虑到你在localhost上使用默认用户和密码）用队列名称替换q_name。

import pika
connection = pika.BlockingConnection()
channel = connection.channel()
q = channel.queue_declare(q_name)
q_len = q.method.message_count

Answer 3

你试过PyRabbit吗？它有一个get_queue_depth() method，听起来就像你要找的那样。

Answer 4

有两种方法可以在AMQP协议中获取队列大小。您可以使用Queue.Declare或Basic.Get。

如果您在使用Basic.Consume到达时消费消息，那么除非您断开连接（超时）并重新声明队列，否则您无法获取此信息，或者获取一条消息但不要获取消息。在较新版本的AMQP中，您可以主动重新排队消息。

至于Pika，我不知道具体细节，但AMQP的Python客户端一直是我的荆棘。通常，您需要使用monkeypatch类来获取所需的信息，或者允许队列使用者超时，以便您可以定期执行其他操作（如记录统计信息）或查找队列中有多少消息。

另一种方法是放弃，并使用Pipe类来运行sudo rabbitmqctl list_queues -p my_vhost。然后解析输出以查找所有队列的大小。如果你这样做，你需要配置/etc/sudoers不要求通常的sudo密码。

我祈祷有更多Pika经验的其他人通过指出你如何做我提到的所有事情来回答这个问题，在这种情况下我会下载Pika并踢掉轮胎。但是，如果这种情况没有发生，并且您在修改Pika代码方面遇到困难，那么请查看haigha。我发现他们的代码比其他Python AMQP客户端库要简单得多，因为它们更贴近AMQP协议。

获取Pika中的队列大小（AMQP Python）

4 个答案: