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

小灰灰 2021-09-24 22:22 249阅读 0赞

目前有个任务,需要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("完成");
    
        }
    }