@Async结合CompletableFuture实现主线程阻塞,CompletableFuture并发执行任务
项目开发中经常会遇到业务需要多任务处理的场景,比如目前我除了的业务就是如此。
我要提供给客户端一个批量查询第三方数据的接口,由于是调用第三方的接口,相应时间受到网络环境的影响,如果需要查询的的数据过多,响应时间就会很长,所有这里使用异步处理,但是需要阻塞主线程,拿到所以数据后由主线程进行处理再返回给用户,是个时候就需要使用多线程的工具,@Async结合CompletableFuture实现主线程阻塞,CompletableFuture并发执行任务。
在主线程中调用多个带有返回值的 @Async 方法,并且等待所有 @Async 方法执行完毕后,再执行主线程的后续操作,而且希望这个过程不阻塞主线程。CompletableFuture 是 Java 提供的一个用于异步编程的工具类,它提供了丰富的方法来处理异步操作的结果。
- 定义公共线程池,并开启异步注解@Async
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.task.AsyncTaskExecutor;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;@Configuration
@EnableAsync
public class AsyncConfig {private static final int MAX_POOL_SIZE = 5;private static final int CORE_POOL_SIZE = 5;@Bean("asyncTaskThreadPool")public AsyncTaskExecutor asyncTaskThreadPool() {ThreadPoolTaskExecutor taskExecutor = new ThreadPoolTaskExecutor();taskExecutor.setMaxPoolSize(MAX_POOL_SIZE);taskExecutor.setCorePoolSize(CORE_POOL_SIZE);taskExecutor.setThreadNamePrefix("async-task-thread-pool");taskExecutor.initialize();return taskExecutor;}}
- 在业务方法上添加@Async注解并返回CompletableFuture,@Async使用时注意最好是使用自定义的线程池
@Async("asyncTaskThreadPool")public CompletableFuture<JSONObject> getApplymentsInfo(String applymentId) {//义务逻辑,,,,,,return CompletableFuture.completedFuture("返回业务数据");}
- 使用时需要注意主线程阻塞,CompletableFuture异步并发执行
public List<JSONObject> getApplymentsInfoList(List<String> applymentIds) throws Exception {List<CompletableFuture<JSONObject>> jsonObjectFutureList = new ArrayList<>();for (String applymentId : applymentIds) {jsonObjectFutureList.add(wechatPayRequest.getApplymentsInfo(applymentId));}// 等待所有 CompletableFuture 完成CompletableFuture<Void> allOfFuture = CompletableFuture.allOf(jsonObjectFutureList.toArray(new CompletableFuture[jsonObjectFutureList.size()]));allOfFuture.get(); // 阻塞主线程,等待所有 CompletableFuture 完成List<JSONObject> jsonObjectList = new ArrayList<>();for (CompletableFuture<JSONObject> completableFuture : jsonObjectFutureList) {jsonObjectList.add(completableFuture.get());}return jsonObjectList;}