基于RabbitMQ实现简单RPC服务

写程序的小王叔叔

已于 2022-05-26 07:08:29 修改

阅读量775

点赞数 3

分类专栏： 🐆 面试 / 转载 / 分享文章标签： java 开发语言架构中间件

于 2020-07-08 15:14:34 首次发布

原文链接：https://mp.weixin.qq.com/s/_YgRZD_oi0CqAdSPSpDUig

版权

🐆 面试 / 转载 / 分享专栏收录该内容

58 篇文章 8 订阅

订阅专栏

小Hub领读：

基于RabbitMq，怎么弄一个RPC框架，来看看代码，应该能看懂哈！

RPC(Remote Procedure Call, 远程过程调用)，是一种计算机通信协议。对于两台机器而言，就是 A 服务器上的应用程序调用 B 服务器上的函数或者方法，由于不在同一个内存空间或机器上运行，因此需要借助于网络通信。

1. RPC 框架

我们首先通过一张图理解 RPC 的工作流程：

因此，实现一个最简单的 RPC 服务，只需要 Client、Server 和 Network，本文就是利用消息中间件 RabbitMQ 作为 Network 载体传输信息，实现简单的 RPC 服务。简单原理可如下图所示：

即：当 Client 发送 RPC 请求时，Client 端是消息生产者，Server 端是消息消费者；当 Server 返回结果时，Server 端是消息生产者，Client 是消息消费者；发送和返回使用不同的队列。

接下来我们通过代码，详细展示一个计算斐波那契数列的 RPC 服务。

2. RPCServer 实现

2.1 Server 初始化

初始化就是声明 RabbitMQ 的链接工厂、链接、信道、队列、交换机等等，并做了绑定，由此构成了 AMQP 的通信结构。

2.2 监听队列并反馈

	
	
		/**

		* 开启server

		*/

		private void startServer() {

		try {

				LOG.info("Waiting for RPC calls.....");

				while (true) {
			
					//获得文本消息
			
					QueueingConsumer.Delivery delivery = consumer.nextDelivery();
			
					BasicProperties props = delivery.getProperties();
			
			
					//返回消息的属性
			
					BasicProperties replyProps = new BasicProperties.Builder().correlationId(props.getCorrelationId()).build();
			
					long receiveTime = System.currentTimeMillis();
			
					JSONObject json = new JSONObject();
	
					try {
			
							String message = new String(delivery.getBody(), "UTF-8");
					
							int n = Integer.parseInt(message);
					
							LOG.info("Got a request: fib(" + message + ")");
					
							json.put("status", "success");
					
							json.put("result", fib(n));
			
					} catch (Exception e) {
				
							json.put("status", "fail");
					
							json.put("reason", "Not a Number!");
					
							LOG.error("receive message failed!", e);
			
					} finally {
			
							long responseTime = System.currentTimeMillis();
					
							json.put("calculateTime", (responseTime - receiveTime));
					
							channel.basicPublish("", props.getReplyTo(), replyProps, json.toString().getBytes("UTF-8"));
					
							channel.basicAck(delivery.getEnvelope().getDeliveryTag(), false);
			
					}
		
				}
		
			} catch (Exception e) {
	
				LOG.error("server failed!", e);
	
			} finally {
	
				if (connection != null) {
		
					try {
			
						connection.close();
			
					} catch (Exception e) {
			
						LOG.error("close failed!", e);
			
					}
		
				}
		
			}

		}

在该方法中使用了一个无限循环，每次处理一条消息。通过调用消费者对象的 nextDelivery 方法来获得 RabbitMQ 队列的最新一条消息。同时通过 getProperties 获取到消息中的反馈信息属性，用于标记客户端 Client 的属性。然后计算斐波那契数列的结果。
最后通过 basicAck 使用消息信封向 RabbitMQ 确认了该消息。

到这里就实现了计算斐波那契数列 RPC 服务的 Server 端。

3. RPCClient 实现

3.1 初始化 CLient

这里声明 AMQP 结构体的方式和 Server 端类似，只不过 Client 端需要多声明一个队列，用于 RPC 的 response。

3.2 发送 / 接收消息


		/**

		* 请求server

		* @param message

		* @return

		* @throws Exception

		*/

		private String requestMessage(String message) throws Exception {

			String response = null;
	
			String corrId = UUID.randomUUID().toString();
	
			BasicProperties props = new BasicProperties.Builder().correlationId(corrId).replyTo(RESPONSE_QUEUE).build();
	
			channel.basicPublish("", QUEUE_NAME, props, message.getBytes("UTF-8"));
	
			while (true) {
	
				QueueingConsumer.Delivery delivery = consumer.nextDelivery();
		
				if (delivery.getProperties().getCorrelationId().equals(corrId)) {
		
					response = new String(delivery.getBody(),"UTF-8");
		
					break;
				}
			}
			return response;

		}

BasicProperties 用于存储你请求消息的属性，这里我设置了 correlationId 和 replyTo 属性，用于 Server 端的返回识别。