如何与sbt一起使用JMH for Scala基准测试?

时间:2014-04-11 15:27:27

标签: scala sbt microbenchmark jmh

我尝试将一起使用,但到目前为止,我还没有设法正确设置它,以便基于.scala的基准测试工作。

由于基于 sbt + .java的基准组合起作用,我试图从该基础开始。 我正在使用sbt 0.13.1。


使用sbt 的

.java基准测试

build.sbt

import AssemblyKeys._

name := "scala-benchmark"

version := "1.0"

scalaVersion := "2.10.3"

scalacOptions += "-deprecation"

libraryDependencies += "org.openjdk.jmh" % "jmh-core" % "0.5.5"

libraryDependencies += "org.openjdk.jmh" % "jmh-java-benchmark-archetype" % "0.5.5"

libraryDependencies += "org.openjdk.jmh" % "jmh-generator-annprocess" % "0.5.5"

libraryDependencies += "org.openjdk.jmh" % "jmh-generator-bytecode" % "0.5.5"

assemblySettings

jarName in assembly := "microbenchmarks.jar"

test in assembly := {}

mainClass in assembly := Some("org.openjdk.jmh.Main")

获得单身"脂肪" jar最后, sbt-assembly 插件是必需的:

项目/ assembly.sbt

addSbtPlugin("com.eed3si9n" % "sbt-assembly" % "0.11.2")

一个简单的基准:

的src /主/ JAVA /应用程序/基准/ JAVA / benchmark2 / Benchmark2.java

package app.benchmark.java.benchmark2;

import org.openjdk.jmh.annotations.GenerateMicroBenchmark;
import org.openjdk.jmh.runner.Runner;
import org.openjdk.jmh.runner.RunnerException;
import org.openjdk.jmh.runner.options.Options;
import org.openjdk.jmh.runner.options.OptionsBuilder;

public class Benchmark2 {
    @GenerateMicroBenchmark
    public int run() {
        int result = 0;
        for (int i = 0; i < 10; i++) {
            result += i * i;
        }
        return result;
    }
}

运行sbt assembly会显示此输出:

  

$ sbt assembly
  [...]
  [info]编译2个Scala源和2个Java源到...
  [警告]警告:支持的源版本&#39; RELEASE_6&#39;来自注释处理器&org.openjdk.jmh.generators.GenerateMicroBenchmarkProcessor&#39;少于-source&#39; 1.8&#39;
  [警告] 1警告
  [info]包括:jmh-java-benchmark-archetype-0.5.5.jar
  [info]包括:jmh-generator-bytecode-0.5.5.jar
  [info]包括:jopt-simple-4.6.jar
  [info]包括:jmh-generator-reflection-0.5.5.jar
  [info]包括:jmh-generator-annprocess-0.5.5.jar
  [info]包括:asm-4.2.jar
  [info]包括:commons-math3-3.2.jar
  [info]包括:jmh-core-0.5.5.jar
  [info]包括:scala-library.jar
  [...]
  [info]包装/home/scala-2.10/vc/rhaag/scala/scala-benchmark/target/scala-2.10/microbenchmarks.jar ...

,生成的microbenchmarks.jar包含运行基准测试所需的所有内容:

$ java -jar target/scala-2.10/microbenchmarks.jar -wi 3 -i 3 -f 1 app.benchmark.java.benchmark2.Benchmark2.run 

[...] 

Benchmark                  Mode   Samples         Mean   Mean error    Units 

a.b.j.b.Benchmark2.run    thrpt         3   607555,968    70243,275   ops/ms 

到目前为止一切顺利。


使用sbt

的Scala基准测试

从那个基础我尝试切换到基于.scala的基准:

build.sbt

用Scala替换Java原型

libraryDependencies += "org.openjdk.jmh" % "jmh-scala-benchmark-archetype" % "0.5.5"

不起作用,因为下载失败。

这有效:

libraryDependencies += "org.openjdk.jmh" % "jmh-scala-benchmark-archetype" % "0.5.5" from "http://repo1.maven.org/maven2/org/openjdk/jmh/jmh-scala-benchmark-archetype/0.5.5/jmh-scala-benchmark-archetype-0.5.5.jar"

另一个简单的基准:

的src /主/阶/应用程序/基准/阶/ benchmark2 / Benchmark2.scala

package app.benchmark.scala.benchmark2

import org.openjdk.jmh.annotations.GenerateMicroBenchmark
import org.openjdk.jmh.runner.Runner
import org.openjdk.jmh.runner.RunnerException
import org.openjdk.jmh.runner.options.Options
import org.openjdk.jmh.runner.options.OptionsBuilder

class Benchmark2 {
  @GenerateMicroBenchmark
  def run() = {
    Seq.range(0, 10).map(i => i * i).sum
  }
}

现在sbt assembly创建了jar文件,但是target/scala-2.10/microbenchmarks.jar#META-INF/MicroBenchmarks没有列出Scala基准,java -jar target/scala-2.10/microbenchmarks.jar -l也没有显示这些基准。


资源:

如何为Scala集成(基于字节码的)JMH处理器? 或者从另一个角度来看:为什么(基于注释的)JMH处理器会自动获取并生成基于Java的基准测试?

1 个答案:

答案 0 :(得分:10)

我已经实现了一个实际可行的 sbt-jmh 插件:https://github.com/ktoso/sbt-jmh

目前不支持构建benchmarks.jar,但您只需键入run -i 10 .*MyBenchmark.*即可按预期工作(为您执行所有多步编译)。

我希望这会有所帮助,欢呼!