Java 监控线程池所有任务是否执行完毕

Java 监控线程池所有任务是否执行完毕

场景引入

在最近的工作中遇到一个需要批量生产百万数据并写入数据库的需求，先通过单线程的方式去实现，但是感觉效率一般，然后通过多线程的方式去改进，但是遇到下面的问题：

生成数据的时候，每生成一条数据，就将数据丢入到集合中进行缓存，当集合中的数据量达到 500 的时候，就进行一次数据库写入操作，并清空集合。

那么当我生成的数据量为 1200 的时候，就会出现最后 200 条没有写入到数据库的问题

那么要解决这个问题，就要知道线程池的任务什么时候执行完毕，网上找了很多方法如：

• 方案一 - 直接调用是否终止的方法

executor.isTerminated()

• 方案二 - 判断活跃线程数

executor.getActiveCount() == 0

• 方案三 - 设置一个较长的等待时间

executor.awaitTermination(10000, TimeUnit.SECONDS);

但是都没有达到我的目的（也有可能是我使用的姿势不对）

最后尝试了下面的方法

功能实现

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.CollectionUtils;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ThreadPoolExecutor;

@Slf4j
public class ThreadPoolTest {
   

    <