From b7aeeb04c45b95eb72513843297a5832df205a6a Mon Sep 17 00:00:00 2001 From: David Ho Date: Wed, 13 May 2026 11:41:49 -0700 Subject: [PATCH 1/2] Add method to return shard classes from BenchmarkRunner --- .../awssdk/benchmark/BenchmarkRunner.java | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java b/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java index 6cef787fef3d..5be32bc6b44f 100644 --- a/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java +++ b/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java @@ -27,6 +27,7 @@ import java.util.Collection; import java.util.List; import java.util.stream.Collectors; +import java.util.stream.Stream; import org.apache.commons.cli.CommandLine; import org.apache.commons.cli.CommandLineParser; import org.apache.commons.cli.DefaultParser; @@ -109,6 +110,21 @@ public class BenchmarkRunner { private static final Logger log = Logger.loggerFor(BenchmarkRunner.class); + static List getShardClasses(String shard) { + switch (shard) { + case "sync": return SYNC_BENCHMARKS; + case "async": return ASYNC_BENCHMARKS; + case "protocol": return PROTOCOL_BENCHMARKS; + case "coldStart": return COLD_START_BENCHMARKS; + case "metrics": return Stream.concat(METRIC_BENCHMARKS.stream(), METRIC_PUBLISHER_BENCHMARKS.stream()) + .collect(Collectors.toList()); + case "all": return Stream.of(SYNC_BENCHMARKS, ASYNC_BENCHMARKS, PROTOCOL_BENCHMARKS, + COLD_START_BENCHMARKS, METRIC_BENCHMARKS, METRIC_PUBLISHER_BENCHMARKS) + .flatMap(List::stream).collect(Collectors.toList()); + default: throw new IllegalArgumentException("Unknown shard: " + shard); + } + } + private final List benchmarksToRun; private final BenchmarkResultProcessor resultProcessor; private final BenchmarkRunnerOptions options; @@ -120,6 +136,11 @@ private BenchmarkRunner(List benchmarksToRun, BenchmarkRunnerOptions opt } public static void main(String... args) throws Exception { + if (args.length >= 2 && args[0].equals("--list")) { + System.out.println(String.join("|", getShardClasses(args[1]))); + return; + } + List benchmarksToRun = new ArrayList<>(); benchmarksToRun.addAll(SYNC_BENCHMARKS); benchmarksToRun.addAll(ASYNC_BENCHMARKS); From 3dc0573bab8380b74a26b13efec47c01723f43db Mon Sep 17 00:00:00 2001 From: David Ho Date: Wed, 13 May 2026 13:02:05 -0700 Subject: [PATCH 2/2] Add method to return shard classes from BenchmarkRunner --- .../awssdk/benchmark/BenchmarkRunner.java | 22 ++++++++++--------- 1 file changed, 12 insertions(+), 10 deletions(-) diff --git a/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java b/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java index 5be32bc6b44f..0b28b61a0e4e 100644 --- a/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java +++ b/test/sdk-benchmarks/src/main/java/software/amazon/awssdk/benchmark/BenchmarkRunner.java @@ -110,6 +110,16 @@ public class BenchmarkRunner { private static final Logger log = Logger.loggerFor(BenchmarkRunner.class); + private final List benchmarksToRun; + private final BenchmarkResultProcessor resultProcessor; + private final BenchmarkRunnerOptions options; + + private BenchmarkRunner(List benchmarksToRun, BenchmarkRunnerOptions options) { + this.benchmarksToRun = benchmarksToRun; + this.resultProcessor = new BenchmarkResultProcessor(); + this.options = options; + } + static List getShardClasses(String shard) { switch (shard) { case "sync": return SYNC_BENCHMARKS; @@ -125,19 +135,11 @@ static List getShardClasses(String shard) { } } - private final List benchmarksToRun; - private final BenchmarkResultProcessor resultProcessor; - private final BenchmarkRunnerOptions options; - - private BenchmarkRunner(List benchmarksToRun, BenchmarkRunnerOptions options) { - this.benchmarksToRun = benchmarksToRun; - this.resultProcessor = new BenchmarkResultProcessor(); - this.options = options; - } - public static void main(String... args) throws Exception { if (args.length >= 2 && args[0].equals("--list")) { + // CHECKSTYLE:OFF System.out.println(String.join("|", getShardClasses(args[1]))); + // CHECKSTYLE:ON return; }