Java的concurrent包里面的CountDownLatch其实可以把它看作一个计数器,只不过这个计数器的操作是原子操作,同时只能有一个线程去操作这个计数器,也就是同时只能有一个线程去减这个计数器里面的值。
你可以向CountDownLatch对象设置一个初始的数字作为计数值,任何调用这个对象上的await()方法都会阻塞,直到这个计数器的计数值被其他的线程减为0为止。
CountDownLatch的一个非常典型的应用场景是:有一个任务想要往下执行,但必须要等到其他的任务执行完毕后才可以继续往下执行。假如我们这个想要继续往下执行的任务调用一个CountDownLatch对象的await()方法,其他的任务执行完自己的任务后调用同一个CountDownLatch对象上的countDown()方法,这个调用await()方法的任务将一直阻塞等待,直到这个CountDownLatch对象的计数值减到0为止
比如某个逻辑是要等5个线程执行完毕 才能执行某个操作:
package thread;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.CountDownLatch;
public class CountDownLatchTest {
public static void main(String[] args) {
CountDownLatch countDownLatch = new CountDownLatch(5);
MyTask myTask = new MyTask(countDownLatch);
Thread t1 = new Thread(myTask,"线程1");
Thread t2 = new Thread(myTask,"线程2");
Thread t3 = new Thread(myTask,"线程3");
Thread t4 = new Thread(myTask,"线程4");
Thread t5 = new Thread(myTask,"线程5");
List<Thread> threadList = new ArrayList<>();
threadList.add(t1);
threadList.add(t2);
threadList.add(t3);
threadList.add(t4);
threadList.add(t5);
for(int i=0;i<threadList.size();i++){
threadList.get(i).start();
}
try {
countDownLatch.await();
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println("线程执行完毕");
}
}
class MyTask implements Runnable{
private CountDownLatch countDownLatch;
public MyTask(CountDownLatch countDownLatch) {
this.countDownLatch = countDownLatch;
}
@Override
public void run() {
try {
Thread.sleep(2000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(Thread.currentThread().getName()+"执行完毕");
countDownLatch.countDown();
}
}
上面就是执行完毕5个线程后 main线程才能执行。有点类似Thread类中的join方法,但是不同的是join方法是有顺序的,