From 30d52c6b18a7a262e9e69eea4465766ed69629d0 Mon Sep 17 00:00:00 2001 From: Ning Yu Date: Mon, 3 Jun 2024 10:19:24 +0800 Subject: [PATCH] fix(perf): increase request timeout of the admin client Signed-off-by: Ning Yu --- .../org/apache/kafka/tools/automq/perf/ConsumerService.java | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tools/src/main/java/org/apache/kafka/tools/automq/perf/ConsumerService.java b/tools/src/main/java/org/apache/kafka/tools/automq/perf/ConsumerService.java index 99d5bef116..4af15d11b8 100644 --- a/tools/src/main/java/org/apache/kafka/tools/automq/perf/ConsumerService.java +++ b/tools/src/main/java/org/apache/kafka/tools/automq/perf/ConsumerService.java @@ -24,6 +24,7 @@ import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.Future; +import java.util.concurrent.TimeUnit; import java.util.concurrent.atomic.AtomicLong; import java.util.stream.Collectors; import java.util.stream.Stream; @@ -59,7 +60,10 @@ public class ConsumerService implements AutoCloseable { private final String groupSuffix; public ConsumerService(String bootstrapServer) { - this.admin = Admin.create(Map.of(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG, bootstrapServer)); + Properties properties = new Properties(); + properties.put(ConsumerConfig.BOOTSTRAP_SERVERS_CONFIG, bootstrapServer); + properties.put(ConsumerConfig.REQUEST_TIMEOUT_MS_CONFIG, (int) TimeUnit.SECONDS.toMillis(300)); + this.admin = Admin.create(properties); this.groupSuffix = new SimpleDateFormat("HHmmss").format(System.currentTimeMillis()); }