import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
import java.util.concurrent.ThreadPoolExecutor;
/**
* @author wdj
* Created on 2020/10/28 14:55
*/
@Configuration
public class MyThreadPoolTaskExecutor {
/**
* 默认异步线程池
*
* @return
*/
@Bean
public ThreadPoolTaskExecutor taskExecutor() {
ThreadPoolTaskExecutor pool = new ThreadPoolTaskExecutor();
// 获取JAVA虚拟机的可用处理器数量。IO密集型建议核心线程数是该值2倍;计算密集型建议核心线程数是该值1倍
int processorNum = Runtime.getRuntime().availableProcessors();
//设置核心线程数
pool.setCorePoolSize(processorNum * 2);
//最大线程数
pool.setMaxPoolSize(processorNum * 4);
//缓冲队列数
pool.setQueueCapacity(100);
//线程池名前缀
pool.setThreadNamePrefix("user-");
//允许线程空闲时间(单位:默认为秒)
pool.setKeepAliveSeconds(300);
// 设置拒绝策略,当线程池阻塞队列已满时对新任务的处理。调节机制,即饱和时回退主线程执行
pool.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
// 等待所有任务结束后再关闭线程池
pool.setWaitForTasksToCompleteOnShutdown(true);
// 初始化
pool.initialize();
return pool;
}
}
编写了异步线程池代码后,还需要在启动类上开启异步线程支持,加上注解
@EnableAsync //开启线程异步支持
以及在需要用到异步线程支持的方法上加上注解
@Async
使用@Async需要注意的地方,如果@Async是标注在无返回值的方法上,那么就像正常方法那样编写即可,但如果是有返回值的方法,返回的数据类型为Future类型,其为一个接口,具体的结果类型为AsyncResult,例如:
@Async
public Future<String> asyncMethodWithReturnType(){}
//返回的结果类似
Future<String> future = ....;
//可以通过future.isDone() 判断是否执行完毕
//通过future.get()获取返回值