带参数的组件component初始化分离为base和引用初始化两部分

2023-10-29

component初始化需要读取配置文件的参数，这种component的初始化，如果是直接在一个文件初始化，比较容易实现。

但是现在要把这个component分为两部分初始化，base部分放在公共仓库，引用base在config实现初始化。这样做，方便统一管理base，需要使用的模块，自行引入base初始化使用。但当初始化需要配置文件的参数时，实现会麻烦一些。

本文提供一种方式，实现component的分离，同时支持初始化参数传入。

ThreadPoolExecutorConfig

这是单文件初始化的component

import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

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

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 自定义线程池
 *     <p>使用注入 @Autowired private Executor threadPoolExecutor;
 * @date 2023/3/14 6:32 PM
 */
@Configuration
@EnableAsync
public class ThreadPoolExecutorConfig {

    @Value("${thread.pool.core-pool-size}")
    private Integer corePoolSize;

    @Value("${thread.pool.max-pool-size}")
    private Integer maxPoolSize;

    @Bean(name = "threadPoolExecutor")
    public Executor threadPoolExecutor() {

        ThreadPoolTaskExecutor threadPoolExecutor = new ThreadPoolTaskExecutor();

        // 核心线程数
        threadPoolExecutor.setCorePoolSize(corePoolSize);
        // 最大线程数
        threadPoolExecutor.setMaxPoolSize(maxPoolSize);

        // 队列长度
        threadPoolExecutor.setQueueCapacity(maxPoolSize * 10);

        // 线程空闲时间
        threadPoolExecutor.setKeepAliveSeconds(300);

        // 线程名字前缀
        threadPoolExecutor.setThreadNamePrefix("xxx-Executor-");

        /**
         * 拒绝处理策略 CallerRunsPolicy()：交由调用方线程运行，比如 main 线程。 AbortPolicy()：直接抛出异常。
         * DiscardPolicy()：直接丢弃。 DiscardOldestPolicy()：丢弃队列中最老的任务。
         */
        threadPoolExecutor.setRejectedExecutionHandler(new ThreadPoolExecutor.CallerRunsPolicy());

        // 线程初始化
        threadPoolExecutor.initialize();

        return threadPoolExecutor;
    }
}

初始化分离为两部分

BaseThreadPoolExecutor

base模块，业务模块引入使用。

import lombok.AllArgsConstructor;
import org.springframework.scheduling.annotation.EnableAsync;
import org.springframework.scheduling.concurrent.ThreadPoolTaskExecutor;

import java.util.concurrent.Executor;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 自定义线程池
 *     <p>使用注入 @Autowired private Executor threadPoolExecutor;
 * @date 2023/3/14 6:32 PM
 */
@EnableAsync
@AllArgsConstructor
public class BaseThreadPoolExecutor {

    private Integer corePoolSize;

    private Integer maxPoolSize;

    public Executor threadPoolExecutor() {

        ThreadPoolTaskExecutor threadPoolExecutor = new ThreadPoolTaskExecutor();

        // 核心线程数
        threadPoolExecutor.setCorePoolSize(corePoolSize);
        // 最大线程数
        threadPoolExecutor.setMaxPoolSize(maxPoolSize);

        // 队列长度
        threadPoolExecutor.setQueueCapacity(maxPoolSize * 10);

        // 线程空闲时间
        threadPoolExecutor.setKeepAliveSeconds(300);

        // 线程名字前缀
        threadPoolExecutor.setThreadNamePrefix("xxx-Executor-");

        /**
         * 拒绝处理策略 CallerRunsPolicy()：交由调用方线程运行，比如 main 线程。 AbortPolicy()：直接抛出异常。
         * DiscardPolicy()：直接丢弃。 DiscardOldestPolicy()：丢弃队列中最老的任务。
         */
        threadPoolExecutor.setRejectedExecutionHandler(
                new java.util.concurrent.ThreadPoolExecutor.CallerRunsPolicy());

        // 线程初始化
        threadPoolExecutor.initialize();

        return threadPoolExecutor;
    }
}

ThreadPoolExecutorConfig

初始化类，引入base实现。

import com.opengate.base.threadpool.BaseThreadPoolExecutor;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.util.concurrent.Executor;

/**
 * @author 曳戈泰尔
 * @version 1.0
 * @description 自定义线程池
 *     <p>使用注入 @Autowired private Executor threadPoolExecutor;
 * @date 2023/3/14 6:32 PM
 */
@Configuration
public class ThreadPoolExecutorConfig {

  @Value("${thread.pool.core-pool-size}")
  private Integer corePoolSize;

  @Value("${thread.pool.max-pool-size}")
  private Integer maxPoolSize;

  /** 注入Base模块的线程池，传入初始化参数 */
  @Bean(name = "threadPoolExecutor")
  public Executor baseThreadPoolExecutor() {

    return new BaseThreadPoolExecutor(corePoolSize, maxPoolSize).threadPoolExecutor();
  }
}