RocketMQ重平衡策略你用过几种？

川长思鸟来 2024-03-25 19:27 104阅读 0赞

### 前言 ###

我们简单讲述了`RocketMQ`消费者是如何实现重平衡的，我们在源码中发现默认的重平衡策略是平均分配策略`AllocateMessageQueueAveragely`，另外我们还可以设置其他的重平衡策略，你知道有哪几种嘛？你用过其中的哪些分配策略呢？

### AllocateMessageQueueAveragely【平均分配】 ###

在上一篇文章中，我们简单说明了`平均分配`的结果，下面我们根据源码来看一下：

@Override
        public List<MessageQueue> allocate(String consumerGroup, String currentCID, List<MessageQueue> mqAll,
            List<String> cidAll) {
    
            List<MessageQueue> result = new ArrayList<MessageQueue>();
            // 检查参数，不合理的参数就不做分配了
            if (!check(consumerGroup, currentCID, mqAll, cidAll)) {
                return result;
            }
            // cidAll和mqAll已经排序好了，保证所有消费者实例拿到的顺序一致；
            // 获取当前消费者ID在cidAll中的索引位置
            int index = cidAll.indexOf(currentCID);
            // 计算取模结果
            int mod = mqAll.size() % cidAll.size();
            // 计算每隔消费者平均分配到的数量
            int averageSize =
                mqAll.size() <= cidAll.size() ? 1 : (mod > 0 && index < mod ? mqAll.size() / cidAll.size()
                    + 1 : mqAll.size() / cidAll.size());
            // 计算起始索引值
            int startIndex = (mod > 0 && index < mod) ? index * averageSize : index * averageSize + mod;
            // 计算遍历此时
            int range = Math.min(averageSize, mqAll.size() - startIndex);
            for (int i = 0; i < range; i++) {
                // 计算目标MessageQueue，并添加到结果集中
                result.add(mqAll.get((startIndex + i) % mqAll.size()));
            }
            // 返回结果集
            return result;
        }
    复制代码

举个例子说明一下：

假如有7个`MessageQueue`，3个消费者，分配结果如下：

> 1.消费者C1分配到的MessageQueue索引值为\[0,1,2\]；
> 
> 2.消费者C2分配到的MessageQueue索引值为\[3,4\]；
> 
> 3.消费者C3分配到的MessageQueue索引值为\[5,6\]；

![format_png][]

如果说`MessageQueue`数量小于消费者的数量，比如当前`MessageQueue`数量为2，消费者数量为3，那么第三个消费者是不会被分配`MessageQueue`的，只有前两个消费者各1个`MessageQueue`；

### AllocateMessageQueueAveragelyByCircle【平均交叉分配】 ###

这个`平均交叉`的分配策略实现方式更简单：

[format_png]: https://image.dandelioncloud.cn/pgy_files/images/2024/03/25/e071924794ed48188331f39442aa804d.png