Python rabbitmq的使用（六）

前面的例子都有个共同点，就是发送端发送消息出去后没有结果返回。若是只是单纯发送消息，固然没有问题了，可是在实际中，经常会须要接收端将收到的消息进行处理以后，返回给发送端。

处理方法描述：发送端在发送信息前，产生一个接收消息的临时队列，该队列用来接收返回的结果。其实在这里接收端、发送端的概念已经比较模糊了，由于发送端也一样要接收消息，接收端一样也要发送消息，因此这里笔者使用另外的示例来演示这一过程。

示例内容：假设有一个控制中心和一个计算节点，控制中心会将一个天然数N发送给计算节点，计算节点将N值加1后，返回给控制中心。这里用center.py模拟控制中心，compute.py模拟计算节点。

compute.py代码分析

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33

#!/usr/bin/env python #coding=utf8 import pika   #链接rabbitmq服务器 connection = pika.BlockingConnection(pika.ConnectionParameters(         host='localhost')) channel = connection.channel()   #定义队列 channel.queue_declare(queue='compute_queue') print ' [*] Waiting for n'   #将n值加1 def increase(n):     return n + 1   #定义接收到消息的处理方法 def request(ch, method, properties, body):     print " [.] increase(%s)"  % (body,)       response = increase(int(body))       #将计算结果发送回控制中心     ch.basic_publish(exchange='',                      routing_key=properties.reply_to,                      body=str(response))     ch.basic_ack(delivery_tag = method.delivery_tag)   channel.basic_qos(prefetch_count=1) channel.basic_consume(request, queue='compute_queue')   channel.start_consuming()

计算节点的代码比较简单，值得一提的是，原来的接收方法都是直接将消息打印出来，这边进行了加一的计算，并将结果发送回控制中心。

center.py代码分析

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43

#!/usr/bin/env python #coding=utf8 import pika   class Center(object):     def __init__(self):         self.connection = pika.BlockingConnection(pika.ConnectionParameters(                 host='localhost'))           self.channel = self.connection.channel()                   #定义接收返回消息的队列         result = self.channel.queue_declare(exclusive=True)         self.callback_queue = result.method.queue           self.channel.basic_consume(self.on_response,                                    no_ack=True,                                    queue=self.callback_queue)       #定义接收到返回消息的处理方法     def on_response(self, ch, method, props, body):         self.response = body                 def request(self, n):         self.response = None         #发送计算请求，并声明返回队列         self.channel.basic_publish(exchange='',                                    routing_key='compute_queue',                                    properties=pika.BasicProperties(                                          reply_to = self.callback_queue,                                          ),                                    body=str(n))         #接收返回的数据         while self.response is None:             self.connection.process_data_events()         return int(self.response)   center = Center()   print " [x] Requesting increase(30)" response = center.request(30) print " [.] Got %r" % (response,)

上例代码定义了接收返回数据的队列和处理方法，而且在发送请求的时候将该队列赋值给reply_to，在计算节点代码中就是经过这个参数来获取返回队列的。

打开两个终端，一个运行代码python compute.py，另一个终端运行center.py，若是执行成功，应该就能看到效果了。

笔者在测试的时候，出了些小问题，就是在center.py发送消息时没有指明返回队列，结果compute.py那边在计算完结果要发回数据时报错，提示routing_key不存在，再次运行也报错。用rabbitmqctl list_queues查看队列，发现compute_queue队列有1条数据，每次从新运行compute.py的时候，都会从新处理这条数据。后来使用/etc/init.d/rabbitmq-server restart从新启动下rabbitmq就ok了。