java - JMH 基准测试 - 比较替代实现的运行时间的简洁方法

Question

我有同一个自定义类的两个实现:

private List<String> a = Util.myCustomClass();
private List<String> b = Util2.myCustomClass();

并且想比较它们每个函数的运行时间(它们都具有相同的名称)。目前，我的基准测试如下:

@Benchmark
public boolean contains_val_a() {
   return a.contains(val);
}

@Benchmark
public boolean contains_val_b() {
  return b.contains(val);
}

我对大约 25 个不同的函数重复这个并行结构(因为有两种实现，每个函数都写了两次)。有没有办法让我只编写 25 个 @Benchmark 函数并让 jmh 为这两个实现运行每个函数？

Answer 1

您可以使用 @Param 定义要加载的(字符串)类，并让它在 @Setup 方法中初始化正在测试的类，如下所示在此示例中:https://github.com/ktoso/sbt-jmh/blob/master/src/sbt-test/sbt-jmh/jmh-run/src/main/scala/org/openjdk/jmh/samples/JMHSample_27_Params.scala

本质上

@Param(Array("a", "b"))
val name: String = ""

var thing: CommonInterface = ""

@Setup
def setup(): Unit = name match {
  case "a" => new A
  case "b" => new B
}

然后，JMH 将在测试结果中包含一个“param”列，以便您知道哪个结果对应哪个值。