如何抑制打印到jmh基准内控制台的方法的输出？

Question

我使用jmh来比较在控制台上大量打印的几种方法的性能。事实上，这些都是fizzbazz问题的不同解决方案。

fizzbuzz的每个实现都是一个实现FizzBazz接口的类，并且有一个print（）方法。所有这些实现都被编译到一个单独的jar中。

我试图在我的基准测试中抑制这种方法的输出。因此，我需要在运行基准测试之前用数组替换输出流，并在测试完成后返回。

如果我在每个基准方法中执行此操作，如下所示：

@Benchmark
public void naiveTest() {
    PrintStream originalOut = System.out;
    ByteArrayOutputStream outContent = new ByteArrayOutputStream();
    System.setOut(new PrintStream(outContent));

    new NaiveFizzBuzz().print(N); // N is upper limit
    
    System.setOut(originalOut);
}

然后一切正常。但这是一堆重复的代码，我不想摆脱它。此外，此代码会中断测量，因为每次运行测试时都会调用它。

我正在尝试使用状态来完成任务。

@State(Scope.Benchmark)
public static class MyState {
    private PrintStream originalOut;

    @Setup
    public void doSetup() {
        originalOut = System.out;
        ByteArrayOutputStream outContent = new ByteArrayOutputStream();
        System.setOut(new PrintStream(outContent));
    }

    @TearDown
    public void doTearDown() {
        System.setOut(originalOut);
    }
}

我的基准如下所示：

@Benchmark
public void naiveTest() {
    new NaiveFizzBuzz().print(N);
}

// Lots of similar benchmarks here...

但它不起作用。

如何将替换输出流的所有代码放在一个地方，以便在每次测试运行之前自动执行？

Answer 1

我找到了问题的解决方案。它包括这样一个事实，即我需要将我的状态对象作为参数传递给每个基准。

@Benchmark
public void naiveTest(MyState state) {
    new NaiveFizzBuzz().print(N);
}

完全按照我的需要工作。

不幸的是，jmh的留档不是很人性化，所以我花了一段时间才找到这一点。

我做了一些研究，得出结论，可以回答我自己的问题。如果不是这样，请纠正我。然后我会编辑这个问题。