io.github.xiaoxuetu.threadpool.MultiThreadPoolAutoConfigure Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of multithreadpool-spring-boot-starter Show documentation
spring boot starter for multi threadpools
The newest version!
package io.github.xiaoxuetu.threadpool;

import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.task.TaskDecorator;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.util.Map;

/**
 * 线程池自动配置器
 */
@Slf4j
@Configuration
@EnableConfigurationProperties(MultiThreadPoolProperties.class)
public class MultiThreadPoolAutoConfigure {

    /**
     * 线程池属性
     */
    private MultiThreadPoolProperties multiThreadPoolProperties;

    /**
     * 应用上线文
     */
    private ApplicationContext applicationContext;

    /**
     * 任务修饰器
     */
    private final ObjectProvider taskDecorator;

    /**
     * 线程池缓存
     */
    @Resource
    private MultiThreadPoolManager multiThreadPoolManager;

    /**
     * 构造方法
     *
     * @param multiThreadPoolProperties 线程池配置
     * @param applicationContext        应用上下文
     * @param taskDecorator             任务修饰器
     */
    public MultiThreadPoolAutoConfigure(MultiThreadPoolProperties multiThreadPoolProperties,
                                        ApplicationContext applicationContext,
                                        ObjectProvider taskDecorator) {
        this.multiThreadPoolProperties = multiThreadPoolProperties;
        this.applicationContext = applicationContext;
        this.taskDecorator = taskDecorator;
        ;
    }

    /**
     * 初始化线程池
     * 
     * 目前初始化时机是在该类初始化后
     */
    @PostConstruct
    public void init() {
        for (MultiThreadPoolProperties.ThreadPoolProperties properties : multiThreadPoolProperties.getExecutors()) {
            multiThreadPoolManager
                    .createThreadPoolTaskExecutor((ConfigurableApplicationContext) applicationContext,
                    properties, taskDecorator.getIfUnique());
        }

    }


    /**
     * 获取线程池管理器
     *
     * @return 线程池管理器
     */
    @Bean
    @ConditionalOnMissingBean
    public MultiThreadPoolManager multiThreadPoolManager() {
        return new MultiThreadPoolManager();
    }
}