[rocketmq] 03/04: [benchmark] Add clientRebalanceEnable

[zhangyang]

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

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

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

    [benchmark] Add clientRebalanceEnable
    
    Signed-off-by: zhangyang21 <zhangyan...@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;
     }