fix(tools/perf): fix the perf tools await count (#2219)

Signed-off-by: Robin Han <[email protected]>
AutoMQ · Dec 12, 2024 · 0265f44 · 0265f44
1 parent 1666892
commit 0265f44
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tools/src/main/java/org/apache/kafka/tools/automq/PerfCommand.java b/tools/src/main/java/org/apache/kafka/tools/automq/PerfCommand.java
@@ -179,11 +179,12 @@ private void waitTopicsReady(boolean hasConsumer) {
     private void waitTopicsReadyWithConsumer() {
         long start = System.nanoTime();
         boolean ready = false;
+        int expectPartitionCount = config.topics * config.partitionsPerTopic;
         while (System.nanoTime() < start + TOPIC_READY_TIMEOUT_NANOS) {
-            int sent = producerService.probe();
+            producerService.probe();
             int received = readyPartitions.size();
-            LOGGER.info("Waiting for topics to be ready... sent: {}, received: {}", sent, received);
-            if (received >= sent) {
+            LOGGER.info("Waiting for topics to be ready... sent: {}, received: {}", expectPartitionCount, received);
+            if (received >= expectPartitionCount) {
                 ready = true;
                 break;
             }