RabbitMQ：使用python发布/订阅消息

        在上一章节中，我们学习了RabbitMQ：基础，运行和管理，在本节中我们来学习一下如何发布和订阅rabbitmq消息，我们使用python来开发应用程序。

        我们先来看一下发布/订阅rabbitmq消息的流程。先来看消息生产者Publisher如何发布消息流程：

引用

        1、打开RabbitMQ连接;
        2、创建Channel通道；
        3、声名一个exchange交换机；
        4、生成一条消息；
        5、发布消息；
        6、关闭Channel通道；
        7、关闭RabbitMQ连接。
       

        然后我们再来看一下Consumer订阅流程和消费消息：

引用

        1、打开RabbitMQ连接;
        2、创建Channel通道；
        3、声名一个exchange交换机；
        4、声名一个queue队列；
        5、将queue队列绑定到exchange交换机；
        6、消费消息；
        7、关闭Channel通道；
        8、关闭RabbitMQ连接。
       

        在写代码之前，我们需要先安装一下使用python链接RabbitMQ的依赖库——pika。我们使用easy_install来安装(easy_install的具体安装过程请参看Centos 6 上安装easy_instal)：

引用

        # ./easy_install pika       
       

        好了，准备工作已经好了，现在我们开始写代码：

引用

        # mkdir -p /data/rabbitmq-pika
        # cd /data/rabbitmq-pika
        # mkdir c2 
        # cd c2     
       

        在本节中，我们主要来开发一个带Publisher confirms的消息发布和订阅程序。我们先来开发消息生产者hello_world_producer_with_comfirms.py：

引用

        # touch hello_world_producer_with_comfirms.py
        # chmod +x hello_world_producer_with_comfirms.py
        # vi hello_world_producer_with_comfirms.py
       

        hello_world_producer_with_comfirms.py代码如下：
       

#coding=utf-8

import pika,sys
from pika import spec

#在"/"虚拟主机vhost上通过用户guest建立channel通道
user_name = 'guest'
user_passwd = 'guest'
target_host = 'localhost'
vhost = '/'
cred = pika.PlainCredentials(user_name,user_passwd)
conn_params = pika.ConnectionParameters(target_host,
                                        virtual_host = vhost,
                                        credentials = cred)
conn_broker = pika.BlockingConnection(conn_params)
channel = conn_broker.channel()

#定义消息发布后publisher接受到的确认信息处理函数
def confirm_handler(frame):
    if type(frame.method) == spec.Confirm.SelectOk:
        """生产者创建的channel处于‘publisher comfirms’模式"""
        print 'Channel in "confirm" mode!'
    elif type(frame.method) == spec.Basic.Nack:
        """生产者接受到消息发送失败并且消息丢失的消息"""
        print 'Message lost!'
    elif type(frame.method) == spec.Basic.ack:
        if frame.method.delivery_tag in msg_ids:
            """生产者接受到成功发布的消息"""
            print 'Confirm received!'
            msg_ids.remove(frame.method.delivery_tag)

#将生产者创建的channel处于"publisher confirms"模式
channel.confirm_delivery(callback = confirm_handler)

#创建一个direct类型的、持久化的、没有consumer时队列是否自动删除的exchage交换机
channel.exchange_declare(exchange = 'hello-exch',
                              type = 'direct',
                              passive = False,
                              durable = True,
                              auto_delete = False)
#使用接收到的信息创建消息
#使用接收到的信息创建消息
msg = sys.argv[1]
msg_props = pika.BasicProperties()
msg_props.content_type = 'text/plain'
#持久化消息
msg_props.delivery_mode = 2
msg_ids = []

print 'ready to publish...'
#发布消息
channel.basic_publish(body = msg,
                      exchange = 'hello-exch',
                      properties = msg_props,
                      routing_key = 'hala')
print 'published!'
msg_ids.append(len(msg_ids) + 1)
print len(msg_ids)
channel.close()
conn_broker.close()
       

       接下来，我们先来开发一个带消息确认信息的消费者hello_world_consumer_with_ack.py：

引用

        # hello_world_consumer_with_ack.py
        # chmod +x hello_world_consumer_with_ack.py
        # vi hello_world_consumer_with_ack.py
       

        hello_world_consumer_with_ack.py代码如下：

#coding=utf-8

import pika

#在"/"虚拟主机vhost上通过用户guest建立channel通道
user_name = 'guest'
user_passwd = 'guest'
target_host = 'localhost'
vhost = '/'
cred = pika.PlainCredentials(user_name,user_passwd)
conn_params = pika.ConnectionParameters(target_host,
                                        virtual_host = vhost,
                                        credentials = cred)
conn_broker = pika.BlockingConnection(conn_params)
conn_channel = conn_broker.channel()

#创建一个direct类型的、持久化的、没有consumer时，队列是否自动删除exchage交换机
conn_channel.exchange_declare(exchange = 'hello-exch',
                              type = 'direct',
                              passive = False,
                              durable = True,
                              auto_delete = False)

#创建一个持久化的、没有consumer时队列是否自动删除的名为“hell-queue”
conn_channel.queue_declare(queue = 'hello-queue',
                           durable = True,
                           auto_delete = False)

#将“hello-queue”队列通过routing_key绑定到“hello-exch”交换机
conn_channel.queue_bind(queue = 'hello-queue',
                   exchange = 'hello-exch',
                   routing_key = 'hala')

#定义一个消息确认函数，消费者成功处理完消息后会给队列发送一个确认信息，然后该消息会被删除
def ack_info_handler(channel,method,header,body):
    """ack_info_handler """    
    print 'ack_info_handler() called!'
    if body == 'quit':
        channel.basic_cancel(consumer_tag = 'hello-hala')
        channel.stop_sonsuming()
    else:
        print body
        channel.basic_ack(delivery_tag = method.delivery_tag)

conn_channel.basic_consume(ack_info_handler,
                           queue = 'hello-queue',
                           no_ack = False,
                           consumer_tag = 'hello-hala')

print 'ready to consume msg...'
conn_channel.start_consuming()
        

       

        测试：

引用

        # /opt/mq/rabbitmq/sbin/rabbitmq-server start &
        # python ./hello_world_consumer_with_ack.py
        # python ./hello_world_producer_with_comfirms.py 'hello-world'
       

        对于上面的hello_world_producer_with_comfirms.py，本人调试了很长时间，期间总是在：

引用

        #将生产者创建的channel处于"publisher confirms"模式
        channel.confirm_delivery(callback = confirm_handler)
       

        报错，最后在重新下载pika-0.9.13.tar.gz并安装后才成功执行上述代码：

引用

       # wget https://pypi.python.org/packages/source/p/pika/pika-0.9.13.tar.gz --no-check-certificate
        # chmod +x pika-0.9.13.tar.gz
        # tar xzvf pika-0.9.13.tar.gz
        # cd pika-0.9.13
        # python setup.py install
       

评论

1 楼 fbwfbi 2014-03-14  

发现使用pika-0.9.13的版本依然出错：
Traceback (most recent call last):
  File "hello_world_producer_with_comfirms.py", line 34, in <module>
    channel.confirm_delivery(callback = confirm_handler)
TypeError: confirm_delivery() got an unexpected keyword argument 'callback'

将confirm_delivery(callback = confirm_handler)去掉里面的callback参数即可