| | |
| | | package com.ruoyi.framework.config; |
| | | |
| | | import java.util.concurrent.ScheduledExecutorService; |
| | | import java.util.concurrent.ScheduledThreadPoolExecutor; |
| | | import java.util.concurrent.ThreadPoolExecutor; |
| | | import com.ruoyi.common.utils.Threads; |
| | | import com.ruoyi.common.utils.reflect.ReflectUtils; |
| | | import com.ruoyi.framework.config.properties.ThreadPoolProperties; |
| | | import org.apache.commons.lang3.concurrent.BasicThreadFactory; |
| | | import org.springframework.beans.factory.annotation.Autowired; |
| | | import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
| | | import org.springframework.context.annotation.Bean; |
| | | import org.springframework.context.annotation.Configuration; |
| | | import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
| | | import com.ruoyi.common.utils.Threads; |
| | | |
| | | import java.util.concurrent.RejectedExecutionHandler; |
| | | import java.util.concurrent.ScheduledExecutorService; |
| | | import java.util.concurrent.ScheduledThreadPoolExecutor; |
| | | |
| | | /** |
| | | * 线程池配置 |
| | | * |
| | | * @author ruoyi |
| | | * @author Lion Li |
| | | **/ |
| | | @Configuration |
| | | public class ThreadPoolConfig |
| | | { |
| | | // 核心线程池大小 |
| | | private int corePoolSize = 50; |
| | | public class ThreadPoolConfig { |
| | | |
| | | // 最大可创建的线程数 |
| | | private int maxPoolSize = 200; |
| | | |
| | | // 队列最大长度 |
| | | private int queueCapacity = 1000; |
| | | |
| | | // 线程池维护线程所允许的空闲时间 |
| | | private int keepAliveSeconds = 300; |
| | | @Autowired |
| | | private ThreadPoolProperties threadPoolProperties; |
| | | |
| | | @Bean(name = "threadPoolTaskExecutor") |
| | | public ThreadPoolTaskExecutor threadPoolTaskExecutor() |
| | | { |
| | | @ConditionalOnProperty(prefix = "thread-pool", name = "enabled", havingValue = "true") |
| | | public ThreadPoolTaskExecutor threadPoolTaskExecutor() { |
| | | ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); |
| | | executor.setMaxPoolSize(maxPoolSize); |
| | | executor.setCorePoolSize(corePoolSize); |
| | | executor.setQueueCapacity(queueCapacity); |
| | | executor.setKeepAliveSeconds(keepAliveSeconds); |
| | | // 线程池对拒绝任务(无线程可用)的处理策略 |
| | | executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); |
| | | executor.setMaxPoolSize(threadPoolProperties.getMaxPoolSize()); |
| | | executor.setCorePoolSize(threadPoolProperties.getCorePoolSize()); |
| | | executor.setQueueCapacity(threadPoolProperties.getQueueCapacity()); |
| | | executor.setKeepAliveSeconds(threadPoolProperties.getKeepAliveSeconds()); |
| | | RejectedExecutionHandler handler = ReflectUtils.newInstance(threadPoolProperties.getRejectedExecutionHandler().getClazz()); |
| | | executor.setRejectedExecutionHandler(handler); |
| | | return executor; |
| | | } |
| | | |
| | |
| | | * 执行周期性或定时任务 |
| | | */ |
| | | @Bean(name = "scheduledExecutorService") |
| | | protected ScheduledExecutorService scheduledExecutorService() |
| | | { |
| | | return new ScheduledThreadPoolExecutor(corePoolSize, |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build()) |
| | | { |
| | | protected ScheduledExecutorService scheduledExecutorService() { |
| | | return new ScheduledThreadPoolExecutor(threadPoolProperties.getCorePoolSize(), |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build()) { |
| | | @Override |
| | | protected void afterExecute(Runnable r, Throwable t) |
| | | { |
| | | protected void afterExecute(Runnable r, Throwable t) { |
| | | super.afterExecute(r, t); |
| | | Threads.printException(r, t); |
| | | } |