闭锁CountDownLatch的用法

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

package concurrent;

class LatchDemo implements Runnable {

    @Override
    public void run() {
        for (int i = ; i <= ; 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 = ; i < ; 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 = ; i < ; i++)
                System.out.println(i);
        } finally {
            //使用finally确保倒计时操作一定可以执行
            latch.countDown();
        }
    }

}

public class TestCountDownLatch {

    public static void main(String[] args) {

        //参数5对应要等待的线程数量，我们在下面创建了5个线程
        CountDownLatch latch = new CountDownLatch();

        LatchDemo ld = new LatchDemo(latch);

        long start = System.currentTimeMillis();

        for (int i = ; i < ; 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));

    }

}
           

　　运行程序，发现可以正常计算整个程序的执行时间：