一、前言 |
之前我们都是单向发送消息,客户端发送消息给服务端,那么问题来了,我现在发一个命令给远程客户端,让它去执行,执行之后的结果,我想把这个结果再返回。这个模型叫什么呐,这种模型叫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中,那么客户端就会收到自己的消息,就会形成一个死循坏,把自己给玩死了。