RabbitMQ rpc实现
| 一、前言 |
之前我们都是单向发送消息，客户端发送消息给服务端，那么问题来了，我现在发一个命令给远程客户端，让它去执行，执行之后的结果，我想把这个结果再返回。这个模型叫什么呐，这种模型叫RPC=>remote procedure call。
怎么返回这个消息呢？
答：就server 端和客户端既是消费者，又是生产者。
想要深入了解RabbitMQ：猛击这里
| 二、模型图 |
| 三、逻辑代码 |
3.1、RPC Client
__author__ = "Junesu"
import pika,uuid,time
class FibonacciRpcClient(object):
"斐波那契数列rpc客户端"
def __init__(self):
# 建立连接，指定服务器的ip地址
self.connection = pika.BlockingConnection(pika.ConnectionParameters
(host="localhost"))
# 建立一个会话，每个channel代表一个会话任务
self.channel = self.connection.channel()
# 声明回调队列，再次声明的原因是，服务器和客户端可能先后开启，该声明是幂等的，多次声明，但只生效一次
result = self.channel.queue_declare(exclusive=True)
# 将次队列指定为当前客户端的回调队列
self.callback_queue = result.method.queue
# 客户端订阅回调队列，当回调队列中有响应时，调用`on_response`方法对响应进行处理;
self.channel.basic_consume(self.on_response,no_ack=True,
queue=self.callback_queue)
# 对回调队列中的响应进行处理的函数
def on_response(self,ch,method,props,body):
print("---->",method,props)
if self.corr_id == props.correlation_id: #我发过去的结果就是我想要的结果，保持数据的一致性
self.response = body
# 发出RPC请求
def call(self,n):
# 初始化 response
self.response = None
#生成correlation_id
self.corr_id = str(uuid.uuid4())
# 发送RPC请求内容到RPC请求队列`rpc_queue`，同时发送的还有`reply_to`和`correlation_id`
self.channel.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() #非阻塞版的start_consumer()
print("no msg....")
time.sleep(0.5)
return int(self.response)
if __name__ == "__main__":
fibonacci_rpc = FibonacciRpcClient()# 建立客户端
print(" [x] Requesting fib(30)")
response = fibonacci_rpc.call(30)# 发送RPC请求
print(" [.] Got %r" % response)
注：
- 我是想不阻塞，而是想每过一段时间，就过来检查一下，就不能用start_consumer,而是用connection.process_data_evevts()，它是不阻塞的，如果收到消息就收到，收不到消息也返回，就继续往下执行。
- reply_to就是想让服务器执行完命令之后，把结果返回到这个queue里面。
- 在while self.respose is None中的代码我可以不做time.sleep，我这边可以发消息给服务器端，这个消息不一定按顺序发给服务器端，如果不做self.corr_id == props.correlation_id的验证，那数据就可能对不上了。
3.2、RPC Server
import pika
connection = pika.BlockingConnection(pika.ConnectionParameters(host="localhost"))
channel = connection.channel()
channel.queue_declare(queue="rpc_queue")# 声明RPC请求队列
def fib(n): # 数据处理方法
"斐波那契数列"
if n == 0:
return 0
elif n == 1:
return 1
else:
return fib(n-1)+fib(n-2)
# 对RPC请求队列中的请求进行处理
def on_request(ch,method,props,body):
n = int(body)
print(" [.] fib(%s)" % n)
response = fib(n) # 调用数据处理方法
# 将处理结果(响应)发送到回调队列
ch.basic_publish(exchange="",
routing_key=props.reply_to,
properties=pika.BasicProperties(correlation_id=\
props.correlation_id),#props的是客户端的发过来的信息，这边把correlation_id返回给客户端做验证
body=str(response))
ch.basic_ack(delivery_tag=method.delivery_tag)
# 负载均衡，同一时刻发送给该服务器的请求不超过一个
channel.basic_qos(prefetch_count=1)
# 服务器订阅RPC请求队列，当队列中有请求时，将调用`on_request`方法处理请求
channel.basic_consume(on_request,queue="rpc_queue")
print(" [x] Awaiting RPC requests")
channel.start_consuming()
注：props.reply_to，这个就是客户端返回过来的queue。
问：如果客户端和服务用的是同一个queue，会有什么影响？
答：如果客户端也发到rpc_queue中，那么客户端就会收到自己的消息，就会形成一个死循坏，把自己给玩死了。
