在Java中,CountDownLatch是一种并发工具,它允许一个或多个线程等待一组操作完成。
CountDownLatch的使用方法如下:
创建一个CountDownLatch对象,指定需要等待的操作数量。例如,如果有n个操作需要等待完成,则可以创建一个CountDownLatch对象并将其初始化为n。
在每个操作完成之后,调用CountDownLatch的countDown()方法来减少计数器的值。这表示一个操作已经完成。
在需要等待所有操作完成的地方,调用CountDownLatch的await()方法。这将导致调用线程等待,直到计数器的值为0。
当所有操作都完成时,计数器的值将变为0,调用线程将被释放。
下面是一个简单的示例代码,演示了CountDownLatch的使用:
import java.util.concurrent.CountDownLatch;
public class CountDownLatchExample {
public static void main(String[] args) throws InterruptedException {
int numberOfThreads = 3;
CountDownLatch latch = new CountDownLatch(numberOfThreads);
for (int i = 0; i < numberOfThreads; i++) {
Thread thread = new Thread(new Worker(latch));
thread.start();
}
latch.await();
System.out.println("All threads have completed their work");
}
static class Worker implements Runnable {
private final CountDownLatch latch;
public Worker(CountDownLatch latch) {
this.latch = latch;
}
@Override
public void run() {
// 执行一些操作
System.out.println("Thread " + Thread.currentThread().getId() + " is working");
latch.countDown();
}
}
}
在上面的示例中,创建了一个CountDownLatch对象,并指定需要等待的线程数量为3。然后,使用一个for循环创建3个工作线程,并在每个工作线程中调用countDown()方法来减少计数器的值。
最后,调用latch.await()方法,这将导致主线程等待,直到计数器的值为0。当所有工作线程都完成时,计数器的值将变为0,主线程会继续执行,并打印"All threads have completed their work"。
这就是CountDownLatch的基本使用方法。通过使用CountDownLatch,可以很方便地实现等待一组操作完成的功能。