线程池使用记录并使用Future类返回任务执行结果

线程池使用记录

ExecutorService executor = Executors.newFixedThreadPool(5);
RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
JSONArray goodsJson = ExcelUtils.readMultipartFile(file);
try{
    Callable<List<AddGoodsReq>> task = () -> {
        RequestContextHolder.setRequestAttributes(requestAttributes);
        return goodsService.importExcel(goodsJson);
    };
    Future<List<AddGoodsReq>> futureGoodsReqList = executor.submit(task);
    List<AddGoodsReq> addGoodsReqList = futureGoodsReqList.get();
    removeExistGoods(addGoodsReqList);

    for (AddGoodsReq addGoodsReq : addGoodsReqList) {
        try {
            executor.submit(() -> {
                // 线程上下文传递
                RequestContextHolder.setRequestAttributes(requestAttributes);
                shopGoodsService.addGoods(addGoodsReq);
            });
        } catch (Exception e) {
            e.printStackTrace(); // 处理任务执行过程中的异常
            throw new RuntimeException("添加失败: " + e.getMessage(), e);
        }
    }
}catch (InterruptedException | ExecutionException e){
    Thread.currentThread().interrupt(); // 重设中断状态
    e.printStackTrace();
} finally {
    // 关闭线程池
    executor.shutdown();
    try {
        // 等待所有任务执行完成，或者设置一个合理的超时时间
        executor.awaitTermination(1, TimeUnit.HOURS);
    } catch (InterruptedException e) {
        e.printStackTrace();
    }
}

代码解析：

// 拿到主线程的上下文属性，包含请求头信息等内容
RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();

---

// 使用Callable异步执行并返回结果的任务
Callable<List<AddGoodsReq>> task = () -> {
		// 在每个子线程中设置上下文属性，确保子线程的与主线程拥有相同的请求信息
        RequestContextHolder.setRequestAttributes(requestAttributes);
        // 调用方法执行具体的任务，并返回一个 List<AddGoodsReq> 类型的结果
        return goodsService.importExcel(goodsJson);
};

---

// 将 task 提交给线程池进行执行，并返回一个Future对象 futureGoodsReqList,Future<List<AddGoodsReq>> 表示一个未来可能会完成并返回 List<AddGoodsReq> 类型结果的任务
Future<List<AddGoodsReq>> futureGoodsReqList = executor.submit(task);
// 阻塞等待task中的任务执行完毕，并将结果存储到addGoodsReqList中
List<AddGoodsReq> addGoodsReqList = futureGoodsReqList.get();

至此，实现了使用线程池异步执行任务并使用Future返回结果的流程。