RocketMQ源码分析(六)之Broker消息接收

jannals

已于 2022-08-17 21:22:59 修改

阅读量311

点赞数

分类专栏： # RocketMQ源码分析文章标签： RocketMQ源码 Broker消息接收

于 2022-08-17 20:58:25 首次发布

本文链接：https://blog.csdn.net/usagoole/article/details/126394165

版权

RocketMQ源码分析专栏收录该内容

21 篇文章 12 订阅

订阅专栏

文章目录

版本
Broker处理消息流程
- SendMessageProcessor

版本

基于rocketmq-all-4.3.1版本

Broker处理消息流程

RocketMQ客户端发送消息方法是MQClientAPIImpl#sendMessage,发送消息的请求命令是RequestCode.SEND_MESSAGE(10)

public SendResult sendMessage(
    final String addr,
    final String brokerName,
    final Message msg,
    final SendMessageRequestHeader requestHeader,
    final long timeoutMillis,
    final CommunicationMode communicationMode,
    final SendCallback sendCallback,
    final TopicPublishInfo topicPublishInfo,
    final MQClientInstance instance,
    final int retryTimesWhenSendFailed,
    final SendMessageContext context,
    final DefaultMQProducerImpl producer
) throws RemotingException, MQBrokerException, InterruptedException {
    long beginStartTime = System.currentTimeMillis();
    RemotingCommand request = null;
    if (sendSmartMsg || msg instanceof MessageBatch) {
        SendMessageRequestHeaderV2 requestHeaderV2 = SendMessageRequestHeaderV2.createSendMessageRequestHeaderV2(requestHeader);
        request = RemotingCommand.createRequestCommand(msg instanceof MessageBatch ? RequestCode.SEND_BATCH_MESSAGE : RequestCode.SEND_MESSAGE_V2, requestHeaderV2);
    } else {
        request = RemotingCommand.createRequestCommand(RequestCode.SEND_MESSAGE, requestHeader);
    }
    ...省略...
}

在Broker端源码可以找到注册RequestCode.SEND_MESSAGE命令的地方是BrokerController#registerProcessor。对应的处理类是SendMessageProcessor

public void registerProcessor() {
   SendMessageProcessor sendProcessor = new SendMessageProcessor(this);
   sendProcessor.registerSendMessageHook(sendMessageHookList);
   sendProcessor.registerConsumeMessageHook(consumeMessageHookList);
	//可以看到处理类是SendMessageProcessor
   this.remotingServer.registerProcessor(RequestCode.SEND_MESSAGE, sendProcessor, this.sendMessageExecutor);
  ...省略...
}

SendMessageProcessor

SendMessageProcessor#processRequest是客户端发送消息的处理方法（单个消息和批量消息），此方法是一个模板方法，增加钩子处理函数，批量消息和单条消息处理逻辑。SendMessageProcessor#processRequest整体的流程如下
单条消息和批量消息都是调用AbstractSendMessageProcessor#msgCheck进行主要参数检查。批量消息不支持私信队列，因为只有消费失败时Consumer才会发送单条消息到私信队列，并不会发送批量消息进去死信队列，所以不存在重试Topic
AbstractSendMessageProcessor#msgCheck的执行逻辑
- 检查Broker是否有写权限
- 检查topic是否可以进行消息发送，主要针对默认主题，默认主题不能发送消息，仅供路由查找
- 如果Topic不存在，则创建Topic。在NameServer端存储Topic的配置信息，默认路径为${ROCKET_HOME}/store/config/topic.json。