本文已参与「新人创作礼」活动,一起开启掘金创作之路。
一、导读
java多线程的使用很多,这里介绍一下springboot中的多线程的使用,使用线程连接池。源码地址传送门
二、关键代码
- CustomThreadPoolTaskExecutor 类代码
public class CustomThreadPoolTaskExecutor extends ThreadPoolTaskExecutor {
private void showThreadPoolInfo(String prefix) {
ThreadPoolExecutor threadPoolExecutor = getThreadPoolExecutor();
if (null == threadPoolExecutor) {
return;
}
log.info("{}, {},taskCount [{}], completedTaskCount [{}], activeCount [{}], queueSize [{}]",
this.getThreadNamePrefix(), prefix, threadPoolExecutor.getTaskCount(),
threadPoolExecutor.getCompletedTaskCount(), threadPoolExecutor.getActiveCount(),
threadPoolExecutor.getQueue().size());
}
@Override
public void execute(Runnable task) {
showThreadPoolInfo("1. do execute");
super.execute(task);
}
@Override
public void execute(Runnable task, long startTimeout) {
showThreadPoolInfo("2. do execute");
super.execute(task, startTimeout);
}
@Override
public Future<?> submit(Runnable task) {
showThreadPoolInfo("1. do submit");
return super.submit(task);
}
@Override
public <T> Future<T> submit(Callable<T> task) {
showThreadPoolInfo("2. do submit");
return super.submit(task);
}
@Override
public ListenableFuture<?> submitListenable(Runnable task) {
showThreadPoolInfo("1. do submitListenable");
return super.submitListenable(task);
}
@Override
public <T> ListenableFuture<T> submitListenable(Callable<T> task) {
showThreadPoolInfo("2. do submitListenable");
return super.submitListenable(task);
}
}
- ThreadPoolConfig 配置类代码
@Configuration
@EnableAsync
public class ThreadPoolConfig {
@Value("${thread.pool.coreSize}")
private int coreSize;
@Value("${thread.pool.maxSize}")
private int maxSize;
@Value("${thread.pool.queueSize}")
private int queueSize;
@Value("${thread.pool.threadNamePrefix}")
private String threadNamePrefix;
/**
* 线程池配置bean
* @return Executor
*/
@Bean
public Executor asyncServiceExecutor() {
log.info("start asyncServiceExecutor");
// 使用 自定义 CustomThreadPoolTaskExecutor
ThreadPoolTaskExecutor executor = new CustomThreadPoolTaskExecutor();
// 配置核心线程数
executor.setCorePoolSize(coreSize);
// 配置最大线程数
executor.setMaxPoolSize(maxSize);
// 配置队列大小
executor.setQueueCapacity(queueSize);
// 配置线程池中的线程的名称前缀
executor.setThreadNamePrefix(threadNamePrefix);
// rejection-policy:当pool已经达到max size的时候,如何处理新任务
// CALLER_RUNS:不在新线程中执行任务,而是有调用者所在的线程来执行
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
// 执行初始化
executor.initialize();
return executor;
}
}
- ThreadService 使用类
@Service
public class ThreadService {
/**
* 带返回值
* @return
*/
@Async("asyncServiceExecutor")
public Future<String> index1() {
log.info("start executeAsync");
try {
Thread.sleep(2000);
} catch (Exception e) {
e.printStackTrace();
}
log.info("end executeAsync");
return new AsyncResult<>("1");
}
/**
* 不带返回值
* @return
*/
@Async("asyncServiceExecutor")
public void index2() {
log.info("start executeAsync");
try {
Thread.sleep(2000);
} catch (Exception e) {
e.printStackTrace();
}
log.info("end executeAsync");
}
}
三、后记
- 最后说一下线程池,引用阿里巴巴开发规范的解释
- 【强制】线程池不允许使用Executors去创建,而是通过ThreadPoolExecutor的方式,这样的处理方式让写的同学更加明确线程池的运行规则,规避资源耗尽的风险。
说明:Executors各个方法的弊端:
1) newFixedThreadPool和newSingleThreadExecutor: 主要问题是堆积的请求处理队列可能会耗费非常大的内存,甚至OOM。
2) newCachedThreadPool和newScheduledThreadPool: 主要问题是线程数最大数是Integer.MAX_VALUE,可能会创建数量非常多的线程,甚至OOM。
- 【强制】创建线程或线程池时请指定有意义的线程名称,方便出错时回溯。
- 现实程序中是绝对要避免直接显示创建线程,而不使用线程池。显示创建线程会导致资源不合理利用已经导致oom的问题,线程创建和销毁都会消耗巨大资源,使用线程池来管理线程是明智的选择。