【十一】Java多线程J.U.C之 Countdownlatch

J.U.C是JDK 1.5提供的包 java.util.concurrent

Countdownlatch允许一个或多个线程等待直到在其他线程中一组操作执行完成。

一、Countdownlatch原理

主线程TA调用await()后等待T1 T2 T3三个线程都执行了countDown，计数器cnt=0后，主线程开始继续执行。

强调：执行countDown()的线程，并不会因为执行了countDown()方法而被blocked，执行countDown()方法只是计数减1而已，会继续之后后面的代码。只有执行await()方法的线程才会被blocked。

这个点是countdownlatch跟cyclicBarrier的区别之一，cyclicBarrier到达屏障点的线程会被blocked，所有线程都到达屏障点则所有线程同时唤醒。

另一个区别是同一个countdownlatch对象计数为0后不能重复使用了，而同一个cyclicBarrier对象在调用reset()方法后还能重复使用。

二、使用场景

有任务A和任务B，任务B必须在任务A完成之后再做。任务A能被分为n部分，这n部分之间的任务互不影响。把这n部分任务分给不同的线程，当A任务完成后，通知做B任务的线程开始执行任务，执行B任务的线程，可以是一个或多个。

三、代码示例

主线程等待T1 T2 T3三个线程执行完成后，统计一共花费的时间。

package com.sid.thread.CountDownLatchTest;

import java.util.Date;
import java.util.concurrent.CountDownLatch;

/**
 * @program: thread-test
 * @description:    某个线程需要等待一个或多个线程（多个线程是同步的）操作结束（或达到某种状态）才开始执行
 *                    示例：main线程，等待3个线程执行完了，main线程再统计总共时间
 * @author: Sid
 * @date: 2018-11-27 14:28
 * @since: 1.0
 **/
public class CountDownLatchDemo {
    public static void main(String[] args) throws InterruptedException {
        int totalThread = 3;
        long start = System.currentTimeMillis();
        CountDownLatch countDown = new CountDownLatch(totalThread);
        for(int i = 0; i < totalThread; i++) {
            final String threadName = "Thread " + i;
            new Thread(() -> {
                System.out.println(String.format("%s\t%s %s", new Date(), threadName, "started"));
                try {
                    Thread.sleep(1000);
                } catch (Exception ex) {
                    ex.printStackTrace();
                }
                /**
                 * 如果当前计数器的值>1，则将其减1，强调：执行countDown.countDown()的线程并不会blocked，只是计数减1，然后继续往后面执行
                 * 若当前值=1，则将其置为0并唤醒所有通过await等待的线程；
                 * 若当前值=0，则什么也不做直接返回。
                 */
                countDown.countDown();
                System.out.println(String.format("%s\t%s %s", new Date(), threadName, "ended"));
            }).start();
        }

        /**
         * 读取当前计数器的值，一般用于调试或者测试。
         * */
        long count = countDown.getCount();
        System.out.println("count is :"+count);
        /**
         * await()
         * 等待计数器的值为0，若计数器的值为0则该方法返回；
         * 若等待期间该线程被中断，则抛出InterruptedException并清除该线程的中断状态。
         *
         * await(long timeout, TimeUnit unit) 在指定的时间内等待计数器的值为0，
         * 若在指定时间内计数器的值变为0，则该方法返回true；
         * 若指定时间内计数器的值仍未变为0，则返回false；
         * 若指定时间内计数器的值变为0之前当前线程被中断，则抛出InterruptedException并清除该线程的中断状态。
         * */
        countDown.await();

        long stop = System.currentTimeMillis();
        System.out.println(String.format("Total time : %sms", (stop - start)));
    }
}

运行结果