CountDownLatch是Java中一个多线程工具类,它可以用于控制线程的顺序,可以让主线程等待其他线程完成任务之后再继续执行,或者让多个线程之间相互等待。
CountDownLatch的核心思想是,设置一个计数器,每次线程完成任务后,计数器减1,当计数器减到0时,所有等待该计数器的线程开始执行。它的构造函数中需要传入一个整数参数,表示需要等待的线程数量。
使用CountDownLatch需要以下几步:
1.创建CountDownLatch实例并指定等待线程数量
CountDownLatch countDownLatch = new CountDownLatch(n);
2.在需要等待的线程中,调用countDown()方法,计数器减1
countDownLatch.countDown();
3.在主线程中调用await()方法,等待其他线程完成任务
countDownLatch.await();
下面是一个示例代码:
import java.util.concurrent.CountDownLatch;
public class CountDownLatchExample {
public static void main(String[] args) throws InterruptedException {
int n = 5; // 等待5个线程完成任务
CountDownLatch countDownLatch = new CountDownLatch(n);
for (int i = 0; i < n; i++) {
Thread t = new Thread(() -> {
try {
System.out.println(Thread.currentThread().getName() + " is working");
Thread.sleep(1000);
System.out.println(Thread.currentThread().getName() + " done");
countDownLatch.countDown(); // 计数器减1
} catch (InterruptedException e) {
e.printStackTrace();
}
});
t.start();
}
countDownLatch.await(); // 等待其他线程完成任务
System.out.println("All tasks done");
}
}
输出结果:
Thread-0 is working
Thread-1 is working
Thread-2 is working
Thread-3 is working
Thread-4 is working
Thread-2 done
Thread-1 done
Thread-0 done
Thread-4 done
Thread-3 done
All tasks done
在这个例子中,创建了5个线程,并让每个线程睡眠1秒钟,表示完成一个任务。在每个线程完成任务后,调用了countDown()方法,计数器减1。
在主线程中,调用了await()方法,等待所有线程完成任务。当所有线程的计数器都减为0时,主线程才会继续执行,输出"All tasks done"。