背景
近来, 在用多个线程去下载多个资源文件, 用Junit测试时踩到了一个坑~涉世太浅~debug过程,发现有时只有一个线程在下载资源就结束了,其它线程还没开始就完事了~~真是奇了个怪,于是谷歌了下,发现Junit测试不支持多线程。恍然大悟, 坑有点深,以此记录,铭记于心!
下面模拟下Junit测试不支持多线程的场景.
import org.junit.jupiter.api.Test;
public class JunitDemo {
private int i = 10;
@Test
public void test() {
for (int i = 0; i < this.i; i ++) {
new Thread(new Runner()).start();
}
}
class Runner implements Runnable {
@Override
public void run() {
System.out.println("威少、安东尼、保罗乔治组成雷霆三巨头!");
}
}
}
多次执行上述程序,会发现不会打印出10行结果。Junit测试,在程勋运行完毕后,就立即把结束,才不管后台启动了多个线程在处理什么。
解决方案。
方案比较多。这里瞎歪歪几种
- 通过callable接口异步获取结果。交给FutureTask处理。在get值的时候当前线程会被阻塞起来。
- 通过CountDownLatch计数来实现
eg :
private int i = 10;
/*
* 线程计数器
* 将线程数量初始化
* 每执行完成一条线程,调用countDown()使计数器减1
* 主线程调用方法await()使其等待,当计数器为0时才被执行
*/
private CountDownLatch latch = new CountDownLatch(i);
@Test
public void test() {
for (int i = 0; i < this.i; i ++) {
new Thread(new Runner()).start();
}
try {
latch.await(); // 主线程等待
} catch (InterruptedException e) {
e.printStackTrace();
}
}
class Runner implements Runnable {
@Override
public void run() {
System.out.println("威少、安东尼、保罗乔治组成雷霆三巨头!");
latch.countDown(); // 执行完毕,计数器减1
}
参考文献