CountDownLatch的使用

/ 多线程 / 没有评论 / 308浏览

在开发多线程程序时,常常有这样的需求就是希望主线程可以等待子线程执行完后在执行主线程的逻辑,按照我们之学过的知识,我们可以用join()方法来实现此需求。

public class Test {
public static void main(String[] args) throws InterruptedException {
Thread thread1 = new Thread(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
});
Thread thread2 = new Thread(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
});
thread1.start();
thread2.start();
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
}
任务: main
任务: Thread-0
任务: Thread-1

下面我们调用join()方法来满足我们的需求。

public class Test {
public static void main(String[] args) throws InterruptedException {
Thread thread1 = new Thread(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
});
Thread thread2 = new Thread(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
});
thread1.start();
thread2.start();
thread1.join();
thread2.join();
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
}
任务: Thread-1
任务: Thread-0
任务: main

有关join()方法的使用,在这里就不做过多说明了,想详细了解的可以去看看这篇文章

下面我们用CountDownLatch来满足我们的需求。

CountDownLatch:允许一个或多个线程等待其它线程完成操作。CountDownLatch的构造方法必须接收一个int类型的参数。这个参数也就是要等待多少个的线程的数量。如果我们要等待2个线程,那我们就参数就写2。当我们调用CountDownLatch中的countDown()方法时,参数会自动减1,当参数变成0时,则不会在阻塞当前线程。

public class Test {
public static void main(String[] args) throws InterruptedException {
final CountDownLatch countDownLatch = new CountDownLatch(2);
Thread thread1 = new Thread(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
countDownLatch.countDown();
}
});
Thread thread2 = new Thread(new Runnable() {
@Override
public void run() {
try {
Thread.sleep(1000);
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
countDownLatch.countDown();
}
});
thread1.start();
thread2.start();
countDownLatch.await();
System.out.println(String.format("任务: %s", Thread.currentThread().getName()));
}
}
任务: Thread-1
任务: Thread-0
任务: main

如果CountDownLatch构造方法里的参数大于调用CountDownLatch中的countDown()方法的数量时,也就是说参数会一直不等于0。那么当前线程就会一直等待。在实际的多线程开发中常常用CountDownLatch来满足我们上述的需求,因为它和join()相比,可以直接在线程的内部处理,而不需要在当前线程中调用join()。