| | |
| | | |
| | | import java.util.concurrent.ScheduledExecutorService; |
| | | import java.util.concurrent.ScheduledThreadPoolExecutor; |
| | | import java.util.concurrent.ThreadFactory; |
| | | import java.util.concurrent.ThreadPoolExecutor; |
| | | |
| | | /** |
| | |
| | | * 核心线程数 = cpu 核心数 + 1 |
| | | */ |
| | | private final int core = Runtime.getRuntime().availableProcessors() + 1; |
| | | |
| | | private ScheduledExecutorService scheduledExecutorService; |
| | | |
| | | @Bean(name = "threadPoolTaskExecutor") |
| | | @ConditionalOnProperty(prefix = "thread-pool", name = "enabled", havingValue = "true") |
| | |
| | | */ |
| | | @Bean(name = "scheduledExecutorService") |
| | | protected ScheduledExecutorService scheduledExecutorService() { |
| | | return new ScheduledThreadPoolExecutor(core, |
| | | new BasicThreadFactory.Builder().namingPattern("schedule-pool-%d").daemon(true).build(), |
| | | ThreadFactory threadFactory; |
| | | // 是否启用虚拟线程 |
| | | if (SpringUtils.isVirtual()) { |
| | | // 虚拟线程必须为守护线程,即 daemon 只能是 true |
| | | threadFactory = new BasicThreadFactory.Builder() |
| | | .daemon(true) |
| | | .namingPattern("virtual-schedule-pool-%d") |
| | | .wrappedFactory(Thread.ofVirtual().factory()) |
| | | .build(); |
| | | } else { |
| | | threadFactory = new BasicThreadFactory.Builder() |
| | | .daemon(true) |
| | | .namingPattern("schedule-pool-%d") |
| | | .build(); |
| | | } |
| | | ScheduledThreadPoolExecutor scheduledThreadPoolExecutor = new ScheduledThreadPoolExecutor(core, |
| | | threadFactory, |
| | | 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); |