[ISSUE #8933] feat: DefaultPullConsumer add balance switch (#8934)

Author: humkum

client/src/main/java/org/apache/rocketmq/client/consumer/DefaultMQPullConsumer.java

@@ -88,6 +88,8 @@ public class DefaultMQPullConsumer extends ClientConfig implements MQPullConsume
 
     private int maxReconsumeTimes = 16;
 
+    private boolean enableRebalance = true;
+
     public DefaultMQPullConsumer() {
         this(MixAll.DEFAULT_CONSUMER_GROUP, null);
     }
@@ -468,4 +470,12 @@ public void setMaxReconsumeTimes(final int maxReconsumeTimes) {
     public void persist(MessageQueue mq) {
         this.getOffsetStore().persist(queueWithNamespace(mq));
     }
+
+    public boolean isEnableRebalance() {
+        return enableRebalance;
+    }
+
+    public void setEnableRebalance(boolean enableRebalance) {
+        this.enableRebalance = enableRebalance;
+    }
 }

client/src/main/java/org/apache/rocketmq/client/impl/consumer/DefaultMQPullConsumerImpl.java

@@ -381,13 +381,20 @@ public Set<SubscriptionData> subscriptions() {
 
     @Override
     public void doRebalance() {
+        if (!defaultMQPullConsumer.isEnableRebalance()) {
+            return;
+        }
         if (this.rebalanceImpl != null) {
             this.rebalanceImpl.doRebalance(false);
         }
     }
 
     @Override
     public boolean tryRebalance() {
+        if (!defaultMQPullConsumer.isEnableRebalance()) {
+            return true;
+        }
+
         if (this.rebalanceImpl != null) {
             return this.rebalanceImpl.doRebalance(false);
         }