| | |
| | | package com.ruoyi.framework.config; |
| | | |
| | | 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 java.util.concurrent.RejectedExecutionHandler; |
| | | import java.util.concurrent.ScheduledExecutorService; |
| | | import java.util.concurrent.ScheduledThreadPoolExecutor; |
| | | import java.util.concurrent.ThreadPoolExecutor; |
| | | |
| | | /** |
| | | * 线程池配置 |
| | |
| | | private ThreadPoolProperties threadPoolProperties; |
| | | |
| | | @Bean(name = "threadPoolTaskExecutor") |
| | | @ConditionalOnProperty(prefix = "threadPoolTaskExecutor", name = "enabled", havingValue = "true") |
| | | @ConditionalOnProperty(prefix = "thread-pool", name = "enabled", havingValue = "true") |
| | | public ThreadPoolTaskExecutor threadPoolTaskExecutor() { |
| | | ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); |
| | | executor.setMaxPoolSize(threadPoolProperties.getMaxPoolSize()); |
| | | executor.setCorePoolSize(threadPoolProperties.getCorePoolSize()); |
| | | executor.setQueueCapacity(threadPoolProperties.getQueueCapacity()); |
| | | executor.setKeepAliveSeconds(threadPoolProperties.getKeepAliveSeconds()); |
| | | RejectedExecutionHandler handler; |
| | | switch (threadPoolProperties.getRejectedExecutionHandler()) { |
| | | case "CallerRunsPolicy": |
| | | handler = new ThreadPoolExecutor.CallerRunsPolicy(); |
| | | break; |
| | | case "DiscardOldestPolicy": |
| | | handler = new ThreadPoolExecutor.DiscardOldestPolicy(); |
| | | break; |
| | | case "DiscardPolicy": |
| | | handler = new ThreadPoolExecutor.DiscardPolicy(); |
| | | break; |
| | | default: |
| | | handler = new ThreadPoolExecutor.AbortPolicy(); |
| | | break; |
| | | } |
| | | RejectedExecutionHandler handler = ReflectUtils.newInstance(threadPoolProperties.getRejectedExecutionHandler().getClazz()); |
| | | executor.setRejectedExecutionHandler(handler); |
| | | return executor; |
| | | } |
| | |
| | | @Bean(name = "scheduledExecutorService") |
| | | protected ScheduledExecutorService scheduledExecutorService() { |
| | | return new ScheduledThreadPoolExecutor(threadPoolProperties.getCorePoolSize(), |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build()) { |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build()) { |
| | | @Override |
| | | protected void afterExecute(Runnable r, Throwable t) { |
| | | super.afterExecute(r, t); |