【自定义参数置线程池】SpringBoot 配自定义参数置线程池用于定时任务/异步任务等

import lombok.extern.slf4j.Slf4j;
import org.springframework.aop.interceptor.AsyncUncaughtExceptionHandler;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.AsyncConfigurer;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;
import java.util.concurrent.ThreadPoolExecutor;

/**
 * 重写默认线程池配置
 */
@Slf4j
@Configuration
@EnableAsync
public class ThreadPoolConfig implements AsyncConfigurer {

  @Value("${spring.task.execution.thread-name-prefix}")
  private String namePrefix;

  @Value("${spring.task.execution.pool.core-size}")
  private int corePoolSize;

  @Value("${spring.task.execution.pool.max-size}")
  private int maxPoolSize;

  @Value("${spring.task.execution.pool.queue-capacity}")
  private int queueCapacity;

  @Value("${spring.task.execution.pool.keep-alive}")
  private int keepAliveSeconds;

 
  @Override
  @Bean("myThreadPool")
  public ThreadPoolTaskExecutor getAsyncExecutor() {
    ThreadPoolTaskExecutor executor = new ThreadPoolTaskExecutor();
    // 最大线程数
    executor.setMaxPoolSize(maxPoolSize);
    // 核心线程数
    executor.setCorePoolSize(corePoolSize);
    // 任务队列的大小
    executor.setQueueCapacity(queueCapacity);
    // 线程前缀名
    executor.setThreadNamePrefix(namePrefix);
    // 线程存活时间
    executor.setKeepAliveSeconds(keepAliveSeconds);
    /*
      当poolSize已达到maxPoolSize，如何处理新任务（是拒绝还是交由其它线程处理）
      CallerRunsPolicy：不在新线程中执行任务，而是由调用者所在的线程来执行
     */
    executor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());
    executor.initialize();
    return executor;
  }
 
  /**
   * 异步任务中异常处理
   */
  @Override
  public AsyncUncaughtExceptionHandler getAsyncUncaughtExceptionHandler() {
    return (ex, method, params) -> {
        log.error("=========================={}=======================", ex.getMessage(), ex);
        log.error("exception method:{}", method.getName());
    };
  }
}

#线程池配置
# 线程池的前缀名称
spring.task.execution.thread-name-prefix: pool-admin-thread-
# 核心线程池数
spring.task.execution.pool.core-size: 5
# 最大线程池数
spring.task.execution.pool.max-size: 10
# 任务队列的容量
spring.task.execution.pool.queue-capacity: 5
# 非核心线程的存活时间
spring.task.execution.pool.keep-alive: 60

使用场景1:定时任务使用线程池 在启动类上打上@EnableScheduling注解

示例

@Component
public class UpdateNoticeStatusTask {

    @Async("myThreadPool")
    @Scheduled(cron = "0 0/1 * * * ?")
    public void updateStatus() {
  
    //每隔一分钟执行一次更新状态的操作


    }
}

使用场景2:异步任务使用线程池 在启动类上打上@EnableAsync注解

    @Async("myThreadPool")
    public void async() {
    //里面写异步任务

    }

使用该异步方法

SpringUtil.getBean(xxx.class).async);