jmh

Author: jnape

pom.xml

@@ -58,7 +58,8 @@
         <junit-jupiter.version>5.9.0</junit-jupiter.version>
         <maven-compiler-plugin.version>3.3</maven-compiler-plugin.version>
         <maven-jar-plugin.version>3.1.1</maven-jar-plugin.version>
-        <mockito-junit-jupiter.version>4.2.0</mockito-junit-jupiter.version>
+        <mockito-junit-jupiter.version>4.10.0</mockito-junit-jupiter.version>
+        <jmh.version>1.35</jmh.version>
     </properties>
 
     <dependencies>
@@ -96,7 +97,18 @@
             <version>${mockito-junit-jupiter.version}</version>
             <scope>test</scope>
         </dependency>
-
+        <dependency>
+            <groupId>org.openjdk.jmh</groupId>
+            <artifactId>jmh-core</artifactId>
+            <version>${jmh.version}</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.openjdk.jmh</groupId>
+            <artifactId>jmh-generator-annprocess</artifactId>
+            <version>${jmh.version}</version>
+            <scope>test</scope>
+        </dependency>
     </dependencies>
 
     <build>

src/test/java/com/jnape/palatable/lambda/effect/io/fiber/benchmark/Benchmark.java

@@ -0,0 +1,63 @@
+package com.jnape.palatable.lambda.effect.io.fiber.benchmark;
+
+import org.openjdk.jmh.results.format.ResultFormatType;
+import org.openjdk.jmh.runner.Runner;
+import org.openjdk.jmh.runner.RunnerException;
+import org.openjdk.jmh.runner.options.ChainedOptionsBuilder;
+import org.openjdk.jmh.runner.options.Options;
+import org.openjdk.jmh.runner.options.OptionsBuilder;
+
+import java.io.File;
+
+import static com.jnape.palatable.lambda.functions.builtin.fn2.Filter.filter;
+import static com.jnape.palatable.lambda.functions.builtin.fn3.FoldLeft.foldLeft;
+import static java.lang.String.join;
+import static java.util.Arrays.asList;
+import static java.util.Arrays.copyOf;
+
+public final class Benchmark {
+
+    static final int OPS_PER_BENCHMARK = 1_000_000_000;
+
+    private static final String ROOT_PACKAGE_PATH = join("/", "src", "test", "java");
+    private static final String RESULTS_DIR_NAME  = "results";
+
+    public static void runBenchmarks(Class<?> benchmarkClass, String... methods) throws RunnerException {
+        new Runner(options(benchmarkClass, methods)).run();
+    }
+
+    private static Options options(Class<?> benchmarkClass, String[] methods) {
+        ChainedOptionsBuilder optionsBuilder = new OptionsBuilder()
+                .resultFormat(ResultFormatType.JSON)
+                .result(resultsFilePath(benchmarkClass));
+
+        String benchmarkName = benchmarkClass.getCanonicalName();
+        optionsBuilder = methods.length == 0
+                         ? optionsBuilder.include(benchmarkName)
+                         : foldLeft((b, m) -> b.include(benchmarkName + "." + m),
+                                    optionsBuilder,
+                                    asList(methods));
+
+        return optionsBuilder.build();
+    }
+
+    private static String resultsFilePath(Class<?> benchmarkClass) {
+        String   packageName          = benchmarkClass.getPackage().getName();
+        String[] nestedClassesAndName = benchmarkClass.getCanonicalName().substring(packageName.length() + 1).split("\\.");
+        String   packagePath          = joinPrune("/", packageName.split("\\."));
+        String   nestedClassPath      = joinPrune("/", copyOf(nestedClassesAndName, nestedClassesAndName.length - 1));
+
+        String resultsDirectoryPath = joinPrune("/", ROOT_PACKAGE_PATH, packagePath, RESULTS_DIR_NAME, nestedClassPath);
+        String fileName             = joinPrune(".", nestedClassesAndName[nestedClassesAndName.length - 1], "jmh", "json");
+        File   resultsDirectory     = new File(resultsDirectoryPath);
+        if (!(resultsDirectory.mkdirs() || resultsDirectory.isDirectory())) {
+            throw new IllegalStateException("Failed to create JMH results directory: " + resultsDirectoryPath);
+        }
+        return join("/", resultsDirectoryPath, fileName);
+    }
+
+
+    private static String joinPrune(CharSequence delimiter, CharSequence... parts) {
+        return join(delimiter, filter(s -> !s.isEmpty(), asList(parts)));
+    }
+}