一文详解RocketMQ-Spring的源码解析与实战

打印 上一主题 下一主题

主题 906|帖子 906|积分 2718

摘要:这篇文章主要介绍 Spring Boot 项目使用 rocketmq-spring SDK 实现消息收发的操作流程,同时笔者会从开发者的角度解读 SDK 的设计逻辑。
本文分享自华为云社区《RocketMQ-Spring : 实战与源码解析一网打尽》,作者:勇哥java实战分享。
RocketMQ 是大家耳熟能详的消息队列,开源项目 rocketmq-spring 可以帮助开发者在 Spring Boot 项目中快速整合 RocketMQ。
这篇文章会介绍 Spring Boot 项目使用 rocketmq-spring SDK 实现消息收发的操作流程,同时笔者会从开发者的角度解读 SDK 的设计逻辑。
一 SDK 简介

项目地址:https://github.com/apache/rocketmq-spring
rocketmq-spring 的本质是一个 Spring Boot starter 。
Spring Boot 基于“约定大于配置”(Convention over configuration)这一理念来快速地开发、测试、运行和部署 Spring 应用,并能通过简单地与各种启动器(如 spring-boot-web-starter)结合,让应用直接以命令行的方式运行,不需再部署到独立容器中。
Spring Boot starter 构造的启动器使用起来非常方便,开发者只需要在 pom.xml 引入 starter 的依赖定义,在配置文件中编写约定的配置即可。
下面我们看下 rocketmq-spring-boot-starter 的配置:
1、引入依赖
  1. <dependency>
  2. <groupId>org.apache.rocketmq</groupId>
  3. <artifactId>rocketmq-spring-boot-starter</artifactId>
  4. <version>2.2.3</version>
  5. </dependency>
复制代码
2、约定配置

接下来,我们分别按照生产者和消费者的顺序,详细的讲解消息收发的操作过程。
二 生产者

首先我们添加依赖后,进行如下三个步骤:
1、配置文件中配置如下
  1. rocketmq:
  2.   name-server: 127.0.0.1:9876
  3.   producer:
  4.       group: platform-sms-server-group
  5.     # access-key: myaccesskey
  6.     # secret-key: mysecretkey
  7.   topic: sms-common-topic
复制代码
生产者配置非常简单,主要配置名字服务地址和生产者组。
2、需要发送消息的类中注入 RcoketMQTemplate
  1. @Autowired
  2. private RocketMQTemplate rocketMQTemplate;
  3. @Value("${rocketmq.topic}")
  4. private String smsTopic;
复制代码
3、发送消息,消息体可以是自定义对象,也可以是 Message 对象

rocketMQTemplate 类包含多钟发送消息的方法:

  • 同步发送 syncSend
  • 异步发送 asyncSend
  • 顺序发送 syncSendOrderly
  • oneway发送 sendOneWay
下面的代码展示如何同步发送消息。
  1. String destination = StringUtils.isBlank(tags) ? topic : topic + ":" + tags;
  2. SendResult sendResult =
  3. rocketMQTemplate.syncSend(
  4.             destination,
  5. MessageBuilder.withPayload(messageContent).
  6. setHeader(MessageConst.PROPERTY_KEYS, uniqueId).
  7. build()
  8. );
  9. if (sendResult != null) {
  10. if (sendResult.getSendStatus() == SendStatus.SEND_OK) {
  11. // send message success ,do something
  12. }
  13. }
复制代码
syncSend 方法的第一个参数是发送的目标,格式是:topic + ":" + tags ,
第二个参数是:spring-message 规范的 message 对象 ,而 MessageBuilder 是一个工具类,方法链式调用创建消息对象。
三 消费者

1、配置文件中配置如下
  1. rocketmq:
  2.   name-server: 127.0.0.1:9876
  3.   consumer1:
  4.     group: platform-sms-worker-common-group
  5.     topic: sms-common-topic
复制代码
2、实现消息监听器
  1. @Component
  2. @RocketMQMessageListener(
  3. consumerGroup = "${rocketmq.consumer1.group}", //消费组
  4.     topic = "${rocketmq.consumer1.topic}" //主题
  5. )
  6. public class SmsMessageCommonConsumer implements RocketMQListener<String> {
  7. public void onMessage(String message) {
  8. System.out.println("普通短信:" + message);
  9. }
  10. }
复制代码
消费者实现类也可以实现 RocketMQListener, 在 onMessage 方法里通过 RocketMQ 原生消息对象 MessageExt 获取更详细的消息数据 。
  1. public void onMessage(MessageExt message) {
  2. try {
  3.         String body = new String(message.getBody(), "UTF-8");
  4.         logger.info("普通短信:" + message);
  5. } catch (Exception e) {
  6. logger.error("common onMessage error:", e);
  7. }
  8. }
复制代码
四 源码概览

最新源码中,我们可以看到源码中包含四个模块:
1、rocketmq-spring-boot-parent
该模块是父模块,定义项目所有依赖的 jar 包。
2、rocketmq-spring-boot
核心模块,实现了 starter 的核心逻辑。
3、rocketmq-spring-boot-starter
SDK 模块,简单封装,外部项目引用。
4、rocketmq-spring-boot-samples
示例代码模块。这个模块非常重要,当用户使用 SDK 时,可以参考示例快速开发。
五 starter 实现

我们重点分析下 rocketmq-spring-boot 模块的核心源码:
spring-boot-starter 实现需要包含如下三个部分:
1、定义 Spring 自身的依赖包和 RocketMQ 的依赖包 ;
2、定义spring.factories 文件
在 resources 包下创建 META-INF 目录后,新建 spring.factories 文件,并在文件中定义自动加载类,文件内容是:
  1. org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
  2. org.apache.rocketmq.spring.autoconfigure.RocketMQAutoConfiguration
复制代码
spring boot 会根据文件中配置的自动化配置类来自动初始化相关的 Bean、Component 或 Service。
3、实现自动加载类
在 RocketMQAutoConfiguration 类的具体实现中,我们重点分析下生产者和消费者是如何分别启动的。
▍生产者发送模板类:RocketMQTemplate

RocketMQAutoConfiguration 类定义了两个默认的 Bean :
首先SpringBoot项目中配置文件中的配置值会根据属性条件绑定到 RocketMQProperties 对象 中,然后使用 RocketMQ 的原生 API 分别创建生产者 Bean 和拉取消费者 Bean , 分别将两个 bean 设置到 RocketMQTemplate 对象中。
两个重点需要强调:

  • 发送消息时,将 spring-message 规范下的消息对象封装成 RocketMQ 消息对象


  • 默认拉取消费者 litePullConsumer 。拉取消费者一般用于大数据批量处理场景 。
RocketMQTemplate 类封装了拉取消费者的receive方法,以方便开发者使用。
▍自定义消费者类

下图是并发消费者的例子:
那么 rocketmq-spring 是如何自动启动消费者呢 ?
spring 容器首先注册了消息监听器后置处理器,然后调用 ListenerContainerConfiguration 类的 registerContainer 方法 。
对比并发消费者的例子,我们可以看到: DefaultRocketMQListenerContainer 是对 DefaultMQPushConsumer 消费逻辑的封装。
封装消费消息的逻辑,同时满足 RocketMQListener 泛化接口支持不同参数,比如 String 、MessageExt 、自定义对象 。
首先DefaultRocketMQListenerContainer初始化之后, 获取 onMessage 方法的参数类型 。
然后消费者调用 consumeMessage 处理消息时,封装了一个 handleMessage 方法 ,将原生 RocketMQ 消息对象 MessageExt 转换成 onMessage 方法定义的参数对象,然后调用 rocketMQListener 的 onMessage 方法。
上图右侧标红的代码也就是该方法的精髓:
  1. rocketMQListener.onMessage(doConvertMessage(messageExt));
复制代码
六 写到最后

开源项目 rocketmq-spring 有很多值得学习的地方 ,我们可以从如下四个层面逐层进阶:
1、学会如何使用 :参考 rocketmq-spring-boot-samples 模块的示例代码,学会如何发送和接收消息,快速编码;
2、模块设计:学习项目的模块分层 (父模块、SDK 模块、核心实现模块、示例代码模块);
3、starter 设计思路 :定义自动配置文件 spring.factories 、设计配置属性类 、在 RocketMQ client 的基础上实现优雅的封装、深入理解 RocketMQ 源码等;
4、举一反三:当我们理解了 rocketmq-spring 的源码,我们可以尝试模仿该项目写一个简单的 spring boot starter。
 
点击关注,第一时间了解华为云新鲜技术~

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!
回复

使用道具 举报

0 个回复

倒序浏览

快速回复

您需要登录后才可以回帖 登录 or 立即注册

本版积分规则

美丽的神话

金牌会员
这个人很懒什么都没写!

标签云

快速回复 返回顶部 返回列表