CountDownLatch的使用
场景
1,业务中出现需要压缩A到B天的数据每天生成一个excel(不考虑数据量巨大的情况),并且打包成zip返回给用户,打包数据的时候使用多线程性能会提高很多,但是这里也会有一个问题,比如打包5天的数据,那么我们必须让所有的线程都完成了工作才能进行打包否则会出现数据损失的情况
public class CountDownLatchDemo {
public static void main(String[] args) throws InterruptedException {
ThreadPoolExecutor executor = new ThreadPoolExecutor(5, 20, 10, TimeUnit.MICROSECONDS, new LinkedBlockingQueue<Runnable>(10));
CountDownLatch countdown = new CountDownLatch(10);
executor.execute(() -> {
for (int i = 0; i < 10; i++) {
Thread thread = new Thread("线程" + i) {
@Override
public void run() {
for (int i = 0; i < 10; i++) {
System.out.println("--线程--" + Thread.currentThread().getName() + "i的值为:" + i);
}
System.err.println("--线程--" + Thread.currentThread().getName() + "次数" + countdown.getCount());
countdown.countDown();
}
};
thread.start();
}
});
System.err.println("计数器为:" + countdown.getCount());
countdown.await();
if (countdown.getCount() <= 0) {
System.err.println("线程已经全部结束");
executor.shutdown();
}
System.out.println("线程池已经结束");
// 继续业务代码
}
}
这段代码中最主要的是countdown.await();
如果没有这个代码会导致等待没有执行,导致数据错误