CountDownLatch源码剖析

.

    public static void main(String[] args) throws Exception{
        CountDownLatch latch = new CountDownLatch(2);

        new Thread(()->{
            try {
                Thread.sleep(2000);
                System.out.println("线程1准备执行 countDown 操作");
                latch.countDown();
            }catch (Exception e){
                e.printStackTrace();
            }
        }).start();

        new Thread(()->{
            try {
                Thread.sleep(2000);
                System.out.println("线程2准备执行 countDown 操作");
                latch.countDown();
            }catch (Exception e){
                e.printStackTrace();
            }
        }).start();

        System.out.println("main() 线程准备执行 await");
        latch.await();
        System.out.println("success");
    }

CountDownLatch

  我们在创建 CountDownLatch 的时候,其实他就是一个 sycn 、继承了 aqs,可以猜到它的实现应该也是aps 那一套。首先就是将 count 赋值给了 state 标记锁状态。

CountDownLatch源码剖析

await

  主线程在 await() 的时候就等在了那里,直到所有线程执行完 countDown() , 主线程就继续往下执行。猜想他应该是进入了 aqs 的同步队列,直到其他线程将他唤醒。那我们就看看它的具体实现。

1. tryAcquireShared() 获取 state ,我们初始化的时候设置的是2,那么此时肯定 !=0 。
2. doAcquireSharedInterruptibly() 首先在 addWaiter() 里面死循环两次将 当前线程加入等待队列中。
3. 然后获取到上一个节点 p, 它肯定是一个空的头节点。
4. 然后 tryAcquireShared() 再次判断一下锁占用状态。
5. 此时 r 肯定是>0 的,然后就 p.next=null 将自己从等待队中摘除出来
6. 最后在 parkAndCheckInterrupt() 里面调用 LockSupport.park() 将自己挂起,等待别人的唤醒。

CountDownLatch源码剖析

countDown

1. tryReleaseShared() 先获取 state ,如果 !=0 说明有人持有锁,那就 cas 操作 -1 返回true。
2. doReleaseShared() 首先获取到头节点 h ,肯定不会为空的。
3. 然后如果状态是 Node.SIGNAL 那就 cas 改成 0 。
4. 最后在 unparkSuccessor() 里面获取头节点的下一个节点,也就是我们放在 同步队列 里面的main线程节点,通过 LockSupport.unpark() 唤醒它。

 CountDownLatch源码剖析

 

 

CountDownLatch源码剖析

上一篇:SonarQube+jenkins实现CodeReview环境搭建


下一篇:k8s cpu 高负载排查