KAFKA-12895: Add benchmark

apache · Oct 1, 2024 · a7ebc6c · a7ebc6c
1 parent ab266bc
commit a7ebc6c
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/jmh-benchmarks/src/main/java/org/apache/kafka/jmh/util/ConcurrentMapBenchmark.java b/jmh-benchmarks/src/main/java/org/apache/kafka/jmh/util/ConcurrentMapBenchmark.java
@@ -189,4 +189,23 @@ public void testCopyOnWriteMapEntrySet(Blackhole blackhole) {
             }
         }
     }
+
+    @Benchmark
+    @OperationsPerInvocation(TIMES)
+    public void testConcurrentHashMapComputeIfAbsentReadOnly(Blackhole blackhole) {
+        for (int i = 0; i < TIMES; i++) {
+            blackhole.consume(concurrentHashMap.computeIfAbsent(
+                    ThreadLocalRandom.current().nextInt(0, mapSize),
+                    newValue -> Integer.MAX_VALUE
+            ));
+        }
+    }
+
+    @Benchmark
+    @OperationsPerInvocation(TIMES)
+    public void testConcurrentHashMapGetReadOnly(Blackhole blackhole) {
+        for (int i = 0; i < TIMES; i++) {
+            blackhole.consume(concurrentHashMap.get(ThreadLocalRandom.current().nextInt(0, mapSize)));
+        }
+    }
 }