关于大数据:SpringBootRabbitMQ-实现-RPC-调用

32次阅读

共计 5752 个字符,预计需要花费 15 分钟才能阅读完成。

说到 RPC(Remote Procedure Call Protocol 近程过程调用协定),小伙伴们脑海里蹦出的预计都是 RESTful API、Dubbo、WebService、Java RMI、CORBA 等。

其实,RabbitMQ 也给咱们提供了 RPC 性能,并且应用起来很简略。

明天松哥通过一个简略的案例来和大家分享一下 Spring Boot+RabbitMQ 如何实现一个简略的 RPC 调用。

留神

对于 RabbitMQ 实现 RPC 调用,有的小伙伴可能会有一些误会,心想这还不简略?搞两个音讯队列 queue_1 和 queue_2,首先客户端发送音讯到 queue_1 上,服务端监听 queue_1 上的音讯,收到之后进行解决;解决实现后,服务端发送音讯到 queue_2 队列上,而后客户端监听 queue_2 队列上的音讯,这样就晓得服务端的处理结果了。

这种形式不是不能够,就是有点麻烦!RabbitMQ 中提供了现成的计划能够间接应用,十分不便。接下来咱们就一起来学习下。

  1. 架构
    先来看一个简略的架构图:

这张图把问题说的很明确了:

首先 Client 发送一条音讯,和一般的音讯相比,这条音讯多了两个要害内容:一个是 correlation_id,这个示意这条音讯的惟一 id,还有一个内容是 reply_to,这个示意音讯回复队列的名字。
Server 从音讯发送队列获取音讯并解决相应的业务逻辑,解决实现后,将处理结果发送到 reply_to 指定的回调队列中。
Client 从回调队列中读取音讯,就能够晓得音讯的执行状况是什么样子了。
这种状况其实非常适合解决异步调用。

  1. 实际
    接下来咱们通过一个具体的例子来看看这个怎么玩。

2.1 客户端开发
首先咱们来创立一个 Spring Boot 工程名为 producer,作为音讯生产者,创立时候增加 web 和 rabbitmq 依赖,如下图:

我的项目创立胜利之后,首先在 application.properties 中配置 RabbitMQ 的根本信息,如下:

spring.rabbitmq.host=localhost
spring.rabbitmq.port=5672
spring.rabbitmq.username=guest
spring.rabbitmq.password=guest
spring.rabbitmq.publisher-confirm-type=correlated
spring.rabbitmq.publisher-returns=true
这个配置后面四行都好了解,我就不赘述,前面两行:首先是配置音讯确认形式,咱们通过 correlated 来确认,只有开启了这个配置,未来的音讯中才会带 correlation_id,只有通过 correlation_id 咱们能力将发送的音讯和返回值之间关联起来。最初一行配置则是开启发送失败退回。大数据培训

接下来咱们来提供一个配置类,如下:

@Configuration
public class RabbitConfig {

public static final String RPC_QUEUE1 = "queue_1";
public static final String RPC_QUEUE2 = "queue_2";
public static final String RPC_EXCHANGE = "rpc_exchange";

/**
 * 设置音讯发送 RPC 队列
 */
@Bean
Queue msgQueue() {return new Queue(RPC_QUEUE1);
}

/**
 * 设置返回队列
 */
@Bean
Queue replyQueue() {return new Queue(RPC_QUEUE2);
}

/**
 * 设置交换机
 */
@Bean
TopicExchange exchange() {return new TopicExchange(RPC_EXCHANGE);
}

/**
 * 申请队列和交换器绑定
 */
@Bean
Binding msgBinding() {return BindingBuilder.bind(msgQueue()).to(exchange()).with(RPC_QUEUE1);
}

/**
 * 返回队列和交换器绑定
 */
@Bean
Binding replyBinding() {return BindingBuilder.bind(replyQueue()).to(exchange()).with(RPC_QUEUE2);
}


/**
 * 应用 RabbitTemplate 发送和接管音讯
 * 并设置回调队列地址
 */
@Bean
RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {RabbitTemplate template = new RabbitTemplate(connectionFactory);
    template.setReplyAddress(RPC_QUEUE2);
    template.setReplyTimeout(6000);
    return template;
}


/**
 * 给返回队列设置监听器
 */
@Bean
SimpleMessageListenerContainer replyContainer(ConnectionFactory connectionFactory) {SimpleMessageListenerContainer container = new SimpleMessageListenerContainer();
    container.setConnectionFactory(connectionFactory);
    container.setQueueNames(RPC_QUEUE2);
    container.setMessageListener(rabbitTemplate(connectionFactory));
    return container;
}

}
这个配置类中咱们别离配置了音讯发送队列 msgQueue 和音讯返回队列 replyQueue,而后将这两个队列和音讯交换机进行绑定。这个都是 RabbitMQ 的惯例操作,没啥好说的。

在 Spring Boot 中咱们负责音讯发送的工具是 RabbitTemplate,默认状况下,零碎主动提供了该工具,然而这里咱们须要对该工具从新进行定制,次要是增加音讯发送的返回队列,最初咱们还须要给返回队列设置一个监听器。

好啦,接下来咱们就能够开始具体的音讯发送了:

@RestController
public class RpcClientController {

private static final Logger logger = LoggerFactory.getLogger(RpcClientController.class);

@Autowired
private RabbitTemplate rabbitTemplate;

@GetMapping("/send")
public String send(String message) {
    // 创立音讯对象
    Message newMessage = MessageBuilder.withBody(message.getBytes()).build();

    logger.info("client send:{}", newMessage);

    // 客户端发送音讯
    Message result = rabbitTemplate.sendAndReceive(RabbitConfig.RPC_EXCHANGE, RabbitConfig.RPC_QUEUE1, newMessage);

    String response = "";
    if (result != null) {
        // 获取已发送的音讯的 correlationId
        String correlationId = newMessage.getMessageProperties().getCorrelationId();
        logger.info("correlationId:{}", correlationId);

        // 获取响应头信息
        HashMap<String, Object> headers = (HashMap<String, Object>) result.getMessageProperties().getHeaders();

        // 获取 server 返回的音讯 id
        String msgId = (String) headers.get("spring_returned_message_correlation");

        if (msgId.equals(correlationId)) {response = new String(result.getBody());
            logger.info("client receive:{}", response);
        }
    }
    return response;
}

}
这块的代码其实也都是一些惯例代码,我挑几个要害的节点说下:

音讯发送调用 sendAndReceive 办法,该办法自带返回值,返回值就是服务端返回的音讯。
服务端返回的音讯中,头信息中蕴含了 spring_returned_message_correlation 字段,这个就是音讯发送时候的 correlation_id,通过音讯发送时候的 correlation_id 以及返回音讯头中的 spring_returned_message_correlation 字段值,咱们就能够将返回的音讯内容和发送的音讯绑定到一起,确认出这个返回的内容就是针对这个发送的音讯的。
这就是整个客户端的开发,其实最最外围的就是 sendAndReceive 办法的调用。调用尽管简略,然而筹备工作还是要做足够。例如如果咱们没有在 application.properties 中配置 correlated,发送的音讯中就没有 correlation_id,这样就无奈将返回的音讯内容和发送的音讯内容关联起来。

2.2 服务端开发
再来看看服务端的开发。

首先创立一个名为 consumer 的 Spring Boot 我的项目,创立我的项目增加的依赖和客户端开发创立的依赖是统一的,不再赘述。

而后配置 application.properties 配置文件,该文件的配置也和客户端中的配置统一,不再赘述。

接下来提供一个 RabbitMQ 的配置类,这个配置类就比较简单,单纯的配置一下音讯队列并将之和音讯交换机绑定起来,如下:

@Configuration
public class RabbitConfig {

public static final String RPC_QUEUE1 = "queue_1";
public static final String RPC_QUEUE2 = "queue_2";
public static final String RPC_EXCHANGE = "rpc_exchange";

/**
 * 配置音讯发送队列
 */
@Bean
Queue msgQueue() {return new Queue(RPC_QUEUE1);
}

/**
 * 设置返回队列
 */
@Bean
Queue replyQueue() {return new Queue(RPC_QUEUE2);
}

/**
 * 设置交换机
 */
@Bean
TopicExchange exchange() {return new TopicExchange(RPC_EXCHANGE);
}

/**
 * 申请队列和交换器绑定
 */
@Bean
Binding msgBinding() {return BindingBuilder.bind(msgQueue()).to(exchange()).with(RPC_QUEUE1);
}

/**
 * 返回队列和交换器绑定
 */
@Bean
Binding replyBinding() {return BindingBuilder.bind(replyQueue()).to(exchange()).with(RPC_QUEUE2);
}

}
最初咱们再来看下音讯的生产:

@Component
public class RpcServerController {

private static final Logger logger = LoggerFactory.getLogger(RpcServerController.class);
@Autowired
private RabbitTemplate rabbitTemplate;

@RabbitListener(queues = RabbitConfig.RPC_QUEUE1)
public void process(Message msg) {logger.info("server receive : {}",msg.toString());
    Message response = MessageBuilder.withBody(("i'm receive:"+new String(msg.getBody())).getBytes()).build();
    CorrelationData correlationData = new CorrelationData(msg.getMessageProperties().getCorrelationId());
    rabbitTemplate.sendAndReceive(RabbitConfig.RPC_EXCHANGE, RabbitConfig.RPC_QUEUE2, response, correlationData);
}

}
这里的逻辑就比较简单了:

服务端首先收到音讯并打印进去。
服务端提取出原音讯中的 correlation_id。
服务端调用 sendAndReceive 办法,将音讯发送给 RPC_QUEUE2 队列,同时带上 correlation_id 参数。
服务端的音讯收回后,客户端将收到服务端返回的后果。

OK,功败垂成。

2.3 测试
接下来咱们进行一个简略测试。

首先启动 RabbitMQ。

接下来别离启动 producer 和 consumer,而后在 postman 中调用 producer 的接口进行测试,如下:

能够看到,曾经收到了服务端的返回信息。

来看看 producer 的运行日志:

能够看到,音讯发送进来后,同时也收到了 consumer 返回的信息。

能够看到,consumer 也收到了客户端发来的音讯。

正文完
 0