@Configuration public class ThreadPoolConfig { // Spring 的线程池 @Bean public ThreadPoolTaskExecutor threadPoolTaskExecutor() { ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); executor.setCorePoolSize(10); executor.setMaxPoolSize(20); executor.setQueueCapacity(200); executor.setThreadNamePrefix("MyExecutor-"); executor.initialize(); return executor; } } -------------------- @RestController public class MyController { @Autowired private ThreadPoolTaskExecutor executor; @GetMapping("/async") public String asyncMethod() { executor.execute(() -> { // 异步操作 }); return "success"; } } -------------------- 或者 @Autowired @Qualifier("threadPoolTaskExecutor") private ThreadPoolTaskExecutor threadPoolTaskExecutor; StudyPlanVo vo = new StudyPlanVo(); CompletableFuture<Void> basicFuture = CompletableFuture.runAsync(() -> { StudyPlanBasic basic = studyPlanBasicMapper.findByTaskId(taskDataId); vo.setBasic(basic); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanBasicMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> moleculeFuture = CompletableFuture.runAsync(() -> { List<StudyPlanMolecule> moleculeList = studyPlanMoleculeMapper.findByTaskId(taskDataId); vo.setMoleculeList(moleculeList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanMoleculeMapper.findByTaskId", ex); return null; }); CompletableFuture.allOf(basicFuture, moleculeFuture).get(1L, TimeUnit.MINUTES);
@Configuration // 启用异步任务，guava ThreadFactoryBuilder public class AsyncConfiguration { final int cpuNum = Runtime.getRuntime().availableProcessors(); // java 的线程池 @Bean("threadPoolExecutor") public ThreadPoolExecutor threadPoolExecutor() { final ThreadFactory queueThreadFactory = new ThreadFactoryBuilder() .setNameFormat("queue-pool-%d").build(); return new ThreadPoolExecutor(cpuNum, cpuNum*3, 60L, TimeUnit.SECONDS, new LinkedBlockingQueue<>(120), queueThreadFactory, new ThreadPoolExecutor.CallerRunsPolicy()); } } @Autowired @Qualifier("threadPoolExecutor") private ThreadPoolExecutor threadPoolExecutor; ------------------------------------- public StudyPlanVo studyPlan(String taskDataId) throws Exception { StudyPlanVo vo = new StudyPlanVo(); CompletableFuture<Void> basicFuture = CompletableFuture.runAsync(() -> { StudyPlanBasic basic = studyPlanBasicMapper.findByTaskId(taskDataId); vo.setBasic(basic); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanBasicMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> moleculeFuture = CompletableFuture.runAsync(() -> { List<StudyPlanMolecule> moleculeList = studyPlanMoleculeMapper.findByTaskId(taskDataId); vo.setMoleculeList(moleculeList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanMoleculeMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> prescriptionFuture = CompletableFuture.runAsync(() -> { List<StudyPlanPrescription> prescriptionList = studyPlanPrescriptionMapper.findByTaskId(taskDataId); if (!CollectionUtils.isEmpty(prescriptionList)) { prescriptionList.sort((o1, o2) -> { long a1 = parseSequence(o1.getSequenceNumber()); long a2 = parseSequence(o2.getSequenceNumber()); return Long.compare(a1, a2); }); } vo.setPrescriptionList(prescriptionList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanPrescriptionMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> testGroupFuture = CompletableFuture.runAsync(() -> { List<StudyPlanTestGroup> testGroupList = studyPlanTestGroupMapper.findByTaskId(taskDataId); vo.setTestGroupList(testGroupList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanTestGroupMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> conditionFuture = CompletableFuture.runAsync(() -> { List<StudyPlanCondition> conditionList = studyPlanConditionMapper.findByTaskId(taskDataId); vo.setConditionList(conditionList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanConditionMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> proteinFuture = CompletableFuture.runAsync(() -> { List<StudyPlanProtein> proteinList = studyPlanProteinMapper.findByTaskId(taskDataId); vo.setProteinList(proteinList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanProteinMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> reagentFuture = CompletableFuture.runAsync(() -> { List<StudyPlanReagent> reagentList = studyPlanReagentMapper.findByTaskId(taskDataId); vo.setReagentList(reagentList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanReagentMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> consumableFuture = CompletableFuture.runAsync(() -> { List<StudyPlanConsumable> consumableList = studyPlanConsumableMapper.findByTaskId(taskDataId); vo.setConsumableList(consumableList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanConsumableMapper.findByTaskId", ex); return null; }); CompletableFuture<Void> numFuture = CompletableFuture.runAsync(() -> { List<StudyPlanNum> numList = studyPlanNumMapper.findByTaskId(taskDataId); vo.setNumList(numList); }, threadPoolExecutor).exceptionally(ex -> { log.error("Failed to studyPlanNumMapper.findByTaskId", ex); return null; }); CompletableFuture.allOf(basicFuture, moleculeFuture, prescriptionFuture, testGroupFuture, conditionFuture, proteinFuture, reagentFuture, consumableFuture, numFuture) .get(1L, TimeUnit.MINUTES); return vo; }
import java.util.concurrent.CompletableFuture; import java.util.concurrent.ExecutionException; public class CompletableFutureAllOfDemo { public static void main(String[] args) throws ExecutionException, InterruptedException { // 任务A：模拟耗时操作（如日志打印） CompletableFuture<Void> taskA = CompletableFuture.runAsync(() -> { System.out.println("任务A执行中..."); sleep(1000); // 模拟耗时1秒 System.out.println("任务A执行完毕"); }); // 任务B：模拟数据处理 CompletableFuture<Void> taskB = CompletableFuture.runAsync(() -> { System.out.println("任务B执行中..."); sleep(1500); // 模拟耗时1.5秒 System.out.println("任务B执行完毕"); }); // 任务C：模拟文件上传 CompletableFuture<Void> taskC = CompletableFuture.runAsync(() -> { System.out.println("任务C执行中..."); sleep(800); // 模拟耗时0.8秒 System.out.println("任务C执行完毕"); }); // 任务D：模拟接口调用 CompletableFuture<Void> taskD = CompletableFuture.runAsync(() -> { System.out.println("任务D执行中..."); sleep(1200); // 模拟耗时1.2秒 System.out.println("任务D执行完毕"); }); // 关键：用 allOf 组合所有任务，等待全部完成 CompletableFuture<Void> allTasks = CompletableFuture.allOf(taskA, taskB, taskC, taskD); // 方式1：阻塞等待所有任务完成（适合主线程需要等待结果的场景） allTasks.join(); // join() 抛出 unchecked 异常（无需显式捕获） // 或 allTasks.get(); // get() 抛出 checked 异常（需 try-catch） System.out.println("\n===== 所有任务（A/B/C/D）执行完毕 ====="); // 方式2：非阻塞注册“所有任务完成后执行的逻辑”（适合回调场景） allTasks.thenRun(() -> { System.out.println("\n===== 回调：所有任务（A/B/C/D）执行完毕 ====="); // 这里可以写后续逻辑（如汇总结果、触发下一个流程） }).join(); // 若需等待回调执行，可再加 join() } // 辅助方法：模拟耗时 private static void sleep(long millis) { try { Thread.sleep(millis); } catch (InterruptedException e) { Thread.currentThread().interrupt(); } } }
