发布于 

CountDownLatch线程计数器

CountDownLatch的使用

概念

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

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

使用场景

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

CountDownLatch的使用

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
// 首先创建一个线程计数器 构造参数为线程数量
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的所有方法

1
2
3
4
5
6
7
8
9
10
11
12
13
// 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(){};

本站由 @binvv 使用 Stellar 主题创建。
本博客所有文章除特别声明外,均采用 CC BY-NC-SA 4.0 许可协议,转载请注明出处。