CountDownLatch的使用

CountDownLatch的概念

CountDownLatch（闭锁），它是java.util.cucurrent包下的工具类，能够保证一个（调用CountDownLatch.await()的）线程在其他线程完成各自的工作后再执行。

CountDownLatch实现机制

CountDownLatch内存在一个计数器字段，计数器的值是创建CountDownLatch对象时设置的，调用CountDownLatch.await()的线程需要等待计数器的值变成0才能继续执行。每完成了一个任务，计数器的值就会减一。当计数器值到达0时，它表示所有需要结束的线程已经完成了任务，调用await()方法的线程就可以恢复执行任务。

CountDownLatch提供的方法

方法都非常简单，就不详细解析。

CountDownLatch的使用场景

1、保证线程执行顺序，类似join()。

package com.su.mybatis.oracle.controller;

import java.util.concurrent.CountDownLatch;

public class Test {

    public static void main(String[] args) throws InterruptedException {
        System.out.println(Thread.currentThread().getName() + "开始...");
        final CountDownLatch latch = new CountDownLatch(2);
        new Thread(){
            @Override
            public void run() {
                System.out.println("线程"+Thread.currentThread().getName()+"开始...");
                try {
                    //模拟业务
                    Thread.sleep(1000);
                }catch (Exception e){
                    e.printStackTrace();
                }finally {
                    System.out.println("线程"+Thread.currentThread().getName()+"结束...");
                    latch.countDown();
                }
            }
        }.start();

        new Thread(){
            @Override
            public void run() {
                System.out.println("线程"+Thread.currentThread().getName()+"开始...");
                try {
                    //模拟业务
                    Thread.sleep(2000);
                } catch (Exception e){
                    e.printStackTrace();
                } finally {
                    System.out.println("线程"+Thread.currentThread().getName()+"结束...");
                    latch.countDown();
                }
            }
        }.start();
        try {
            System.out.println("等待线程结束");
            latch.await();
            System.out.println("线程"+Thread.currentThread().getName()+"结束...");
        }catch (Exception e){

        }
}

}

输出结果：

main开始...
线程Thread-0开始...
等待线程结束
线程Thread-1开始...
线程Thread-0结束...
线程Thread-1结束...
线程main结束...

2、模拟并发测试

package com.su.mybatis.oracle.controller;

import java.util.concurrent.CountDownLatch;

public class Test {

    public static void main(String[] args) throws InterruptedException {
        System.out.println(Thread.currentThread().getName() + "开始...");
        final CountDownLatch latch = new CountDownLatch(100000);
        long startTime = System.currentTimeMillis();
        for(int i = 0; i < 100000; i++) {
            new Thread(){
                @Override
                public void run() {
                    System.out.println("线程"+Thread.currentThread().getName()+"开始...");
                    try {
                        Thread.sleep(10);
                    } catch (Exception e){
                        e.printStackTrace();
                    } finally {
                        System.out.println("线程"+Thread.currentThread().getName()+"结束...");
                        latch.countDown();
                    }
                }
            }.start();
        }
        try {
            System.out.println("等待线程结束");
            latch.await();
            System.out.println("线程"+Thread.currentThread().getName()+"结束...,花费时间："
            + (System.currentTimeMillis() - startTime) + "ms");
        }catch (Exception e){
            e.printStackTrace();
        }
    }
}

输出结果：

main开始...
线程Thread-0开始...
线程Thread-1开始...
.
.
.
线程Thread-99999结束...
线程Thread-99998结束...
线程Thread-99997结束...
线程Thread-99996结束...
线程main结束...,花费时间：16443ms

 

 

如果有写的不对的地方，请大家多多批评指正，非常感谢！