From ba0b9d2739d446fb96fe9ebefed834428b821572 Mon Sep 17 00:00:00 2001
From: Jorge Esteban Quilcate Otoya <jorge.quilcate@aiven.io>
Date: Fri, 13 Oct 2023 12:53:47 +0300
Subject: [PATCH] fixup! feat: add benchmarks

---
 .../benchs/transform/DetransformBench.java        | 15 +++++++--------
 1 file changed, 7 insertions(+), 8 deletions(-)

diff --git a/benchmarks/src/main/java/io/aiven/kafka/tieredstorage/benchs/transform/DetransformBench.java b/benchmarks/src/main/java/io/aiven/kafka/tieredstorage/benchs/transform/DetransformBench.java
index b42615702..e4b3f249b 100644
--- a/benchmarks/src/main/java/io/aiven/kafka/tieredstorage/benchs/transform/DetransformBench.java
+++ b/benchmarks/src/main/java/io/aiven/kafka/tieredstorage/benchs/transform/DetransformBench.java
@@ -22,7 +22,6 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.security.SecureRandom;
-import java.util.List;
 import java.util.concurrent.TimeUnit;
 
 import io.aiven.kafka.tieredstorage.manifest.index.ChunkIndex;
@@ -57,16 +56,16 @@
 
 @State(Scope.Benchmark)
 @Fork(value = 1)
-@Warmup(iterations = 5)
-@Measurement(iterations = 10)
+@Warmup(iterations = 4)
+@Measurement(iterations = 16)
 @BenchmarkMode({Mode.Throughput, Mode.SampleTime})
 @OutputTimeUnit(TimeUnit.MILLISECONDS)
 public class DetransformBench extends AesKeyAware {
     static Path segmentPath;
-    @Param({"52428800"})
-    public int contentLength; // 500MiB
-    @Param({"102400", "1048576", "5242880", "10485760"})
-    public int chunkSize; // 100KiB, 1MiB, 5MiB, 10MiB
+    @Param({"5242880", "10485760", "52428800"})
+    public int contentLength; // 5MiB, 10MiB, 50MiB
+    @Param({"10240", "102400", "524288", "1048576", "5242880", "10485760"})
+    public int chunkSize; // 10KiB, 100KiB, 512KiB, 1MiB, 5MiB, 10MiB
     @Param({"false"})
     public boolean compression;
     @Param({"false", "true"})
@@ -114,7 +113,7 @@ public void teardown() throws IOException {
     public int test() throws IOException {
         // Detransform.
         DetransformChunkEnumeration detransformEnum = new BaseDetransformChunkEnumeration(
-            new ByteArrayInputStream(uploadedData), List.of(chunkIndex.chunks().iterator().next()));
+            new ByteArrayInputStream(uploadedData), chunkIndex.chunks());
         if (encryption) {
             detransformEnum = new DecryptionChunkEnumeration(
                 detransformEnum, ivSize, AesKeyAware::decryptionCipherSupplier);