RabbitMQ学习小结（六）—— RPC[Python]

1. 简介

远程过程调用在很多地方都是经常用到，虽然也有诸多的缺点，但是应用还是较为广泛，对于RPC的概念就不做解释了，本节跟着参考资料，实现基于RabbitMQ的RPC如何使用。
接下来我们创建一个模拟RPC服务来返回斐波那契数列。所以需要一个客户端和一个服务端程序。

2. RPC[RabbitMQ]

对于此次操作我们可以理解为客户端对服务端做黑盒操作，客户端不知道服务端的处理逻辑是什么，只知道需要给服务端哪些信息，而服务端从RabbitMQ中取回消息处理完成之后，按照客户端要求将消息发送到指定队列。客户端取到消息后就可以自行处理。基本流程如下图：



在上面的描述中，我们可以读取出几个信息，就是客户端要给服务端指定一些信息，而这些信息包含哪些呢？从上图我们可以看到一个reply_to和correlation_id，这两个属性是AMQP中消息的属性，reply_to是我们上面提到的服务端处理完成之后将结果发送给指定的队列；而correlation_id是将RPC的请求和响应关联起来。
AMQP协议给消息预定义了一系列的14个属性。大多数属性很少会用到，除了以下几个：
delivery_mode（投递模式）：将消息标记为持久的（值为2）或暂存的（除了2之外的其他任何值）。第二篇教程里接触过这个属性，记得吧？

content_type（内容类型）:用来描述编码的mime-type。例如在实际使用中常常使用application/json来描述JOSN编码类型。

reply_to（回复目标）：通常用来命名回调队列。

correlation_id（关联标识）：用来将RPC的响应和请求关联起来。
如果大家还想对这些属性更加深的了解，可以到官网或者中文资料中查阅，或者点击这里看看AMQP协议的一些简介及属性等内容。

总结

服务端（rpc_server.py）

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2016-03-08 20:38:06
# @Author  : mengx (mx472756841@gmail.com)
# @Link    : http://blog.csdn.net/mx472756841 # @Version : $Id$
'''
1. 获取请求消息（消费者）
2. 处理请求消息
3. 返回结果消息（生产者）

'''
import traceback
import pika

def fib(n):
if n == 0:
return 0
elif n == 1:
return 1
else:
return fib(n-1) + fib(n-2)

def on_request(ch, method, props, body):
"""
消费处理函数
根据请求的参数，计算斐波那契数列
"""
n = int(body)
print " [.] fib(%s)" % (n,)

# 计算斐波那契
resp = fib(n)

# 将结果返回请求指定队列
ch.basic_publish(exchange='',
routing_key=props.reply_to,
body=str(resp),
properties=pika.BasicProperties(
correlation_id=props.correlation_id)
)

# 消息确认
ch.basic_ack(delivery_tag=method.delivery_tag)

conn = None
try:
# 创建连接
conn = pika.BlockingConnection(pika.ConnectionParameters(
host='localhost'))

# 创建通道
channel = conn.channel()

# 声明队列
channel.queue_declare('rpc_queue')

# 在同一时刻处理一条消息
channel.basic_qos(prefetch_count=1)

# 消费
channel.basic_consume(on_request, queue='rpc_queue')

# 启动
print ' [*] Waiting for messages. To exit press CTRL+C'
channel.start_consuming()

except:
traceback.print_exc()
finally:
if conn:
conn.close()

客户端（rpc_client.py）

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Date    : 2016-03-08 20:59:55
# @Author  : mengx (mx472756841@gmail.com)
# @Link    : http://blog.csdn.net/mx472756841 # @Version : $Id$
'''
1. 请求服务发送消息（作为生产者，指定接受队列）
2. 接收服务返回消息
3. 输出消息
'''

import pika
import uuid

class FibonacciRpcClient(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):
# 比较消息是否是自己发送的消息，如果是的话赋值给response
if self.corr_id == props.correlation_id:
self.response = body

def call(self, n):
self.response = None
self.corr_id = str(uuid.uuid4())
self.channel.basic_publish(exchange='',
routing_key='rpc_queue',
properties=pika.BasicProperties(
reply_to=self.callback_queue,
correlation_id=self.corr_id,
),
body=str(n))
while self.response is None:
# 等待连接的所有消息
self.connection.process_data_events()

return int(self.response)

fibonacci_rpc = FibonacciRpcClient()

print " [x] Requesting fib(30)"
response = fibonacci_rpc.call(30)
print " [.] Got %r" % (response,)

3. 参考资料

官网资料：http://www.rabbitmq.com/tutorials/tutorial-six-python.html
中文资料：RPC/]http://rabbitmq-into-chinese.readthedocs.org/zh_CN/latest/tutorials_with_python/[6]RPC/