diff options
Diffstat (limited to 'integration/test/resources/play-json/jmh.sc')
-rw-r--r-- | integration/test/resources/play-json/jmh.sc | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/integration/test/resources/play-json/jmh.sc b/integration/test/resources/play-json/jmh.sc new file mode 100644 index 00000000..76ed43fc --- /dev/null +++ b/integration/test/resources/play-json/jmh.sc @@ -0,0 +1,61 @@ +import ammonite.ops._ +import mill._, scalalib._, modules._ + +trait Jmh extends ScalaModule { + + def ivyDeps = super.ivyDeps() ++ Agg(ivy"org.openjdk.jmh:jmh-core:1.19") + + def runJmh(args: String*) = T.command { + val (_, resources) = generateBenchmarkSources() + Jvm.interactiveSubprocess( + "org.openjdk.jmh.Main", + classPath = (runClasspath() ++ generatorDeps()).map(_.path) ++ + Seq(compileGeneratedSources().path, resources), + mainArgs = args, + workingDir = T.ctx.dest + ) + } + + def compileGeneratedSources = T { + val dest = T.ctx.dest + val (sourcesDir, _) = generateBenchmarkSources() + val sources = ls.rec(sourcesDir).filter(_.isFile) + %%("javac", + sources.map(_.toString), + "-cp", + (runClasspath() ++ generatorDeps()).map(_.path.toString).mkString(":"), + "-d", + dest)(wd = dest) + PathRef(dest) + } + + // returns sources and resources directories + def generateBenchmarkSources = T { + val dest = T.ctx().dest + + val sourcesDir = dest / 'jmh_sources + val resourcesDir = dest / 'jmh_resources + + rm(sourcesDir) + mkdir(sourcesDir) + rm(resourcesDir) + mkdir(resourcesDir) + + Jvm.subprocess( + "org.openjdk.jmh.generators.bytecode.JmhBytecodeGenerator", + (runClasspath() ++ generatorDeps()).map(_.path), + mainArgs = Array( + compile().classes.path, + sourcesDir, + resourcesDir, + "default" + ).map(_.toString) + ) + + (sourcesDir, resourcesDir) + } + + def generatorDeps = resolveDeps( + T { Agg(ivy"org.openjdk.jmh:jmh-generator-bytecode:1.19") } + ) +} |