CountDownLanch的使用
CountDownLanch 是一个倒数计数器, 给一个初始值(>=0), 然后每一次调用countDown就会减1, 这很符合等待多个子线程结束的场景: 一个线程结束的时候, countDown一次, 直到所有的线程都countDown了 , 那么所有子线程就都结束了.
await: 会阻塞等待计数器减少到0位置. 带参数的await是多了等待时间.
countDown: 将当前的计数减1
getCount(): 返回当前的计数
显而易见, 我们只需要在子线程执行之前, 赋予初始化countDownLanch, 并赋予线程数量为初始值.
每个线程执行完毕的时候, 就countDown一下.主线程只需要调用await方法, 可以等待所有子线程执行结束。
示例:
package com.test;
import java.util.concurrent.CountDownLatch;
public class CountDownDemo {
public static void main(String[] args) throws Exception {
//定义线程数
int subThreadNum = 5;
//取得一个倒计时器,从5开始
CountDownLatch countDownLatch = new CountDownLatch(subThreadNum);
//依次创建5个线程,并启动
for (int i = 0; i < subThreadNum; i++) {
new SubThread(2000*(i+1), countDownLatch).start();
}
//主线程工作
mainWork();
//等待所有的子线程结束
countDownLatch.await();
System.out.println("Main Thread work done!");
}
private static void mainWork(){
System.out.println("Main thread start work!");
try {
Thread.sleep(2000L);
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
System.out.println("Main Thread work done!");
}
/**
* 子线程类
* @author fuhg
*/
private static class SubThread extends Thread{
private CountDownLatch countDownLatch;
private long workTime;
public SubThread(long workTime,CountDownLatch countDownLatch){
this.workTime = workTime;
this.countDownLatch = countDownLatch;
}
public void run() {
// TODO Auto-generated method stub
try {
System.out.println("Sub thread is starting!");
Thread.sleep(workTime);
System.out.println("Sub thread is stopping!");
} catch (InterruptedException e) {
// TODO Auto-generated catch block
e.printStackTrace();
} finally {
//线程结束时,将计时器减一
countDownLatch.countDown();
}
}
}
}