Merge pull request #162 from rabbitmq/initial-member-count-option

Add --initial-member-count option

Author: acogoluegnes

src/main/java/com/rabbitmq/stream/perf/StreamPerfTest.java

@@ -1,4 +1,4 @@
-// Copyright (c) 2020-2023 Broadcom. All Rights Reserved.
+// Copyright (c) 2020-2024 Broadcom. All Rights Reserved.
 // The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
 //
 // This software, the RabbitMQ Stream Performance Testing Tool, is dual-licensed under the
@@ -243,6 +243,12 @@ public class StreamPerfTest implements Callable<Integer> {
       converter = Utils.DurationTypeConverter.class)
   private Duration maxAge;
 
+  @CommandLine.Option(
+      names = {"--initial-member-count", "-imc"},
+      description = "the number of initial members streams should have.",
+      converter = Utils.PositiveIntegerTypeConverter.class)
+  private int initialMemberCount;
+
   @CommandLine.Option(
       names = {"--leader-locator", "-ll"},
       description =
@@ -1266,6 +1272,10 @@ private void createStream(Environment environment, String stream) {
       streamCreator.maxAge(this.maxAge);
     }
 
+    if (this.initialMemberCount > 0) {
+      streamCreator.initialMemberCount(this.initialMemberCount);
+    }
+
     try {
       streamCreator.create();
     } catch (StreamException e) {