多线程处理大LIST

1.线程池类-ServiceThreadPool

package com.kba.common.service;

import lombok.Data;

import java.util.concurrent.*;

/**
 * @author zhangzhiqing
 * @create 2022-10-18 17:53
 */
@Data
public class ServiceThreadPool {

    private ExecutorService executorService;

    /**
     * int corePoolSize,
     * int maximumPoolSize,
     * long keepAliveTime,
     * TimeUnit unit,
     * BlockingQueue<Runnable> workQueue,
     * RejectedExecutionHandler handler
     */
    public ServiceThreadPool() {
        int cpuNum = Runtime.getRuntime().availableProcessors();
        Double size = cpuNum / (1 - 0.8);
        this.executorService = new ThreadPoolExecutor(size.intValue(), size.intValue() + 10, 60, TimeUnit.SECONDS,
                new ArrayBlockingQueue<>(40), Executors.defaultThreadFactory(), new ThreadPoolExecutor.AbortPolicy());
    }

}

2.拆分list的工具类SplitListUtils(将List 拆分为多个list处理)

package com.kba.common.util;

import org.springframework.util.CollectionUtils;

import java.util.ArrayList;
import java.util.List;

/**
 * 拆分结合工具类
 */
public class SplitListUtils {
    /**
     * 拆分集合
     *
     * @param <T> 泛型对象
     * @param resList 需要拆分的集合
     * @param subListLength 每个子集合的元素个数
     * @return 返回拆分后的各个集合组成的列表
     * 代码里面用到了guava和common的结合工具类
     **/
    public static <T> List<List<T>> split(List<T> resList, int subListLength) {
        if (CollectionUtils.isEmpty(resList) || subListLength <= 0) {
            return new ArrayList();
        }
        List<List<T>> ret = new ArrayList();
        int size = resList.size();
        if (size <= subListLength) {
            // 数据量不足 subListLength 指定的大小
            ret.add(resList);
        } else {
            int pre = size / subListLength;
            int last = size % subListLength;
            // 前面pre个集合，每个大小都是 subListLength 个元素
            for (int i = 0; i < pre; i++) {
                List<T> itemList = new ArrayList();
                for (int j = 0; j < subListLength; j++) {
                    itemList.add(resList.get(i * subListLength + j));
                }
                ret.add(itemList);
            }
            // last的进行处理
            if (last > 0) {
                List<T> itemList = new ArrayList();
                for (int i = 0; i < last; i++) {
                    itemList.add(resList.get(pre * subListLength + i));
                }
                ret.add(itemList);
            }
        }
        return ret;
    }
}

3.具体处理list 的实现逻辑（请忽略业务mapper等代码）

  /**
     * @Description:推荐处方更新逗号隔开的drugName
     *
     * @author zhangzhiqing
     * @date  2023/7/11 9:24
     */
    @Transactional(rollbackFor = Exception.class)
    public void createAdvices() {
        ExecutorService executorService = THREAD_POOL_EXECUTOR.getExecutorService();

        List<ScreenCnCliRecord> totalCnCliRecordList = new ArrayList<>();
        //查询需要拼接处方的列表 疾病前20
        List<ScreenCnCliRecord> allRecordList = createIndexMapper.selectNeedDealRecord();
        //查询所有医嘱
        List<ScreenCliAdvices> advicesList = createIndexMapper.selectAdvicesList();

        List<List<ScreenCnCliRecord>> split = SplitListUtils.split(allRecordList, 1000);
        CountDownLatch countDownLatch = new CountDownLatch(split.size());


        List<Future<List<ScreenCnCliRecord>>> futureList = new ArrayList<>();

        for (List<ScreenCnCliRecord> recordList:split){
            Future<List<ScreenCnCliRecord>> submit  = executorService.submit(()-> {
                List<ScreenCnCliRecord> newCnCliRecordList = new ArrayList<>();

                for(ScreenCnCliRecord entity:recordList){
                    String mrn = entity.getMrn();
                    String orgCode = entity.getOrgCode();
                    List<ScreenCliAdvices> advices = advicesList.stream().filter(e->mrn.equals(e.getMrn())&&orgCode
                            .equals(e.getOrgCode())).collect(Collectors.toList());
                    if(advices!=null&&advices.size()>0){
                        List<String> drugNameList = advices.stream().map(ScreenCliAdvices::getDrugName).collect(Collectors.toList());
                        if(drugNameList!=null&&drugNameList.size()>0){
                            Collections.sort(drugNameList);
                            String drugNames =String.join(",", drugNameList);
                            drugNames = drugNames.replaceAll(" ", "");
                            entity.setDrugName(drugNames);
                            newCnCliRecordList.add(entity);
                        }
                    }
                }
                return newCnCliRecordList;
            });
            futureList.add(submit);
            countDownLatch.countDown();
        }

        try {
            countDownLatch.await();
        } catch (InterruptedException e) {
            e.printStackTrace();
        }

        futureList.forEach(integerFuture -> {
            try {
                totalCnCliRecordList.addAll(integerFuture.get());
            } catch (InterruptedException e) {
                e.printStackTrace();
            } catch (ExecutionException e) {
                e.printStackTrace();
            }
        });

        if(totalCnCliRecordList!=null&&totalCnCliRecordList.size()>0){
            List<List<ScreenCnCliRecord>> splitCnCliRecordList = SplitListUtils.split(totalCnCliRecordList, 1000);
            for(List<ScreenCnCliRecord> entityList:splitCnCliRecordList){
                createIndexMapper.batchUpdateRecordList(entityList);
            }
        }
    }

tips：
feature.get 是阻塞的，为啥还要CountDownLatch 控制，我问了下chartGpt 说是更灵活。