多线程同步挑战：使用CountDownLatch示例

原创﹏ヽ暗。殇╰゛Y 2024-11-30 00:51 9阅读 0赞

CountDownLatch是Java中一种常用的多线程同步工具，它允许一个或多个等待其他线程完成某些工作后才能继续的线程。

以下是一个使用CountDownLatch的简单示例：

```java
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

public class CountDownLatchExample {
    private final CountDownLatch countDownLatch = new CountDownLatch(2);

public void task1() {
        try {
            System.out.println("Task 1 starting...");
            countDownLatch.await(5, TimeUnit.SECONDS);
            System.out.println("Task 1 completed.");
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }

public void task2() {
        try {
            System.out.println("Task 2 starting...");
            if (!countDownLatch.await(3, TimeUnit.SECONDS))) { // 如果等待时间超过任务执行时间，说明其他线程还未执行完 task 1的任务，需要重试 Task 2不再执行任何操作，并重新加入到CountDownLatch中，直到task 1完成并释放CountDownLatch count = countDownLatch.getCount(); if (count <= 0) { System.out.println("All tasks completed!"); countDownLatch.countDown(); } else { System.out.println("Task 2 retrying...")); // 重试 Task 2 } catch (InterruptedException e) { e.printStackTrace(); } }
  
    public static void main(String[] args) {
        CountDownLatchExample example = new CountDownLatchExample();
        
        Thread task1Thread = new Thread(example::task1);
        Thread task2Thread = new Thread(example::task2);

task1Thread.start();
        task2Thread.start();

// 等待所有任务完成
        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }
    }
}
```

在这个例子中，我们有两个线程task1和task2。任务1需要等待3秒，然后继续执行；任务2则直接开始执行。

CountDownLatch在task1完成时会减一，如果所有任务完成后countDownLatch的值为0，则调用countDown()方法，表示所有任务已完成。