RocketMQ消息消费二：长轮询

最新推荐文章于 2023-07-05 09:13:04 发布

yankunhaha

最新推荐文章于 2023-07-05 09:13:04 发布

阅读量858

点赞数

分类专栏： java RocketMQ源码

本文链接：https://blog.csdn.net/yankunhaha/article/details/99084017

版权

RocketMQ的消费模式实际上是通过pull实现，为提高效率，未获取到消息时会挂起请求。本文介绍了长轮询和短轮询的分类，详细讲解了拉取请求的挂起与唤醒逻辑，涉及PullRequestHoldService关键类的作用。

摘要由CSDN通过智能技术生成

概述

前面提到了，RocketMQ没有真正实现push模式消费，而是通过pull来实现。为了提高性能，未拉取到消息时，并不会返回消息未找到，而是将拉取请求挂起，直到有消息到来被唤醒或者超时。

轮询分类

分为长轮询和短轮询。短轮询1s，长轮询push模式为15s，pull模式为20s。

轮询触发逻辑

在 PullMessageProcessor.processRequest() 中

                case ResponseCode.PULL_NOT_FOUND:
                    // broker 允许挂起，拉取请求设置了挂起标记
                    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);
                        // 挂起请求
                        this.brokerController.getPullRequestHoldService().suspendPullRequest(topic, queueId, pullRequest);
                        response = null; // 设置response为null，不会返回数据
                        break;
                    }

PullRequestHoldService 关键类

PullRequestHoldService执行具体挂起逻辑，它继承自 ServiceThread。
先来看挂起拉取请求方法：

    public void suspendPullRequest(final String topic, final int queueId, final PullRequest pullRequest) {
   
        String key = this.buildKey(topic, queueId);
        // 以 topic@queueId为key，保存拉取请求
        ManyPullRequest mpr = this.pullRequestTable.get(key);
        if (null == mpr) {
   
            mpr = new ManyPullRequest();
            ManyPullRequest prev = this.pullRequestTable.putIfAbsent(key, mpr);
            if (prev != null) {