From 8c6adb481aa04862acdcf0a7b72bdc4d4d5060a3 Mon Sep 17 00:00:00 2001
From: Trask Stalnaker <trask.stalnaker@gmail.com>
Date: Tue, 27 Jul 2021 09:40:53 -0700
Subject: [PATCH] Expose some JMH configuration (#3679)

---
 benchmark-overhead-jmh/build.gradle.kts | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/benchmark-overhead-jmh/build.gradle.kts b/benchmark-overhead-jmh/build.gradle.kts
index a5ba2aa77e..d995d50341 100644
--- a/benchmark-overhead-jmh/build.gradle.kts
+++ b/benchmark-overhead-jmh/build.gradle.kts
@@ -29,6 +29,10 @@ tasks {
     }
   }
 
+  // TODO(trask) move to otel.jmh-conventions?
+  val jmhFork = gradle.startParameter.projectProperties["jmh.fork"]?.toInt()
+  val jmhIncludes = gradle.startParameter.projectProperties["jmh.includes"]
+
   named<JMHTask>("jmh") {
     val shadowTask = project(":javaagent").tasks.named<ShadowJar>("shadowJar").get()
     inputs.files(layout.files(shadowTask))
@@ -43,6 +47,13 @@ tasks {
     // see https://github.com/melix/jmh-gradle-plugin/issues/200
     jvmArgsPrepend.add(args.joinToString(" "))
 
+    if (jmhIncludes != null) {
+      includes.addAll(jmhIncludes.split(','))
+    }
+    if (jmhFork != null) {
+      fork.set(jmhFork)
+    }
+
     // TODO(trask) is this ok? if it's ok, move to otel.jmh-conventions?
     outputs.upToDateWhen { false }
   }