在实践过程中,我们经常需要记录一个任务执行的耗时,这是评价代码好坏,评测代码性能,排查业务执行问题的重要操作。那么,你是如何来获取并计算任务执行耗时的呢?通过new Date获得时间进行换算?还是有更好的方案?本文给你答案。
获取任务耗时通常做法
获取任务耗时,最简单的方式就是打印当前时间与任务开始执行时间的差值,实例代码如下:
@Test
public void testElapsedTimes() throws InterruptedException {
long startTime = new Date().getTime();
// do something
Thread.sleep(1000);
System.out.println("执行耗时:" + (new Date().getTime() - startTime) + "ms");
}
复制
上述方式实现简单,逻辑也比较直观。但如果执行大量测试,测试中还有不同的代码逻辑块,那么需要改动的地方就比较多。
改进做法
在上述代码中,如果IDE安装有代码检查工具,则会提示采用System.currentTimeMillis()来获取时间,而不是new Date().getTime()的方式。
改造之后,实现代码如下:
@Test
public void testElapsedTimes1() throws InterruptedException {
long startTime = System.currentTimeMillis();
// do something
Thread.sleep(1000);
System.out.println("执行耗时:" + (System.currentTimeMillis() - startTime) + "ms");
}
复制
在这样的场景下(无需获取更多Date相关信息)也推荐使用System.currentTimeMillis()来获取时间戳。至于为什么,看一下Date的源码实现就知道了。
Date的构造方法:
public Date() {
this(System.currentTimeMillis());
}
复制
Date在构造时,本质上也是先获得了System.currentTimeMillis(),然后再初始化其他信息。既然我们只需要时间戳,那就没必要再构建Date对象了。从性能层面来说,能优化则优化。
Spring的StopWatch
上述两种方式虽然性能和写法有所区别,但本质是一样的。下面我们来讲讲Spring提供的StopWatch类,它不仅可实现上述功能,而且还可以做类似任务执行时间控制,也就是封装了一个对开始时间、结束时间记录操作的Java类。
先通过StopWatch类来实现一下上述功能:
@Test
public void testStopWatch() throws InterruptedException {
StopWatch sw = new StopWatch();
sw.start("开始执行业务");
// do something
Thread.sleep(1000);
sw.stop();
System.out.println(sw.getTotalTimeMillis());
}
复制
通过创建StopWatch对象,然后调用它的start、stop方法来区分执行任务区间,通过getTotalTimeMillis()方法获得总耗时。
乍一看,代码好像还比之前的方式多了,体现不出来什么优势啊!下面我们再来看一个复杂点的示例:
@Test
public void testStopWatch1() throws InterruptedException {
StopWatch sw = new StopWatch();
sw.start("起床");
Thread.sleep(1000);
sw.stop();
sw.start("洗漱");
Thread.sleep(2000);
sw.stop();
sw.start("锁门");
Thread.sleep(500);
sw.stop();
System.out.println(sw.prettyPrint());
System.out.println(sw.getTotalTimeMillis());
System.out.println(sw.getLastTaskName());
System.out.println(sw.getLastTaskInfo());
System.out.println(sw.getTaskCount());
}
复制
执行上述测试示例,打印结果如下:
StopWatch '': running time = 3509166972 ns
---------------------------------------------
ns % Task name
---------------------------------------------
1003330360 029% 起床
2001421734 057% 洗漱
504414878 014% 锁门
3509
锁门
org.springframework.util.StopWatch$TaskInfo@12f40c25
3
复制
此时,看到StopWatch的魅力所在了吗?
- 通过多组start、stop方法,将业务代码块进行区分,可获得不同代码块的执行耗时;
- 可以通过start方法传入taskName,对每个代码块进行命名;
- 可以对总任务耗时、每个任务耗时进行统计分析;
- prettyPrint()方法,可以优雅的打印出统计分析信息;
- getTotalTimeMillis()方法,打印出总耗时;
- getLastTaskName()方法,打印最后一个任务名称;
- getLastTaskInfo()方法,获得最后一个任务的TaskInfo,进而获得更多相关信息;
- getTaskCount()方法,获得任务数;
现在再看,使用StopWatch是不是可以获得更多有用的信息?
StopWatch的实现原理
最后呢,我们再来看一眼源码,了解一下StopWatch的实现机制。
先看StopWatch的start方法实现:
public void start(String taskName) throws IllegalStateException {
if (this.currentTaskName != null) {
throw new IllegalStateException("Can't start StopWatch: it's already running");
}
this.currentTaskName = taskName;
this.startTimeNanos = System.nanoTime();
}
复制
start方法中记录了任务名称和任务执行的时间,基于System.nanoTime()获得。
stop方法实现如下:
public void stop() throws IllegalStateException {
if (this.currentTaskName == null) {
throw new IllegalStateException("Can't stop StopWatch: it's not running");
}
long lastTime = System.nanoTime() - this.startTimeNanos;
this.totalTimeNanos += lastTime;
this.lastTaskInfo = new TaskInfo(this.currentTaskName, lastTime);
if (this.keepTaskList) {
this.taskList.add(this.lastTaskInfo);
}
++this.taskCount;
this.currentTaskName = null;
}
复制
在stop方法中,通过两个时间戳相减获得lastTime,也就是一个任务的执行时间;lastTime累计相加获得总的执行时间;同时,记录任务列表、任务数统计。
而其他get方法,则是对start、stop中获取的数据的进一步统计、分析、格式化输出而已。
小结
有些功能当我们使用习惯了,可能就固守于一个实现方式,但如果去参考学习其他框架中类似功能的实现,往往会有些新的突破。如果你在使用Spring的框架,建议你尝试一下StopWatch这个API,可以让你的时间统计日志更加高端大气。
博主简介:《SpringBoot技术内幕》技术图书作者,酷爱钻研技术,写技术干货文章。