| | |
| | | import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; |
| | | import org.springframework.boot.context.properties.EnableConfigurationProperties; |
| | | import org.springframework.context.annotation.Bean; |
| | | import org.springframework.core.task.VirtualThreadTaskExecutor; |
| | | import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor; |
| | | |
| | | import java.util.concurrent.ScheduledExecutorService; |
| | |
| | | */ |
| | | private final int core = Runtime.getRuntime().availableProcessors() + 1; |
| | | |
| | | private ScheduledExecutorService scheduledExecutorService; |
| | | |
| | | @Bean(name = "threadPoolTaskExecutor") |
| | | @ConditionalOnProperty(prefix = "thread-pool", name = "enabled", havingValue = "true") |
| | | public ThreadPoolTaskExecutor threadPoolTaskExecutor(ThreadPoolProperties threadPoolProperties) { |
| | |
| | | */ |
| | | @Bean(name = "scheduledExecutorService") |
| | | protected ScheduledExecutorService scheduledExecutorService() { |
| | | return new ScheduledThreadPoolExecutor(core, |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build(), |
| | | // daemon 必须为 true |
| | | BasicThreadFactory.Builder builder = new BasicThreadFactory.Builder().daemon(true); |
| | | if (SpringUtils.isVirtual()) { |
| | | builder.namingPattern("virtual-schedule-pool-%d").wrappedFactory(new VirtualThreadTaskExecutor().getVirtualThreadFactory()); |
| | | } else { |
| | | builder.namingPattern("schedule-pool-%d"); |
| | | } |
| | | ScheduledThreadPoolExecutor scheduledThreadPoolExecutor = new ScheduledThreadPoolExecutor(core, |
| | | builder.build(), |
| | | new ThreadPoolExecutor.CallerRunsPolicy()) { |
| | | @Override |
| | | protected void afterExecute(Runnable r, Throwable t) { |
| | |
| | | Threads.printException(r, t); |
| | | } |
| | | }; |
| | | this.scheduledExecutorService = scheduledThreadPoolExecutor; |
| | | return scheduledThreadPoolExecutor; |
| | | } |
| | | |
| | | /** |
| | |
| | | public void destroy() { |
| | | try { |
| | | log.info("====关闭后台任务任务线程池===="); |
| | | ScheduledExecutorService scheduledExecutorService = SpringUtils.getBean("scheduledExecutorService"); |
| | | Threads.shutdownAndAwaitTermination(scheduledExecutorService); |
| | | } catch (Exception e) { |
| | | log.error(e.getMessage(), e); |