线程CountDownLatch使用详解

Posted 2023-03-06 weixin_ancenhw

一、什么是countDownlatch

CountDownLatch是一个同步工具类，它通过一个计数器来实现的,初始值为线程的数量。每当一个线程完成了自己的任务,计数器的值就相应得减1。当计数器到达0时,表示所有的线程都已执行完毕,然后在等待的线程就可以恢复执行任务。

二、方法详解

CountDownLatch(int count)：count为计数器的初始值（一般需要多少个线程执行，count就设为几）。
countDown(): 每调用一次计数器值-1，直到count被减为0，代表所有线程全部执行完毕。
getCount()：获取当前计数器的值。
await(): 等待计数器变为0，即等待所有异步线程执行完毕。
boolean await(long timeout, TimeUnit unit)： 此方法与await()区别：
①此方法至多会等待指定的时间，超时后会自动唤醒，若 timeout 小于等于零，则不会等待
②boolean 类型返回值：若计数器变为零了，则返回 true；若指定的等待时间过去了，则返回 false

三、CountDownLatch应用场景

1. 某个线程需要在其他n个线程执行完毕后再向下执行
2. 多个线程并行执行同一个任务，提高响应速度

@Slf4j
public class CountDownLatchDemo01 
    private static final int THREAD_NUM = 5;

    public static void main(String[] args) 
        log.info("创建线程池");
        ExecutorService executorService = Executors.newFixedThreadPool(THREAD_NUM);
        //创建线程个数
        final CountDownLatch countDownLatch = new CountDownLatch(THREAD_NUM);
        for (int i = 0; i < THREAD_NUM; i++) 
            executorService.execute(()->
                try 
                    System.out.println("子线程："+Thread.currentThread().getName()+"开始执行");
                    //模拟每个线程处理业务处理3秒
                    TimeUnit.SECONDS.sleep(1);
                    System.out.println("子线程："+Thread.currentThread().getName()+"执行完成");
                 catch (InterruptedException e) 
                    throw new RuntimeException(e);
                finally 
                    //当前线程调用此方法，则计数器减1
                    countDownLatch.countDown();
                
            );
        
        try 
            //阻塞当前线程，直到计数器为0，主线程才开始处理。
            countDownLatch.await();
            System.out.println("等待所有子线程执行完成，主线程："+Thread.currentThread().getName()+"");
         catch (InterruptedException e) 
            throw new RuntimeException(e);
        
      executorService.shutdown();