【Java】CountDownLatch使用之多线程场景等待所有线程执行结束

目前有个任务,需要java生成三张图片,生成之后将三张图片进行二次处理

由于生成图片比较耗时,开三个线程进行同时生成

那么问题来了,如何知道三张图片都生成完了呢

两种解决方案一种AtomicInteger二种CountDownLatch都是计数器的概念,具体的暂未了解

 

以下采用CountDownLatch的方式提供例子

 

/**
 * 多线程等待完成测试
 */
public class CountDownLatchTest {

    public static void main(String[] args) throws InterruptedException {

        ExecutorService executor = Executors.newCachedThreadPool();

        final CountDownLatch latch = new CountDownLatch(3);
        executor.execute(new Runnable() {
            @Override
            public void run() {
                try {
                    Thread.sleep(1);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("生成图片1完成");
                latch.countDown();
            }
        });

        executor.execute(new Runnable() {
            @Override
            public void run() {
                try {
                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("生成图片2完成");
                latch.countDown();
            }
        });

        executor.execute(new Runnable() {
            @Override
            public void run() {
                try {
                    Thread.sleep(5000);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
                System.out.println("生成图片3完成");
                latch.countDown();
            }
        });

        System.out.println("等待生成完成");
        latch.await();
        System.out.println("完成");

    }
}