打印jmh基准测试的附加输出

Question

我目前有一个JMH基准测试来衡量所有实现相同接口的各种数据结构的性能。基准测试工作正常，但我想为每个基准测试打印一些额外的信息，这些信息描述了我的数据结构在试验前后的状态。

目前，我正在做下面的代码

@Param({"impl0", "impl1", "impl2"})
String dataStructureImplementation;

DataStructureInterface dataStructure;

List<String> inputData;

@Setup(Level.Trial)
public void setUp() {
    switch (dataStructureImplementation) {
        case "impl0":
            dataStructure = new DataStructureImplementation0();
            break;
        case "impl1":
            dataStructure = new DataStructureImplementation1();
            break;
        case "impl2":
            dataStructure = new DataStructureImplementation2();
            break;
        default:
            throw new IllegalArgumentException();
    }
    setUpDataStructure();
    inputData = readInputDataFromFile();
    System.out.println(dataStructure.getStateRepresentation());
}

@TearDown(Level.Trial)
public void tearDown() {
    System.out.println(dataStructure.getStateRepresentation());
}

@Benchmark
@OperationsPerInvocation(1000000)
public void queryDataStructure(Blackhole bh) {
    for (String datum : inputData)
        bh.consume(dataStructure.query(datum));
}

问题是系统的输出。出来println与JMH的输出混淆。有没有办法避免这种情况，并在JMH完成其试验报告后运行我的打印声明？另一种方法是将其写入一个文件，然后进行合并，但最好得到一个不需要手动合并的报告。

Answer 1

没有开箱即用的方法可以从@Setup和@Teardown方法积累一些自定义格式的输出并将其附加到最终报告中。

一种方法是将所需的输出附加到一个文件中，并在所有基准测试完成后打印其内容。

不过，您需要使用Runner类来运行测试。

@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MILLISECONDS)
public class BenchmarkTestWithMoreOutput {

    public static void main(String[] args) throws RunnerException {

        // This is just a very basic setup just to get an idea
        Options opt = new OptionsBuilder()
                .include(BenchmarkTestWithMoreOutput.class.getSimpleName())
                .forks(2)
                .build();

        Collection<RunResult> results = new Runner(opt).run();

        // Now print your file content. 
        // It will be printed after the default JMH report
   }

    @Benchmark
    public int benchmarkMethod() {
        // code to benchmark
    }
}

RunResult对象包含来自迭代的信息。您可以将其与自定义输出混合并打印一些漂亮的报告。