opensource/rocketmq
3
0
Fork 0
mirror of https://github.com/apache/rocketmq.git synced 2026-09-19 10:54:54 +08:00
Code Issues Packages Projects Releases Wiki Activity

[benchmark] Add clientRebalanceEnable

Browse Source 
Signed-off-by: zhangyang21 <zhangyang21@xiaomi.com>
This commit is contained in:
zhangyang21
2022-02-08 10:38:41 +08:00
committed by RongtongJin
parent a01433094b
commit c998f3fe68
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
example/src/main/java/org/apache/rocketmq/example/benchmark/Consumer.java
+6
View File
@@ -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;
}