多线程之CountDownLatch

概念

CountDownLatch是通过一个计数器来实现的，计数器的初始值为线程的数量。每当一个线程完成了自己的任务后，计数器的值就会减1。当计数器值到达0时，它表示所有的线程已经完成了任务，然后在闭锁上等待的线程就可以恢复执行任务。
构造器中的计数值（count）实际上就是闭锁需要等待的线程数量。这个值只能被设置一次，而且CountDownLatch没有提供任何机制去重新设置这个计数值。

与CountDownLatch的第一次交互是主线程等待其他线程。主线程必须在启动其他线程后立即调用CountDownLatch.await()方法。这样主线程的操作就会在这个方法上阻塞，直到其他线程完成各自的任务。

其他N个线程必须引用闭锁对象，因为他们需要通知CountDownLatch对象，他们已经完成了各自的任务。这种通知机制是通过 CountDownLatch.countDown()方法来完成的；每调用一次这个方法，在构造函数中初始化的count值就减1。所以当N个线程都调 用了这个方法，count的值等于0，然后主线程就能通过await()方法，恢复执行自己的任务。

---

await()和countDown()

示例：

package com.example.demo.three.untils.countdownlouch;

import java.util.Arrays;
import java.util.Random;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * 案例：
 * 取到一批数据，利用两个线程对数据进行逻辑处理
 * 当处理完过后，再用这批数据做一些其它事情
 */
public class CountDownLunchTest1 {
    private static Random random = new Random(System.currentTimeMillis());
    private static ExecutorService executor = Executors.newFixedThreadPool(2);
    private static CountDownLatch latch = new CountDownLatch(9);

    public static void main(String[] args) throws InterruptedException {
        //获取数组
        int[] data = query();
        //两次线程去操作逻辑
        for (int i = 0; i < data.length; i++) {
            executor.execute(new SimpleRunnAble(data, i, latch));

        }
        //保证所有线程执行完毕，再执行下面程序
        latch.await();
        executor.shutdown();
        //获取处理过的数据
        System.out.println("取到数字" + Arrays.toString(data));
    }

    static class SimpleRunnAble implements Runnable {
        private final int[] data;
        private final int index;
        private final CountDownLatch latch;

        SimpleRunnAble(int[] data, int index, CountDownLatch latch) {
            this.data = data;
            this.index = index;
            this.latch = latch;
        }

        @Override
        public void run() {
            try {
                Thread.sleep(random.nextInt(1000));
            } catch (InterruptedException e) {
                e.printStackTrace();
            }

            int value = data[index];
            if (value % 2 == 0) {
                data[index] = value * 2;
            } else {
                data[index] = value * 10;
            }
            System.out.println(Thread.currentThread().getName() + "结束");
            latch.countDown();
        }
    }

    private static int[] query() {
        return new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9};
    }
}

---

await(long timeout, TimeUnit unit)

await(long timeout, TimeUnit unit)：表示等待最长时间。

package com.example.demo.three.untils.countdownlouch;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
public class CountDownLunchTest2 {
    public static void main(String[] args) throws InterruptedException {
        //new CountDownLatch(0)当为0时，latch.await()无效果
        CountDownLatch latch = new CountDownLatch(1);
        new Thread() {
            @Override
            public void run() {
                try {
                    Thread.sleep(1000);
                    latch.countDown();
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }.start();
        /**
         * 最多等待10秒
         * 1.如果10秒内，没有countDown为0，10秒后发行
         * 2.如果10秒内，countDown为0，立刻发行，不用等待10秒
         */
        latch.await(10,TimeUnit.SECONDS);
        System.out.println("==========");
    }
}