更准确的测试Java程序性能——JMH基准测试
什么是JMH
JMH，即Java Microbenchmark Harness，Java平台下的一套微基准测试工具。如果我们需要测试API性能的话，就可以用上这个工具，所以它并不是取代单元测试的。它可以在开发阶段提供性能参考标准，不过这并不代表线上的性能表现，不同的硬件和操作系统也会带来性能差异，所以最终还是需要上到测试或沙箱环境，让测试人员进行压测。
为什么需要JMH
在了解JMH之前，如果需要性能测试，我们通常会使用for循环，或者JMeter。而JMH正是比for循环严谨，比JMeter使用简单的测试工具。
再者，不知道你注意过没有，在使用for循环测试时，第一次或者头几次运行总是最慢的，越到后面越快。从《计算机组成与设计 硬件软件接口》一书中可以了解到，从更底层讲，Java是解释型的语言。虽然Java也需要编译，但是编译后只是字节码，还需要JVM解释成对应宿主机的机器码。解释的优势是可移植性，但是性能较差。在20世纪80和90年代，虽然解释型语言的性能也飞速提升，但是与C语言相比，仍有10倍的性能差距。
为了保持可移植性，同时又提高性能，Java便开发了即时编译器（Just In Time complier），其通过记录运行的程序来找到所谓的“热点”方法，然后将它们直接编译成宿主机的指令序列，即不通过JVM解释那一层。这样以后该方法的运行就会更快。
看到这里也就明白了，为什么程序越到后面就会越快。JMH在真正的测试之前会预热程序，而且还可以通过配置进程数、线程数等参数来使程序更接近实际的运行状况。
如何使用
首先引入Maven依赖：
<dependency>
<groupId>org.openjdk.jmh</groupId>
<artifactId>jmh-core</artifactId>
<version>1.21</version>
</dependency>
<dependency>
<groupId>org.openjdk.jmh</groupId>
<artifactId>jmh-generator-annprocess</artifactId>
<version>1.21</version>
<scope>test</scope>
</dependency>
本案例中，我写了一个简单的小程序，它会从指定目录读取文件夹内容（每行一个数字），然后会对取出来的数字进行排序。排序算法选择了插入排序和归并排序，我们通过基准测试来看看两者的性能差距。
读取文件内容
public class ReadFile {
public static int[] readInteger(String path){
try(BufferedReader in = new BufferedReader(new FileReader(path));) {
List<Integer> temp = new ArrayList<>();
String str;
while ((str = in.readLine()) != null) {
temp.add(Integer.parseInt(str));
}
int[] result = new int[temp.size()];
for(int i=0;i<temp.size();i++){
result[i]=temp.get(i);
}
return result;
} catch (Exception e) {
e.printStackTrace();
return new int[0];
}
}
}
两个排序算法就不贴了，网上可以搜到很多。实际的开发可能会用上SpringBoot，所以还得与Junit整合，并使用自动注入功能。先直接贴上测试代码：
@BenchmarkMode(Mode.All)
@Warmup(iterations = 3)//预热轮数
@Measurement(iterations = 1, time = 1, timeUnit = TimeUnit.SECONDS)
@Threads(8)//线程数
@Fork(0) //fork的次数，如果想用Autowired自动注入，这个填0
@OutputTimeUnit(TimeUnit.MILLISECONDS)
@State(Scope.Benchmark)
@RunWith(SpringRunner.class) //整合SpringBoot的测试运行环境
@SpringBootTest
public class JHMTest {
//想用自动注入功能，对象必须是静态的，fork填0
private static SortService service;
@Autowired
void setService(SortService service){
JHMTest.service =service;
}
@Test
public void executeBenchmark() throws RunnerException {
//JMH的选项配置，除了上面的注解方式的配置，也可以直接在这个Options里面配置。
//其中/Users/xxxx/Desktop/Benchmark.json是结果的输出文件
Options options = new OptionsBuilder().include(this.getClass().getSimpleName())
.output("/Users/xxxx/Desktop/Benchmark.json").build();
new Runner(options).run();
}
@Benchmark
public void insertSortTest(){
int[] arr = ReadFile.readInteger("/Users/xxxxx/Desktop/test.txt");
service.insertSort(arr);
}
@Benchmark
public void mergeSortTest(){
int[] arr = ReadFile.readInteger("/Users/xxxxx/Desktop/test 2.txt");
service.mergeSort(arr);
}
}
上面注释简单写了几个关键点，我们执行executeBenchmark方法，JMH就会执行该类下带有Benchmark注解的方法。最终结果会输出到指定文件中。
其他注解的解释可见图
截图截取自：https://dunwu.github.io/javatech/test/jmh.html#jmh-api
结果查看
打开结果文件，前面一大坨是系统信息，可以简单看看，直接拉到最后，结果如下：
Benchmark Mode Cnt Score Error Units
JHMTest.insertSortTest thrpt 129.302 ops/ms
JHMTest.mergeSortTest thrpt 122.224 ops/ms
JHMTest.insertSortTest avgt 0.065 ms/op
JHMTest.mergeSortTest avgt 0.066 ms/op
JHMTest.insertSortTest sample 122410 0.066 ± 0.002 ms/op
JHMTest.insertSortTest:insertSortTest·p0.00 sample 0.014 ms/op
JHMTest.insertSortTest:insertSortTest·p0.50 sample 0.050 ms/op
JHMTest.insertSortTest:insertSortTest·p0.90 sample 0.106 ms/op
JHMTest.insertSortTest:insertSortTest·p0.95 sample 0.120 ms/op
JHMTest.insertSortTest:insertSortTest·p0.99 sample 0.192 ms/op
JHMTest.insertSortTest:insertSortTest·p0.999 sample 0.492 ms/op
JHMTest.insertSortTest:insertSortTest·p0.9999 sample 11.891 ms/op
JHMTest.insertSortTest:insertSortTest·p1.00 sample 17.334 ms/op
JHMTest.mergeSortTest sample 122055 0.066 ± 0.002 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.00 sample 0.014 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.50 sample 0.050 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.90 sample 0.107 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.95 sample 0.121 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.99 sample 0.187 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.999 sample 0.457 ms/op
JHMTest.mergeSortTest:mergeSortTest·p0.9999 sample 11.957 ms/op
JHMTest.mergeSortTest:mergeSortTest·p1.00 sample 12.419 ms/op
JHMTest.insertSortTest ss 0.020 ms/op
JHMTest.mergeSortTest ss 0.020 ms/op
结果如上，Mode中thrpt代表吞吐量，单位时间内的执行次数。avgt是平均时间，一次执行需要的单位时间。sample是基于采样的执行时间，采样频率由JMH自动控制。ss是单次执行的时间。
从结果上看，两种排序算法的性能相差无几，当然与我们的逻辑太简单也有关系。这次的分享就到这里，大家赶紧用到自己的项目中，测试一下吧。
