RocketMQ高手之路系列之七：RocketMQ之消息发送（四）

cwcnl25131

于 2024-05-03 21:38:32 发布

阅读量633

点赞数 14

分类专栏：程序员文章标签： java-rocketmq rocketmq 面试

本文链接：https://blog.csdn.net/cwcnl25131/article/details/138425315

版权

程序员专栏收录该内容

258 篇文章 0 订阅

订阅专栏

《一线大厂Java面试题解析+核心总结学习笔记+最新讲解视频+实战项目源码》，点击传送门，即可获取！
final Message msg,

//消息需要发送到的消息队列

final MessageQueue mq,

//消息发送模式

final CommunicationMode communicationMode,

//异步消息回调

final SendCallback sendCallback,

//主题路由信息

final TopicPublishInfo topicPublishInfo,

//超时时间

final long timeout)

涉及的具体步骤如下所示：

在这里插入图片描述

我们具体看下源码，相关注释如下所示：

//根据选择的MessageQueue获取对应Broker地址

String brokerAddr = this.mQClientFactory.findBrokerAddressInPublish(mq.getBrokerName());

//如果为获取到

if (null == brokerAddr) {

//从NameServer进行主动更新TOPIC信息

tryToFindTopicPublishInfo(mq.getTopic());

brokerAddr = this.mQClientFactory.findBrokerAddressInPublish(mq.getBrokerName());

}

进行全局ID设置，源码如下：

if (!(msg instanceof MessageBatch)) {

//设置全局唯一ID

MessageClientIDSetter.setUniqID(msg);

}

//消息提的默认大小大小超过4K，则进行zip压缩，并设置消息系统标记

int sysFlag = 0;

boolean msgBodyCompressed = false;

if (this.tryToCompressMessage(msg)) {

sysFlag |= MessageSysFlag.COMPRESSED_FLAG;

msgBodyCompressed = true;

}

//如果事务，则进行系统事务标记

final String tranMsg = msg.getProperty(MessageConst.PROPERTY_TRANSACTION_PREPARED);

if (tranMsg != null && Boolean.parseBoolean(tranMsg)) {

sysFlag |= MessageSysFlag.TRANSACTION_PREPARED_TYPE;

}

是否执行钩子函数，源码如下所示：

//判断是否进行消息发送钩子函数注册了，为一个列表

if (this.hasSendMessageHook()) {

context = new SendMessageContext();

context.setProducer(this);

context.setProducerGroup(this.defaultMQProducer.getProducerGroup());

context.setCommunicationMode(communicationMode);

context.setBornHost(this.defaultMQProducer.getClientIP());

context.setBrokerAddr(brokerAddr);

context.setMessage(msg);

context.setMq(mq);

String isTrans = msg.getProperty(MessageConst.PROPERTY_TRANSACTION_PREPARED);

if (isTrans != null && isTrans.equals(“true”)) {

context.setMsgType(MessageType.Trans_Msg_Half);

}

if (msg.getProperty(“__STARTDELIVERTIME”) != null || msg.getProperty(MessageConst.PROPERTY_DELAY_TIME_LEVEL) != null) {

context.setMsgType(MessageType.Delay_Msg);

}

this.executeSendMessageHookBefore(context);

}

进行消息发送请求包的构建，源码如下所示：

//消息发送请求头

SendMessageRequestHeader requestHeader = new SendMessageRequestHeader();

//设置消息生产组 requestHeader.setProducerGroup(this.defaultMQProducer.getProducerGroup());

requestHeader.setTopic(msg.getTopic());

//设置主题名称

requestHeader.setDefaultTopic(this.defaultMQProducer.getCreateTopicKey());

//设置默认在单个Broker默认队列数

requestHeader.setDefaultTopicQueueNums(this.defaultMQProducer.getDefaultTopicQueueNums());

requestHeader.setQueueId(mq.getQueueId());

//设置系统标记

requestHeader.setSysFlag(sysFlag);

requestHeader.setBornTimestamp(System.currentTimeMillis());

requestHeader.setFlag(msg.getFlag());

requestHeader.setProperties(MessageDecoder.messageProperties2String(msg.getProperties()));

requestHeader.setReconsumeTimes(0);

requestHeader.setUnitMode(this.isUnitMode());

requestHeader.setBatch(msg instanceof MessageBatch);

if (requestHeader.getTopic().startsWith(MixAll.RETRY_GROUP_TOPIC_PREFIX)) {

String reconsumeTimes = MessageAccessor.getReconsumeTime(msg);

if (reconsumeTimes != null) {

requestHeader.setReconsumeTimes(Integer.valueOf(reconsumeTimes));

MessageAccessor.clearProperty(msg, MessageConst.PROPERTY_RECONSUME_TIME);

}

String maxReconsumeTimes = MessageAccessor.getMaxReconsumeTimes(msg);

if (maxReconsumeTimes != null) {

//设置最大重试次数

requestHeader.setMaxReconsumeTimes(Integer.valueOf(maxReconsumeTimes));

MessageAccessor.clearProperty(msg, MessageConst.PROPERTY_MAX_RECONSUME_TIMES);

}

最后根据communicationMode，进行消息发送模式的选择，其中包括了同步发送方式、异步发送方式以及单向发送方式，大致的源码如下所示：

SendResult sendResult = null;

switch (communicationMode) {

//异步方式

case ASYNC:

Message tmpMessage = msg;

if (msgBodyCompressed) {

//If msg body was compressed, msgbody should be reset using prevBody.

//Clone new message using commpressed message body and recover origin massage.

//Fix bug:https://github.com/apache/rocketmq-externals/issues/66

tmpMessage = MessageAccessor.cloneMessage(msg);

最后我们该如何学习？

1、看视频进行系统学习

这几年的Crud经历，让我明白自己真的算是菜鸡中的战斗机，也正因为Crud，导致自己技术比较零散，也不够深入不够系统，所以重新进行学习是很有必要的。我差的是系统知识，差的结构框架和思路，所以通过视频来学习，效果更好，也更全面。关于视频学习，个人可以推荐去B站进行学习，B站上有很多学习视频，唯一的缺点就是免费的容易过时。

另外，我自己也珍藏了好几套视频资料躺在网盘里，有需要的我也可以分享给你：