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