CountDownLatch是Java并发包中非常实用的一个工具类，它可以帮助我们实现线程之间的同步和协作。CountDownLatch的核心思想是通过计数器来控制线程的执行顺序。当计数器的值降为0时，所有等待的线程都会被唤醒，然后开始执行下一步操作。
1.CountDownLatch的源码解读
在Java中，CountDownLatch的实现是基于AbstractQueuedSynchronizer类的。AbstractQueuedSynchronizer是一个非常重要的同步器，Java中的许多并发类都是基于它来实现的，例如Semaphore、ReentrantLock、ReadWriteLock等。
CountDownLatch的核心实现类是Sync，它是一个继承自AbstractQueuedSynchronizer的内部类。下面是Sync类的源码：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
|
private static final class Sync extends AbstractQueuedSynchronizer { Sync(int count) { setState(count); } int getCount() { return getState(); } protected int tryAcquireShared(int acquires) { return (getState() == 0) ? 1 : -1; } protected boolean tryReleaseShared(int releases) { for (;;) { int c = getState(); if (c == 0) return false; int nextc = c-1; if (compareAndSetState(c, nextc)) return nextc == 0; } }} |
Sync类中有三个重要的方法：
- tryAcquireShared(int acquires)：尝试获取锁，如果计数器的值等于0，表示所有线程都已经执行完毕，返回1，否则返回-1，表示获取锁失败。
- tryReleaseShared(int releases)：释放锁，将计数器的值减1，并返回减1后的计数器的值。如果计数器的值减为0，表示所有线程都已经执行完毕，返回true，否则返回false。
- getCount()：返回当前计数器的值。
tryAcquireShared()方法是CountDownLatch的关键所在，它会尝试获取锁。如果计数器的值等于0，说明所有线程都已经执行完毕，可以返回1，表示获取锁成功；否则返回-1，表示获取锁失败。这里使用了AbstractQueuedSynchronizer类的基础方法，即getState()方法，该方法用于获取同步器的状态。
tryReleaseShared()方法用于释放锁，将计数器的值减1，并返回减1后的计数器的值。如果计数器的值减为0，表示所有线程都已经执行完毕，返回true，否则返回false。这里使用了AtomicInteger类的基础方法，即compareAndSetState()方法，该方法用于比较并设置同步器的状态。
2.CountDownLatch的原理解析
CountDownLatch的工作原理非常简单，它通过计数器来控制线程的执行顺序。当计数器的值降为0时，所有等待的线程都会被唤醒，然后开始执行下一步操作。
CountDownLatch是一个多线程协作的工具类，它允许一个或多个线程等待其他线程完成某个操作后再继续执行。CountDownLatch有一个计数器，当计数器的值变为0时，等待的线程就会被唤醒。CountDownLatch的使用方式非常简单，主要包括两个方法：await()和countDown()。
- await()方法：该方法会阻塞当前线程，直到计数器的值变为0。
- countDown()方法：该方法会将计数器的值减1。
下面是一个简单的示例代码：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
public class CountDownLatchDemo { public static void main(String[] args) throws InterruptedException { final int count = 3; final CountDownLatch latch = new CountDownLatch(count); for (int i = 0; i < count; i++) { new Thread(() -> { // 线程执行任务 System.out.println(Thread.currentThread().getName() + " 执行任务..."); // 任务执行完毕，计数器减1 latch.countDown(); }).start(); } // 等待所有任务执行完毕 latch.await(); System.out.println("所有任务执行完毕..."); }} |
在该示例代码中，我们创建了一个CountDownLatch对象，并将计数器初始化为3。然后创建了3个线程，每个线程执行一个任务，任务执行完毕后，将计数器减1。最后，在主线程中调用latch.await()方法等待所有任务执行完毕。
CountDownLatch的实现原理是基于AbstractQueuedSynchronizer类的。当我们调用await()方法时，线程会尝试获取锁，如果计数器的值不为0，则获取锁失败，线程会被加入到同步队列中阻塞。当我们调用countDown()方法时，计数器的值会减1，如果计数器的值减为0，表示所有线程都已经执行完毕，此时同步队列中的线程会被唤醒，继续执行下一步操作。
具体来说，在Sync类中，tryAcquireShared(int acquires)方法会尝试获取锁，如果计数器的值等于0，表示所有线程都已经执行完毕，返回1，否则返回-1，表示获取锁失败。tryReleaseShared(int releases)方法用于释放锁，将计数器的值减1，并返回减1后的计数器的值。如果计数器的值减为0，表示所有线程都已经执行完毕，返回true，否则返回false。
3.CountDownLatch的应用场景
CountDownLatch是一个非常实用的工具类，它可以帮助我们实现线程之间的同步和协作。下面介绍一些CountDownLatch的常见应用场景：
- 等待多个线程执行完毕：如果有多个线程需要执行，但是必须等待所有线程都执行完毕才能进行下一步操作，可以使用CountDownLatch来实现。我们可以创建一个CountDownLatch对象，并将计数器的值初始化为线程数，每个线程执行完毕后，调用countDown()方法将计数器减1。最后，在主线程中调用await()方法等待所有线程执行完毕。
- 控制线程的执行顺序：如果有多个线程需要按照特定的顺序执行，可以使用CountDownLatch来实现。我们可以创建多个CountDownLatch对象，每个对象的计数器的值都为1，表示只有一个线程可以执行。线程执行完毕后，调用下一个CountDownLatch对象的countDown()方法，唤醒下一个线程。
- 等待外部事件的发生：如果我们需要等待一个外部事件的发生，例如某个网络连接的建立或某个文件的读取完成，可以使用CountDownLatch来实现。我们可以在主线程中创建一个CountDownLatch对象，并将计数器的值初始化为1，然后在另一个线程中等待外部事件的发生。当外部事件发生时，调用CountDownLatch对象的countDown()方法，唤醒主线程继续执行。
- 控制并发线程数：如果我们需要控制并发线程的数量，可以使用CountDownLatch来实现。我们可以创建一个CountDownLatch对象，并将计数器的值初始化为线程数量，每个线程执行完毕后，调用countDown()方法将计数器减1。如果某个线程需要等待其他线程执行完毕，可以调用await()方法等待计数器的值变为0。
4.总结
CountDownLatch是一个非常实用的工具类，它可以帮助我们实现线程之间的同步和协作。CountDownLatch的使用非常简单，只需要调用两个方法：await()和countDown()。CountDownLatch的实现原理是基于AbstractQueuedSynchronizer类的，它通过同步队列来实现线程之间的协作。
CountDownLatch的应用场景非常广泛，包括等待多个线程执行完毕、控制线程的执行顺序、等待外部事件的发生、控制并发线程数等。在实际开发中，我们可以根据具体的需求来选择合适的应用场景，使用CountDownLatch来实现线程之间的同步和协作，从而提高程序的性能和可靠性。
到此这篇关于Java并发包工具类CountDownLatch的应用详解的文章就介绍到这了,更多相关Java CountDownLatch内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！
原文链接：https://juejin.cn/post/7226175610212417594
