java動(dòng)態(tài)線程池的簡(jiǎn)單實(shí)現(xiàn)思路

更新時(shí)間：2023年06月05日 08:31:49 作者：耶耶耶耶yeyeye

本文主要介紹了java?動(dòng)態(tài)線程池的簡(jiǎn)單實(shí)現(xiàn)思路，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

什么是動(dòng)態(tài)線程池？

在線程池日常實(shí)踐中我們常常會(huì)遇到以下問(wèn)題：

代碼中創(chuàng)建了一個(gè)線程池卻不知道核心參數(shù)設(shè)置多少比較合適。
參數(shù)設(shè)置好后，上線發(fā)現(xiàn)需要調(diào)整，改代碼重啟服務(wù)非常麻煩。
線程池相對(duì)于開(kāi)發(fā)人員來(lái)說(shuō)是個(gè)黑箱，運(yùn)行情況在出現(xiàn)問(wèn)題前很難被感知。

因此，動(dòng)態(tài)可監(jiān)控線程池一種針對(duì)以上痛點(diǎn)開(kāi)發(fā)的線程池管理工具。主要可實(shí)現(xiàn)功能有：提供對(duì) Spring 應(yīng)用內(nèi)線程池實(shí)例的全局管控、應(yīng)用運(yùn)行時(shí)動(dòng)態(tài)變更線程池參數(shù)以及線程池?cái)?shù)據(jù)采集和監(jiān)控閾值報(bào)警。

已經(jīng)實(shí)現(xiàn)的優(yōu)秀開(kāi)源動(dòng)態(tài)線程池

hippo4j、dynamic-tp.....

實(shí)現(xiàn)思路

核心管理類(lèi)

需要能實(shí)現(xiàn)對(duì)線程池的
服務(wù)注冊(cè)
獲取已經(jīng)注冊(cè)好的線程池

以及對(duì)注冊(cè)號(hào)線程池參數(shù)的刷新。

對(duì)于每一個(gè)線程池，我們使用一個(gè)線程池名字作為標(biāo)識(shí)每個(gè)線程池的唯一ID。

偽代碼實(shí)現(xiàn)

public class DtpRegistry {
    /**
     * 儲(chǔ)存線程池
     */
    private static final Map<String, Executor> EXECUTOR_MAP = new ConcurrentHashMap<>();
    /**
     * 獲取線程池
     * @param executorName 線程池名字
     */
    public static Executor getExecutor(String executorName) {
        return EXECUTOR_MAP.get(executorName);
    }
    /**
     * 線程池注冊(cè)
     * @param executorName 線程池名字
     */
    public static void registry(String executorName, Executor executor) {
        //注冊(cè)
        EXECUTOR_MAP.put(executorName, executorWrapper);
    }
    /**
     * 刷新線程池參數(shù)
     * @param executorName 線程池名字
     * @param properties 線程池參數(shù)
     */
    public static void refresh(String executorName, ThreadPoolProperties properties) {
        Executor executor = EXECUTOR_MAP.get(executorName)
        //刷新參數(shù)
        //.......
    }
}

如何創(chuàng)建線程池？

STEP 1. 我們可以使用yml配置文件的方式配置一個(gè)線程池，將線程池實(shí)例的創(chuàng)建交由Spring容器。

相關(guān)配置

public class DtpProperties {
? ? private List<ThreadPoolProperties> executors;
}
public class ThreadPoolProperties {
? ? /**
? ? ?* 標(biāo)識(shí)每個(gè)線程池的唯一名字
? ? ?*/
? ? private String poolName;
? ? private String poolType = "common";
? ? /**
? ? ?* 是否為守護(hù)線程
? ? ?*/
? ? private boolean isDaemon = false;
? ? /**
? ? ?* 以下都是核心參數(shù)
? ? ?*/
? ? private int corePoolSize = 1;
? ? private int maximumPoolSize = 1;
? ? private long keepAliveTime;
? ? private TimeUnit timeUnit = TimeUnit.SECONDS;
? ? private String queueType = "arrayBlockingQueue";
? ? private int queueSize = 5;
? ? private String threadFactoryPrefix = "-td-";
? ? private String RejectedExecutionHandler;
}

yml example:

spring:
  dtp:
    executors:
      # 線程池1
      - poolName: dtpExecutor1
        corePoolSize: 5
        maximumPoolSize: 10
      # 線程池2
      - poolName: dtpExecutor2
        corePoolSize: 2
        maximumPoolSize: 15

STEP 2 根據(jù)配置信息添加線程池的BeanDefinition

關(guān)鍵類(lèi)

@Slf4j
public class DtpImportBeanDefinitionRegistrar implements ImportBeanDefinitionRegistrar, EnvironmentAware {
? ? private Environment environment;
? ? @Override
? ? public void registerBeanDefinitions(AnnotationMetadata importingClassMetadata, BeanDefinitionRegistry registry) {
? ? ? ? log.info("注冊(cè)");
? ? ? ? //綁定資源
? ? ? ? DtpProperties dtpProperties = new DtpProperties();
? ? ? ? ResourceBundlerUtil.bind(environment, dtpProperties);
? ? ? ? List<ThreadPoolProperties> executors = dtpProperties.getExecutors();
? ? ? ? if (Objects.isNull(executors)) {
? ? ? ? ? ? log.info("未檢測(cè)本地到配置文件線程池");
? ? ? ? ? ? return;
? ? ? ? }
? ? ? ? //注冊(cè)beanDefinition
? ? ? ? executors.forEach((executorProp) -> {
? ? ? ? ? ? BeanUtil.registerIfAbsent(registry, executorProp);
? ? ? ? });
? ? }
? ? @Override
? ? public void setEnvironment(Environment environment) {
? ? ? ? this.environment = environment;
? ? }
}
/**
?*
?* 工具類(lèi)
?*
?*/
public class BeanUtil{
? ? public static void registerIfAbsent(BeanDefinitionRegistry registry, ThreadPoolProperties executorProp) {
? ? ? ? register(registry, executorProp.getPoolName(), executorProp);
? ? }
? ? public static void register(BeanDefinitionRegistry registry, String beanName, ThreadPoolProperties executorProp) {
? ? ? ? Class<? extends Executor> executorType = ExecutorType.getClazz(executorProp.getPoolType());
? ? ? ? Object[] args = assembleArgs(executorProp);
? ? ? ? register(registry, beanName, executorType, args);
? ? }
? ? public static void register(BeanDefinitionRegistry registry, String beanName, Class<?> clazz, Object[] args) {
? ? ? ? BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(clazz);
? ? ? ? for (Object arg : args) {
? ? ? ? ? ? builder.addConstructorArgValue(arg);
? ? ? ? }
? ? ? ? registry.registerBeanDefinition(beanName, builder.getBeanDefinition());
? ? }
? ? private static Object[] assembleArgs(ThreadPoolProperties executorProp) {
? ? ? ? return new Object[]{
? ? ? ? ? ? ? ? executorProp.getCorePoolSize(),
? ? ? ? ? ? ? ? executorProp.getMaximumPoolSize(),
? ? ? ? ? ? ? ? executorProp.getKeepAliveTime(),
? ? ? ? ? ? ? ? executorProp.getTimeUnit(),
? ? ? ? ? ? ? ? QueueType.getInstance(executorProp.getQueueType(), executorProp.getQueueSize()),
? ? ? ? ? ? ? ? new NamedThreadFactory(
? ? ? ? ? ? ? ? ? ? ? ? executorProp.getPoolName() + executorProp.getThreadFactoryPrefix(),
? ? ? ? ? ? ? ? ? ? ? ? executorProp.isDaemon()
? ? ? ? ? ? ? ? ),
? ? ? ? ? ? ? ? //先默認(rèn)不做設(shè)置
? ? ? ? ? ? ? ? RejectPolicy.ABORT.getValue()
? ? ? ? };
? ? }
}

下面解釋一下這個(gè)類(lèi)的作用，environment實(shí)例中儲(chǔ)存著spring啟動(dòng)時(shí)解析的yml配置，所以我們spring提供的Binder將配置綁定到我們前面定義的DtpProperties類(lèi)中，方便后續(xù)使用。接下來(lái)的比較簡(jiǎn)單，就是將線程池的BeanDefinition注冊(cè)到IOC容器中，讓spring去幫我們實(shí)例化這個(gè)bean。

STEP 3. 將已經(jīng)實(shí)例化的線程池注冊(cè)到核心類(lèi) DtpRegistry 中

我們注冊(cè)了 beanDefinition 后，spring會(huì)幫我們實(shí)例化出來(lái)，在這之后我們可以根據(jù)需要將這個(gè)bean進(jìn)行進(jìn)一步的處理，spring也提供了很多機(jī)制讓我們對(duì)bean的生命周期管理進(jìn)行更多的擴(kuò)展。對(duì)應(yīng)到這里我們就是將實(shí)例化出來(lái)的線程池注冊(cè)到核心類(lèi) DtpRegistry 中進(jìn)行管理。

這里我們使用 BeanPostProcessor 進(jìn)行處理。

@Slf4j
public class DtpBeanPostProcessor implements BeanPostProcessor {
? ? private DefaultListableBeanFactory beanFactory;
? ? @Override
? ? public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
? ? ? ? if (bean instanceof DtpExecutor) {
? ? ? ? ? ? //直接納入管理
? ? ? ? ? ? DtpRegistry.registry(beanName, (DtpExecutor) bean);
? ? ? ? }
? ? ? ? return bean;
? ? }
}

這里的邏輯很簡(jiǎn)單，就是判斷一下這個(gè)bean是不是線程池，是就統(tǒng)一管理起來(lái)。

STEP 4. 啟用 BeanDefinitionRegistrar 和 BeanPostProcessor

在springboot程序中，只要加一個(gè)@MapperScan注解就能啟用mybatis的功能，我們可以學(xué)習(xí)其在spring中的啟用方式，自定義一個(gè)注解：

@Target(ElementType.TYPE)
@Retention(RetentionPolicy.RUNTIME)
@Import(DtpImportSelector.class)
public @interface EnableDynamicThreadPool {
}

其中，比較關(guān)鍵的是@Import注解，spring會(huì)導(dǎo)入注解中的類(lèi)DtpImportSelector

而DtpImportSelector這個(gè)類(lèi)實(shí)現(xiàn)了：

public class DtpImportSelector implements DeferredImportSelector {
    @Override
    public String[] selectImports(AnnotationMetadata importingClassMetadata) {
        return new String[]{
                DtpImportBeanDefinitionRegistrar.class.getName(),
                DtpBeanPostProcessor.class.getName()
        };
    }
}

這樣，只要我們?cè)賳?dòng)類(lèi)或者配置類(lèi)上加上@EnableDynamicThreadPool這個(gè)注解，spring就會(huì)將DtpImportBeanDefinitionRegistrar和DtpBeanPostProcessor這兩個(gè)類(lèi)加入spring容器管理，從而實(shí)現(xiàn)我們的線程池的注冊(cè)。

@SpringBootApplication
@EnableDynamicThreadPool
public class Application {
    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }
}

如何實(shí)現(xiàn)線程池配置的動(dòng)態(tài)刷新

首先明確一點(diǎn)，對(duì)于線程池的實(shí)現(xiàn)類(lèi)，例如：ThreadPoolExecutor等，都有提供核心參數(shù)對(duì)應(yīng)的 set 方法，讓我們實(shí)現(xiàn)參數(shù)修改。因此，在核心類(lèi)DtpRegistry中的refresh方法，我們可以這樣寫(xiě)：

public class DtpRegistry {
    /**
     * 儲(chǔ)存線程池
     */
    private static final Map<String, ThreadPoolExecutor> EXECUTOR_MAP = new ConcurrentHashMap<>();
    /**
     * 刷新線程池參數(shù)
     * @param executorName 線程池名字
     * @param properties 線程池參數(shù)
     */
    public static void refresh(String executorName, ThreadPoolProperties properties) {
        ThreadPoolExecutor executor = EXECUTOR_MAP.get(executorName)
        //設(shè)置參數(shù)
        executor.setCorePoolSize(...);
        executor.setMaximumPoolSize(...);
        ......
    }
}

而這些新參數(shù)怎么來(lái)呢?我們可以引入Nacos、Apollo等配置中心，實(shí)現(xiàn)他們的監(jiān)聽(tīng)器方法，在監(jiān)聽(tīng)器方法里調(diào)用DtpRegistry的refresh方法刷新即可。

到此這篇關(guān)于java動(dòng)態(tài)線程池的簡(jiǎn)單實(shí)現(xiàn)思路的文章就介紹到這了,更多相關(guān)java動(dòng)態(tài)線程池內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: