2015-02-08 134 views
0

我希望我的python脚本能够在将任务发送至队列之前不断运行并检查队列大小是否为零。下面是一个片段:RabbitMQ - 如何获取队列大小

#!/usr/bin/env python 
import pika 
import sys 

while True: 

    # establish connection with RabbitMQ server 
    connection = pika.BlockingConnection(pika.ConnectionParameters('localhost')) 
    channel = connection.channel() 

    # create queue 
    # durable=True make sure RabbitMQ never lose our queue 
    rabbit_queue = channel.queue_declare(queue="task_queue", durable=True) 

    queue_size = rabbit_queue.method.message_count 
    if queue_size <= 0: 
     message = get_message() 
     channel.basic_publish(exchange='', 
           routing_key="task_queue", 
           body=message, 
           properties=pika.BasicProperties(
            delivery_mode=2, # make message persistent 
          )) 

     print " [x] Sent %r" % (message,) 
     time.sleep(3) 

    # close connection to flush network buffers AND ensure our message was actually delivered to RabbitMQ 
    connection.close() 

不断重新建立连接是否是一种好的做法?

+0

实在是太宽泛。最正确的答案:这取决于此脚本运行的频率,您的基础架构,典型的AMQP代理负载。 – pinepain 2015-02-08 16:06:28

+0

此脚本作为后台脚本运行。它nv停止(即:while循环)。我创造了大约6个队列。每个队列可能有大约1500个任务 – nuttynibbles 2015-02-08 16:32:40

+0

试试吧。如果你遇到性能问题,那么你一定会知道该怎么做。 – pinepain 2015-02-08 18:05:17

回答

0

我看不出有任何理由不打开/关闭的while循环外移动连接:

#!/usr/bin/env python 

import pika 
import sys 


# establish connection with RabbitMQ server 
connection = pika.BlockingConnection(pika.ConnectionParameters('localhost')) 

try: 
    channel = connection.channel() 

    # create queue 
    # durable=True make sure RabbitMQ never lose our queue 
    channel.queue_declare(queue="task_queue", durable=True) 

    while True: 
     rabbit_queue = channel.queue_declare(queue="task_queue", durable=True, passive=True) 
     queue_size = rabbit_queue.method.message_count 
     if queue_size <= 0: 
      message = get_message() 
      channel.basic_publish(exchange='', 
            routing_key="task_queue", 
            body=message, 
            properties=pika.BasicProperties(
            delivery_mode=2, # make message persistent 
           )) 

      print " [x] Sent %r" % (message,) 
      time.sleep(3) 
finally: 
    connection.close()