CountDownLatch的两种用法

CountDownLatch 的两种用法

1.某一时刻，所有线程一起执行

package juc.countDownLatch;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

/**
 * @Author： xinjingjie
 * @Date：2021/3/3 14:05
 * 到达某一时刻一起执行
 **/
public class CountDownLatchDemo2 {
    public static void main(String[] args) throws InterruptedException {
        CountDownLatch startSignal = new CountDownLatch(1);
        CountDownLatch doneSignal = new CountDownLatch(10);
        for (int i = 0; i < 10; ++i)
            // create and start threads
            new Thread(new Worker(startSignal, doneSignal)).start();

        Thread.sleep(2000);
        System.out.println("比赛开始");
        startSignal.countDown();

        // let all threads proceed doSomethingElse();
        doneSignal.await(); // wait for all to finish
    }

    static class Worker implements Runnable {
        private final CountDownLatch startSignal;
        private final CountDownLatch doneSignal;

        Worker(CountDownLatch startSignal, CountDownLatch doneSignal) {
            this.startSignal = startSignal;
            this.doneSignal = doneSignal;
        }

        @Override
        public void run() {
            try {
                System.out.println(Thread.currentThread().getName() + "正在准备");
                startSignal.await();
                doWork();
                doneSignal.countDown();
                System.out.println(Thread.currentThread().getName() + "完成比赛");
            } catch (InterruptedException ex) {
            } // return;
        }

        void doWork() throws InterruptedException {
            TimeUnit.SECONDS.sleep(1);
            System.out.println(Thread.currentThread().getName() + "正在冲刺");
        }
    }
}

结果：

“C:\Program Files\Java\jdk1.8.0_181\bin\java.exe” "-
Thread-0正在准备
Thread-2正在准备
Thread-5正在准备
Thread-1正在准备
Thread-9正在准备
Thread-4正在准备
Thread-6正在准备
Thread-3正在准备
Thread-7正在准备
Thread-8正在准备
比赛开始
Thread-5正在冲刺
Thread-0正在冲刺
Thread-9正在冲刺
Thread-1正在冲刺
Thread-1完成比赛
Thread-5完成比赛
Thread-9完成比赛
Thread-2正在冲刺
Thread-2完成比赛
Thread-0完成比赛
Thread-4正在冲刺
Thread-4完成比赛
Thread-6正在冲刺
Thread-6完成比赛
Thread-8正在冲刺
Thread-8完成比赛
Thread-3正在冲刺
Thread-3完成比赛
Thread-7正在冲刺
Thread-7完成比赛

2.执行完所有的业务后才执行主线程

package juc.countDownLatch;

import java.util.concurrent.*;

/**
 * @Author： xinjingjie
 * @Date：2021/3/3 14:10
 * 执行完所有的业务后才执行主线程
 **/
public class CountDownLatchDemo3 {
    public static void main(String[] ar) throws InterruptedException {
        CountDownLatch doneSignal = new CountDownLatch(10);
        ThreadPoolExecutor e = new ThreadPoolExecutor(10,
                13,
                1,
                TimeUnit.SECONDS,
                new ArrayBlockingQueue<>(10));
        for (int i = 0; i < 10; ++i)
            //create and start threads
            e.execute(new WorkerRunnable(doneSignal, i));
        doneSignal.await(); // wait for all to finish
        System.out.println("all is done");
        //关闭线程池
        e.shutdown();
    }
}

class WorkerRunnable implements Runnable {
    private final CountDownLatch doneSignal;
    private final int i;

    WorkerRunnable(CountDownLatch doneSignal, int i) {
        this.doneSignal = doneSignal;
        this.i = i;
    }

    @Override
    public void run() {
        doWork(i);
        doneSignal.countDown();
    }

    void doWork(int i) {
        System.out.println("执行第" + i + "个");
    }
}

执行第1个
执行第3个
执行第0个
执行第5个
执行第4个
执行第7个
执行第8个
执行第9个
执行第2个
执行第6个
all is done

CountDownLatch的作用就是允许一个或多个线程等待其他线程完成操作，看起来有点类似join() 方法，但其提供了比 join() 更加灵活的API。
CountDownLatch可以手动控制在n个线程里调用n次countDown()方法使计数器进行减一操作，也可以在一个线程里调用n次执行减一操作。
而 join() 的实现原理是不停检查join线程是否存活，如果 join 线程存活则让当前线程永远等待。所以两者之间相对来说还是CountDownLatch使用起来较为灵活。

https://zhuanlan.zhihu.com/p/148231820