rocketmq 消费端未拉取到消息处理机制

最新推荐文章于 2025-10-28 22:18:36 发布

原创

最新推荐文章于 2025-10-28 22:18:36 发布 · 1.9k 阅读

本文深入探讨RocketMQ中的长轮询机制，详细解析在消费端未能立即获取消息时，broker如何通过suspend机制缓存PullRequest并定时检查，直至有新消息到达或超时，从而实现高效的消息拉取。

一句话概述

我们知道在消费端向broker拉取消息的时候是以PullRequest为驱动的，如果拉取消息成功，那么这个PullRequest会被再次放回消费端的队列中，进而不断的进行拉取；

那在该PullRequest没有拉取到消息的情况下呢？这种情况下broker并不会立即响应consumer，而是会执行一种叫suspend的机制。
该机制会将该拉取任务对象暂时缓存起来，待一定时间后再次执行消息拉取的动作并将拉取的结果返回给客户端；

暂停的时间有两种。A:broker的longPollingEnable = true 时取 requestHeader.getSuspendTimeoutMillis()即消费端传过来的暂停时间；B : broker的longPollingEnable = false 时取的是broker端的配置即this.brokerController.getBrokerConfig().getShortPollingTimeMills()。

关键入口

org.apache.rocketmq.broker.processor.PullMessageProcessor#processRequest(io.netty.channel.Channel, org.apache.rocketmq.remoting.protocol.RemotingCommand, boolean){

//省略 lot code

case ResponseCode.PULL_NOT_FOUND:

                    if (brokerAllowSuspend && hasSuspendFlag) {
                        long pollingTimeMills = suspendTimeoutMillisLong;
                        if (!this.brokerController.getBrokerConfig().isLongPollingEnable()) {
                            pollingTimeMills = this.brokerController.getBrokerConfig().getShortPollingTimeMills();
                        }

                        String topic = requestHeader.getTopic();
                        long offset = requestHeader.getQueueOffset();
                        int queueId = requestHeader.getQueueId();
                        PullRequest pullRequest = new PullRequest(request, channel, pollingTimeMills, this.brokerController.getMessageStore().now(), offset, subscriptionData, messageFilter);

//这里是关键代码，broker会将组装一个PullRequest缓存起来放到PullRequestHoldService中，待处理
                        this.brokerController.getPullRequestHoldService().suspendPullRequest(topic, queueId, pullRequest);
                        response = null;
                        break;
                    }

//省略 lot code

}