闭锁CountDownLatch的用法

矫情吗；* 2022-07-12 04:15 194阅读 0赞

现在有十个线程各自输出0-50000中的所有偶数，假设我们试图使用下面这段代码计算整个程序的执行时间：

package concurrent;
    
    class LatchDemo implements Runnable {
    
        @Override
        public void run() {
            for (int i = 0; i <= 50000; i++)
                System.out.println(i);
        }
    
    }
    
    public class TestCountDownLatch { 
    
        public static void main(String[] args) {
    
            LatchDemo ld = new LatchDemo();
    
            long start = System.currentTimeMillis();
    
            for (int i = 0; i < 5; i++)
                new Thread(ld).start();
    
            long end = System.currentTimeMillis();
    
            System.out.println("total time is " + (end - start));
    
        }
    
    }

会发现无法达到预期的效果，甚至在输出的最后一行根本看不到“total time is XXX”的结果，这是正常的，因为在这个多线程的环境下，或许当其他线程还没有将偶数完全输出，main线程已经将时间计算完毕并且输出了。要解决这个问题，我们需要使main线程中计算时间的代码在其他所有线程运算全部完成之后才执行，这就是CountDownLatch的功能，下面我们使用CountDownLatch来改进程序：

package concurrent;
    
    import java.util.concurrent.CountDownLatch;
    
    class LatchDemo implements Runnable {
    
        private CountDownLatch latch;
    
        public LatchDemo(CountDownLatch latch) {
            this.latch = latch;
        }
    
        @Override
        public void run() {
            try {
                for (int i = 0; i < 50000; i++)
                    System.out.println(i);
            } finally {
                //使用finally确保倒计时操作一定可以执行
                latch.countDown();
            }
        }
    
    }
    
    public class TestCountDownLatch { 
    
        public static void main(String[] args) {
    
            //参数5对应要等待的线程数量，我们在下面创建了5个线程
            CountDownLatch latch = new CountDownLatch(5);
    
            LatchDemo ld = new LatchDemo(latch);
    
            long start = System.currentTimeMillis();
    
            for (int i = 0; i < 5; i++)
                new Thread(ld).start();
    
            //等待其他所有线程的运算全部完成
            try {
                latch.await();
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
    
            long end = System.currentTimeMillis();
    
            System.out.println("total time is " + (end - start));
    
        }
    
    }

运行程序，发现可以正常计算整个程序的执行时间：  
　　  
　　![image\_1b8iupr8jfcpohjd6s1ucb145i9.png-27kB][image_1b8iupr8jfcpohjd6s1ucb145i9.png-27kB]

[image_1b8iupr8jfcpohjd6s1ucb145i9.png-27kB]: /images/20220711/9aacf1958c83443a87046854140735c7.png