java - JMH Benchmarking - 比较替代实现运行时间的简洁方法

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”列，因此您知道哪个结果对应哪个值。