| | |
| | | import org.springframework.context.annotation.Configuration; |
| | | import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
| | | |
| | | import java.util.concurrent.RejectedExecutionHandler; |
| | | import java.util.concurrent.ScheduledExecutorService; |
| | | import java.util.concurrent.ScheduledThreadPoolExecutor; |
| | | import java.util.concurrent.ThreadPoolExecutor; |
| | |
| | | @Configuration |
| | | public class ThreadPoolConfig { |
| | | |
| | | /** |
| | | * 核心线程数 = cpu 核心数 + 1 |
| | | */ |
| | | private final int core = Runtime.getRuntime().availableProcessors() + 1; |
| | | |
| | | @Autowired |
| | | private ThreadPoolProperties threadPoolProperties; |
| | | |
| | |
| | | @ConditionalOnProperty(prefix = "thread-pool", name = "enabled", havingValue = "true") |
| | | public ThreadPoolTaskExecutor threadPoolTaskExecutor() { |
| | | ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor(); |
| | | executor.setMaxPoolSize(threadPoolProperties.getMaxPoolSize()); |
| | | executor.setCorePoolSize(threadPoolProperties.getCorePoolSize()); |
| | | executor.setMaxPoolSize(core); |
| | | executor.setCorePoolSize(core * 2); |
| | | 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; |
| | | } |
| | | executor.setRejectedExecutionHandler(handler); |
| | | executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy()); |
| | | return executor; |
| | | } |
| | | |
| | |
| | | */ |
| | | @Bean(name = "scheduledExecutorService") |
| | | protected ScheduledExecutorService scheduledExecutorService() { |
| | | return new ScheduledThreadPoolExecutor(threadPoolProperties.getCorePoolSize(), |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build()) { |
| | | return new ScheduledThreadPoolExecutor(core, |
| | | 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); |