【JAVA-并发】CountDownLatch用法和源码浅析

1.CountDownLatch是java并法包中的一员，用于同步一个或者多个任务，强制他们等待有其他任务执行的一组操作完成，常用于多线程之间的协作任务，即一个或者多个线程等待其他线程全部完成在执行的业务场景。

2.具体项目场景应用，电子发票批量下载任务，由于下载任务耗时而且比较多，而且用户需要等待下载任务处理完成给予提示。这时候用到了CountDownLatch，开启了多个子任务线程同时执行，主线程等待子任务线程全部完成后，给予前端用户提示下载完成。这样提高了下载效率。

3. CountDownLatch的具体用法：

ExecutorService service = Executors.newCachedThreadPool();

CountDownLatch cdOrder = new CountDownLatch(3)；

For(int i=0; i<3; i++){

Runnable runnable = new Runnable(){

public void run(){

System.out.print(“执行下载任务…”)

cdOrder.countDown();

}

}

service.execute(runnable);

}

cdOrder.await();

4.具体的源码剖析：

a. CountDownLatch的构造函数：传入计数数值

b.分析它的一个内部类Sync如下：

Sync继承AbstractQueuedSynchronizer类，它的父类中有个state属性如下：

/**

* The synchronization state.

*/

Private volatileintstate;

这个state是用volatile修饰，说明这个属性具有线程之间的可见性

c.再看看方法countDown（）的源码：

继续跟进去，先尝试去释放共享所：

Sync类覆盖了父类的方法tryReleaseShared：

从这段代码可以看出：先是获得共享变量state，如果等于0，返回false；否则对共享变量state减一操作，调用compareAndSetState()方法：

其中内部就是使用CAS原理，对共享变量设置值，CAS操作包含三个操作数内存位置（V）、预期原值（A）和新值（B）。如果内存位置的值与预期原值想匹配，那么处理器会自动将该位置值更新为新值；否则，处理器不做任何操作，保证原子性操作达到线程安全目的。

d.接下来再看方法doReleaseShared()

从代码可以看出，当tryReleaseShared方法返回来true，也就是即state=0的情况。会调用doReleaseShared()方法去唤醒等待的线程。如果头节点状态为-1，表示需要唤醒后续节点

e.最后的方法await():

代码继续跟进：

主要是最后这个方法doAcquireSharedInterruptibly，开启自旋模式一直等待，直到锁被全部释放。由于采用了公平锁，所以要将节点放到队列里,保证先到先执行；获取当前节点的前继节点p,如果当前节点的前继节点等于head头节点，则表示当前节点是等待队列中的第一个节点。满足先到先执行的公平顺序，则尝试释放锁,再次获取当前剩余的共享锁数量是否为0，如果为0，则表示可以进行后续的唤醒动作