[rocketmq] 05/17: [benchmark] Add clientRebalanceEnable

[ji...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

jinrongtong pushed a commit to branch 5.0.0-alpha
in repository https://gitbox.apache.org/repos/asf/rocketmq.git

commit c998f3fe68fbd7fcf993ed7c094564fa66ef4a77
Author: zhangyang21 <zh...@xiaomi.com>
AuthorDate: Thu Nov 18 15:49:13 2021 +0800

    [benchmark] Add clientRebalanceEnable
    
    Signed-off-by: zhangyang21 <zh...@xiaomi.com>
---
 .../main/java/org/apache/rocketmq/example/benchmark/Consumer.java   | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/example/src/main/java/org/apache/rocketmq/example/benchmark/Consumer.java b/example/src/main/java/org/apache/rocketmq/example/benchmark/Consumer.java
index 7d26509..046ac94 100644
--- a/example/src/main/java/org/apache/rocketmq/example/benchmark/Consumer.java
+++ b/example/src/main/java/org/apache/rocketmq/example/benchmark/Consumer.java
@@ -64,6 +64,7 @@ public class Consumer {
         final double failRate = commandLine.hasOption('r') ? Double.parseDouble(commandLine.getOptionValue('r').trim()) : 0.0;
         final boolean msgTraceEnable = commandLine.hasOption('m') && Boolean.parseBoolean(commandLine.getOptionValue('m'));
         final boolean aclEnable = commandLine.hasOption('a') && Boolean.parseBoolean(commandLine.getOptionValue('a'));
+        final boolean clientRebalanceEnable = commandLine.hasOption('c') ? Boolean.parseBoolean(commandLine.getOptionValue('c')) : true;
 
         String group = groupPrefix;
         if (Boolean.parseBoolean(isSuffixEnable)) {
@@ -132,6 +133,7 @@ public class Consumer {
         consumer.setConsumeThreadMin(threadCount);
         consumer.setConsumeThreadMax(threadCount);
         consumer.setInstanceName(Long.toString(System.currentTimeMillis()));
+        consumer.setClientRebalance(clientRebalanceEnable);
 
         if (filterType == null || expression == null) {
             consumer.subscribe(topic, "*");
@@ -218,6 +220,10 @@ public class Consumer {
         opt.setRequired(false);
         options.addOption(opt);
 
+        opt = new Option("c", "clientRebalanceEnable", true, "Client Rebalance Enable, Default: true");
+        opt.setRequired(false);
+        options.addOption(opt);
+
         return options;
     }