package com.ruoyi.framework.config;
|
|
import com.ruoyi.common.utils.Threads;
|
import lombok.extern.slf4j.Slf4j;
|
import org.apache.commons.lang3.concurrent.BasicThreadFactory;
|
import org.springframework.context.annotation.Bean;
|
import org.springframework.context.annotation.Configuration;
|
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;
|
|
import java.util.concurrent.ScheduledExecutorService;
|
import java.util.concurrent.ScheduledThreadPoolExecutor;
|
import java.util.concurrent.ThreadPoolExecutor;
|
|
/**
|
* 线程池配置
|
*
|
* @author ruoyi
|
**/
|
@Slf4j
|
@Configuration
|
public class ThreadPoolConfig
|
{
|
// 核心线程池大小
|
private int corePoolSize = 50;
|
|
// 最大可创建的线程数
|
private int maxPoolSize = 200;
|
|
// 队列最大长度
|
private int queueCapacity = 1000;
|
|
// 线程池维护线程所允许的空闲时间
|
private int keepAliveSeconds = 300;
|
|
@Bean(name = "threadPoolTaskExecutor")
|
public ThreadPoolTaskExecutor threadPoolTaskExecutor()
|
{
|
ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
|
// 核心线程池大小
|
executor.setCorePoolSize(corePoolSize);
|
// 最大可创建的线程数
|
executor.setMaxPoolSize(maxPoolSize);
|
// 队列最大长度
|
executor.setQueueCapacity(queueCapacity);
|
// 线程池维护线程所允许的空闲时间
|
executor.setKeepAliveSeconds(keepAliveSeconds);
|
// 线程名称前缀
|
executor.setThreadNamePrefix("async-task-executor-");
|
// 设置线程工厂 (用于给线程命名，便于排查问题)
|
executor.setThreadFactory(new BasicThreadFactory.Builder().namingPattern("async-task-%d").daemon(true).build());
|
// 线程池对拒绝任务 (无线程可用) 的处理策略 - 调用者运行策略
|
executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
|
// 初始化
|
executor.initialize();
|
return executor;
|
}
|
|
/**
|
* 执行周期性或定时任务
|
*/
|
@Bean(name = "scheduledExecutorService")
|
protected ScheduledExecutorService scheduledExecutorService()
|
{
|
return new ScheduledThreadPoolExecutor(corePoolSize,
|
new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build(),
|
new ThreadPoolExecutor.CallerRunsPolicy())
|
{
|
@Override
|
protected void afterExecute(Runnable r, Throwable t)
|
{
|
super.afterExecute(r, t);
|
Threads.printException(r, t);
|
}
|
};
|
}
|
}
|
