add deserialize ByteBuffer benchmark

apache · Dec 25, 2024 · 1b0cecd · 1b0cecd
1 parent e21b864
commit 1b0cecd
Showing 1 changed file with 18 additions and 2 deletions.
diff --git a/...o-benchmarks/src/test/java/org/apache/paimon/benchmark/bitmap/RoaringBitmapBenchmark.java b/...o-benchmarks/src/test/java/org/apache/paimon/benchmark/bitmap/RoaringBitmapBenchmark.java
@@ -30,6 +30,7 @@
 import java.io.File;
 import java.io.FileOutputStream;
 import java.io.IOException;
+import java.nio.ByteBuffer;
 import java.nio.file.Path;
 import java.util.Random;
 
@@ -65,7 +66,7 @@ public void testDeserialize() throws Exception {
                         .setOutputPerIteration(true);
 
         benchmark.addCase(
-                "without-buffer",
+                "deserialize(DataInput)",
                 10,
                 () -> {
                     try (LocalFileIO.LocalSeekableInputStream seekableStream =
@@ -78,7 +79,7 @@ public void testDeserialize() throws Exception {
                 });
 
         benchmark.addCase(
-                "with-buffer",
+                "deserialize(DataInput, byte[])",
                 10,
                 () -> {
                     try (LocalFileIO.LocalSeekableInputStream seekableStream =
@@ -90,6 +91,21 @@ public void testDeserialize() throws Exception {
                     }
                 });
 
+        benchmark.addCase(
+                "deserialize(ByteBuffer)",
+                10,
+                () -> {
+                    try (LocalFileIO.LocalSeekableInputStream seekableStream =
+                                    new LocalFileIO.LocalSeekableInputStream(file);
+                            DataInputStream input = new DataInputStream(seekableStream)) {
+                        byte[] bytes = new byte[(int) file.length()];
+                        input.readFully(bytes);
+                        new RoaringBitmap().deserialize(ByteBuffer.wrap(bytes));
+                    } catch (IOException e) {
+                        throw new RuntimeException(e);
+                    }
+                });
+
         benchmark.run();
     }
 }