blob: 76ed43fc12205a24c065082670a0b5045cca245a (
plain) (
tree)
|
|
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") }
)
}
|