CountDownLatch线程计数器

CountDownLatch的使用

概念

countDownLatch这个类使一个线程等待其他线程各自执行完毕后再执行。

countDownLatch是通过一个计数器来实现的，计数器的初始值是线程的数量。每当一个线程执行完毕后，计数器的值就-1，当计数器的值为0时，表示所有线程都执行完毕，然后在闭锁上等待的线程就可以恢复工作了。

使用场景

在一些时候由于数据库连表查询过于缓慢，并且无法在数据库sql层进行优化，这时我们需要在代码中使用线程去分页查询数据，多个线程都执行完毕后将数据封装进一个list再进行返回，这时需要保证所有线程执行完毕后才能返回。

CountDownLatch的使用

// 首先创建一个线程计数器 构造参数为线程数量
CountDownLatch countDownLatch = new CountDownLatch(3);

// 创建一个结果集合
List<> list = new ArrayList<>();

// 循环分页查询
for (int i = 1; i <= 3; i++) {
    // executorService 是一个线程池对象，这里直接使用了，也可以通过new Thread(){ public void run() }.start();匿名内部类方式创建线程
    executorService.execute(() -> {
        // 调用查询
        List<> list1 = mapper.find(i * 10, 10);
        // 将结果添加到集合中
        list.addAll(list1);
        // 执行完毕调用线程计数器-1 方法
        countDownLatch.countDown();
    });
}

try {
    // 调用await()方法的线程会被挂起，它会等待直到count值为0才继续执行
    countDownLatch.await();
    System.out.println("返回数据添加完成");
} catch (Exception e) {
    e.printStackTrace();
}

CountDownLatch的所有方法

// countDownLatch类中只提供了一个构造器 count为计数值
public CountDownLatch(int count){};

// 调用await()方法的线程会被挂起，它会等待直到count值为0才继续执行
public void await() throws InterruptedException {};
// 和await()类似，只不过等待一定的时间后count值还没变为0的话就会继续执行
public boolean await(long timeout, TimeUnit unit) throws InterruptedException {};
// 将count值减1
public void countDown() {};
// 返回当前计数
public long getCount(){};
// 返回计数器标识和当前计数
public String toString(){};