「多线程」多线程中的CountDownLatch

目录

CountDownLatch的介绍

CountDownLatch的原理

CountDownLatch的代码实现

CountDownLatch的注意事项

---

CountDownLatch的介绍

        多线程中的 CountDownLatch（倒计时门闩）是一种同步工具，用于控制线程的执行顺序。它基于一个计数器，可以让一个或多个线程等待其他线程完成特定操作。

CountDownLatch的原理

1. 创建一个 CountDownLatch 对象，并设置初始计数值。这个计数值表示需要等待的线程数量。

2. 在需要等待的线程中，调用 countDown() 方法来通知 CountDownLatch 一个操作已经完成。

3. 调用 await() 方法的线程将会被阻塞，直到计数值减到零。

4. 当所有的操作都完成时，计数值将会变为零，阻塞在 await() 方法上的线程将会被释放，继续执行后续的操作。

使用 CountDownLatch 可以实现以下几种场景：

1. 等待多个线程完成某个操作后再继续执行：可以创建一个 CountDownLatch 对象，设置计数值为等待的线程数量，然后每个线程完成操作时调用 countDown() 方法，主线程调用 await() 方法等待所有线程完成。

2. 控制并发任务的同时开始和结束：可以使用 CountDownLatch 来启动多个线程，然后在主线程上调用 await() 方法等待所有线程启动后，同时开始执行任务。然后在任务执行完毕后，通过 countDown() 方法通知主线程任务已经完成。

CountDownLatch的代码实现

        在下面的示例中，创建了 5 个线程（Worker 类）来模拟任务的执行，每个线程睡眠 1 秒钟后完成任务，并调用 countDown() 方法通知主线程。主线程在调用 await() 方法后等待所有线程执行完毕，最终输出 "All workers have completed their tasks."。

import java.util.concurrent.CountDownLatch;

public class CountDownLatchExample {
    public static void main(String[] args) throws InterruptedException {
        int threadCount = 5;
        CountDownLatch latch = new CountDownLatch(threadCount);

        for (int i = 0; i < threadCount; i++) {
            Thread thread = new Thread(new Worker(latch));
            thread.start();
        }

        // 等待所有线程执行完成
        latch.await();

        System.out.println("All workers have completed their tasks.");
    }

    static class Worker implements Runnable {
        private final CountDownLatch latch;

        public Worker(CountDownLatch latch) {
            this.latch = latch;
        }

        @Override
        public void run() {
            // 模拟耗时操作
            try {
                Thread.sleep(1000);
            } catch (InterruptedException e) {
                Thread.currentThread().interrupt();
            }

            System.out.println("Task completed.");

            // 完成任务后调用 countDown()
            latch.countDown();
        }
    }
}

 CountDownLatch的注意事项

在使用 CountDownLatch 时，以下是一些需要注意的事项：

1. 计数值的设置：创建 CountDownLatch 对象时，需要提前确定等待的线程数量。确保计数器的初始值与实际等待的线程数量相匹配，否则可能导致线程无法继续执行或永久阻塞。

2. countDown() 方法的调用：在等待的线程中，必须在完成操作后调用 countDown() 方法来递减计数值。通常在 finally 块内调用，以确保即使发生异常也能正确减少计数值。

3. 阻塞等待：调用 await() 方法的线程将会被阻塞，直到计数值减到零。这意味着需要注意避免在主线程中调用 await() 方法，以免造成程序的假死。

4. 计数值不能重置：一旦计数值减到零，就不能再增加。如果需要进行多个阶段的等待操作，可以考虑使用其他同步工具，如 CyclicBarrier 或 Semaphore。

5. 异常处理：在使用 CountDownLatch 时，需要捕获并处理可能抛出的异常，例如在等待过程中线程被中断。需要适当处理线程的中断状态，以保证程序的正常运行。

6. 性能考虑：使用过多的 CountDownLatch 对象可能会对性能产生负面影响，因此在设计和使用时要谨慎。如果有大量的线程需要等待，可以考虑使用其他更高级的并发工具，如 ExecutorService 或 CompletionService。

7. CountDownLatch 的计数值一旦减到零就不能再增加。如果需要在多个阶段使用，可以考虑使用 CyclicBarrier 或 Semaphore 等其他同步工具。

        总之，在使用 CountDownLatch 时，确保计数值的正确设置，正确调用 countDown() 方法，并处理可能的异常情况。理解这些注意事项可以帮助您更好地使用 CountDownLatch 来控制线程的执行顺序。