RocketMQ源码解读——同一消费组下不同消费者订阅关系不同时

@(rocketmq源码解读)

先解释一下题目，我们假设有一个Producer和两个Consumer，Producer向TOPICA和TOPICB发送消息，两个Consumer分别订阅两个topic。我们看下这时候会出现的问题，以及根据源码分析一下为什么出现问题。

现象

现象其实还是比较隐蔽的，broker上会打印：the consumer's subscription not exist，group ...的日志（Consumer端也会打印类似的日志）。

还会有一些subscription changed, group: ...类似的日志，并且如果仔细的话还会发现，其中一个消费者消费消息时，另外一个就不会消费。

源码分析

我们看一下为什么会导致这样的问题，一开始生看或者debug都是很难下手，这时候可能就需要使用必杀技（一般不外传那种）——问。

问天问地，谷歌百度必应。我直接问了一个大神——芋艿。大神说这种情况会出问题，具体原因他也记不清了，导致这种现象的问题应该是消费关系不停地相互覆盖。

好了，听到这句话我们就有入口了，至少知道应该从Broker上找起。

顺藤摸瓜找到了原因，下面一起看一下源码。

首先我们知道，消费者的两种实现（推和拉）中都维护一个MQClientInstance，这个类非常重要，在启动消费者的时候，都会去启动这个类，我们看下启动的代码，其中有这么一部分：

// Start various schedule tasks
this.startScheduledTask();

这里启动了好多定时任务，我们追进去看一下：

this.scheduledExecutorService.scheduleAtFixedRate(new Runnable() {

    @Override
    public void run() {
        try {
            MQClientInstance.this.cleanOfflineBroker();
            //定时发送心跳
            MQClientInstance.this.sendHeartbeatToAllBrokerWithLock();
        } catch (Exception e) {
            log.error("ScheduledTask sendHeartbeatToAllBroker exception", e);
        }
    }
}, 1000, this.clientConfig.getHeartbeatBrokerInterval(), TimeUnit.MILLISECONDS);

这里我们看到，消费者会定时发送心跳给Broker，我们继续追进去，最后找到sendHeartbeatToAllBroker方法：

//给所有的broker发送心跳
if (!this.brokerAddrTable.isEmpty()) {
    long times = this.sendHeartbeatTimesTotal.getAndIncrement();
    Iterator<Entry<String, HashMap<Long, String>>> it = this.brokerAddrTable.entrySet().iterator();
    while (it.hasNext()) {
        Entry<String, HashMap<Long, String>> entry = it.next();
        String brokerName = entry.getKey();
        HashMap<Long, String> oneTable = entry.getValue();
        if (oneTable != null) {
            for (Map.Entry<Long, String> entry1 : oneTable.entrySet()) {
                Long id = entry1.getKey();
                String addr = entry1.getValue();
                if (addr != null) {
                    if (consumerEmpty) {
                        if (id != MixAll.MASTER_ID)
                            continue;
                    }

                    try {
	                    //真正发送心跳的部分
                        int version = this.mQClientAPIImpl.sendHearbeat(addr, heartbeatData, 3000);
                        if (!this.brokerVersionTable.containsKey(brokerName)) {
                            this.brokerVersionTable.put(brokerName, new HashMap<String, Integer>(4));
                        }
                        this.brokerVersionTable.get(brokerName).put(addr, version);
                        if (times % 20 == 0) {
                            log.info("send heart beat to broker[{} {} {}] success", brokerName, id, addr);
                            log.info(heartbeatData.toString());
                        }
                    } catch (Exception e) {
                        if (this.isBrokerInNameServer(addr)) {
                            log.info("send heart beat to broker[{} {} {}] failed", brokerName, id, addr);
                        } else {
                            log.info("send heart beat to broker[{} {} {}] exception, because the broker not up, forget it", brokerName,
                                id, addr);
                        }
                    }
                }
            }
        }
    }
}

这里会向所有的Broker发送心跳，我们根据我们的例子，这时候Broker是一台，我们再去Broker上看一下Broker如何处理心跳消息，我们根据发送的是HEART_BEAT类型的消息，可以在Broker上看到，这类消息使用ClientManageProcessor处理，我们看下处理心跳的部分（heartBeat方法）：

//循环所有发送过来的数据
for (ConsumerData data : heartbeatData.getConsumerDataSet()) {
	//根据消费组的名字获取broker上记录的消费消息
    SubscriptionGroupConfig subscriptionGroupConfig = this.brokerController.getSubscriptionGroupManager().findSubscriptionGroupConfig(data.getGroupName());
    boolean isNotifyConsumerIdsChangedEnable = true;
    if (null != subscriptionGroupConfig) {
        isNotifyConsumerIdsChangedEnable = subscriptionGroupConfig.isNotifyConsumerIdsChangedEnable();
        int topicSysFlag = 0;
        if (data.isUnitMode()) {
            topicSysFlag = TopicSysFlag.buildSysFlag(false, true);
        }
        String newTopic = MixAll.getRetryTopic(data.getGroupName());
        this.brokerController.getTopicConfigManager().createTopicInSendMessageBackMethod(
            newTopic,
            subscriptionGroupConfig.getRetryQueueNums(),
            PermName.PERM_WRITE | PermName.PERM_READ, topicSysFlag);
    }
	//注册消费者
    boolean changed = this.brokerController.getConsumerManager().registerConsumer(
        data.getGroupName(),
        clientChannelInfo,
        data.getConsumeType(),
        data.getMessageModel(),
        data.getConsumeFromWhere(),
        data.getSubscriptionDataSet(),
        isNotifyConsumerIdsChangedEnable
    );

    if (changed) {
        log.info("registerConsumer info changed {} {}",
            data.toString(),
            RemotingHelper.parseChannelRemoteAddr(ctx.channel())
        );
    }
}

我们可以看到，broker会根据consumer放过来的消息，获取自己这边记录的消费者订阅的信息，注意，获取时是按照消费组获取的，我们看下registerConsumer：

//根据消费组获取消费者信息
ConsumerGroupInfo consumerGroupInfo = this.consumerTable.get(group);
if (null == consumerGroupInfo) {
    ConsumerGroupInfo tmp = new ConsumerGroupInfo(group, consumeType, messageModel, consumeFromWhere);
	//注意这里，这里consumerTable的键就是group
    ConsumerGroupInfo prev = this.consumerTable.putIfAbsent(group, tmp);
    consumerGroupInfo = prev != null ? prev : tmp;
}
boolean r1 =
    consumerGroupInfo.updateChannel(clientChannelInfo, consumeType, messageModel,
        consumeFromWhere);
boolean r2 = consumerGroupInfo.updateSubscription(subList);
if (r1 || r2) {
    if (isNotifyConsumerIdsChangedEnable) {
        this.consumerIdsChangeListener.handle(ConsumerGroupEvent.CHANGE, group, consumerGroupInfo.getAllChannel());
    }
}
this.consumerIdsChangeListener.handle(ConsumerGroupEvent.REGISTER, group, subList);
return r1 || r2;

我们注意ConsumerGroupInfo prev = this.consumerTable.putIfAbsent(group, tmp);这里，这句话告诉我们consumerTable中存放的消费者信息是按照消费组来的，那么一个组的消费信息如果不一样，按照我们的例子中，则订阅了TOPICA的消费者心跳信息告诉Broker：我们组订阅的是TOPICA！然后Broker就记录下来了。过了一会订阅了TOPICB的消费者心跳信息高速Broker：我们订阅的是TOPICB！

这里就导致了订阅消息相互覆盖，那么拉取消息时，肯定有一个消费者没法拉到消息，因为Broker上查询不到订阅信息。

至此我们就知道了导致上述现象的原因。