Kitty中的动态线程池支持Nacos,Apollo多配置中心了

目录

• 回顾昨日
• nacos集成
  • Spring Cloud Alibaba 方式
  • Nacos Spring Boot 方式
• Apollo集成
• 自研配置中心对接
• 无配置中心对接
• 实现源码分析
  • 兼容Apollo和Nacos NoClassDefFoundError
  • Apollo自动刷新问题

回顾昨日

上篇文章 《一时技痒，撸了个动态线程池，源码放Github了》（https://mp.weixin.qq.com/s/JM9idgFPZGkRAdCpw0NaKw）发出后很多读者私下问我这个能不能用到工作中，用肯定是可以用的，本身来说是对线程池的扩展，然后对接了配置中心和监控。

目前用的话主要存在下面几个问题：

• 还没发布到Maven中央仓库（后续会做），可以自己编译打包发布到私有仓库（临时方案）
• 耦合了Nacos，如果你项目中没有用Nacos或者用的其他的配置中心怎么办？（本文内容）
• 只能替换业务线程池，像一些框架中的线程池无法替换（构思中）

本文的重点就是介绍如何对接Nacos 和 Apollo，因为一开始就支持了Nacos，但是支持的方式是依赖了Spring Cloud Alibaba ，如果是没有用Spring Cloud Alibaba 如何支持，也是需要扩展的。

Nacos集成

Nacos集成的话分两种方式，一种是你的项目使用了Spring Cloud Alibaba ，另一种是只用了Spring Boot 方式的集成。

Spring Cloud Alibaba方式

加入依赖：

<dependency>
    <groupId>com.cxytiandi</groupId>
    <artifactId>kitty-spring-cloud-starter-dynamic-thread-pool</artifactId>
</dependency>

然后在Nacos中增加线程池的配置，比如：

kitty.threadpools.executors[0].threadPoolName=TestThreadPoolExecutor
kitty.threadpools.executors[0].corePoolSize=4
kitty.threadpools.executors[0].maximumPoolSize=4
kitty.threadpools.executors[0].queueCapacity=5
kitty.threadpools.executors[0].queueCapacityThreshold=22

然后在项目中的bootstrap.properties中配置要使用的Nacos data-id。

spring.cloud.nacos.config.ext-config[0].data-id=kitty-cloud-thread-pool.properties
spring.cloud.nacos.config.ext-config[0].group=BIZ_GROUP
spring.cloud.nacos.config.ext-config[0].refresh=true

Nacos Spring Boot方式

如果你的项目只是用了Nacos的Spring Boot Starter，比如下面：

<dependency>
  <groupId>com.alibaba.boot</groupId>
  <artifactId>nacos-config-spring-boot-starter</artifactId>
</dependency>

那么集成的步骤跟Spring Cloud Alibaba方式一样，唯一不同的就是配置的加载方式。使用@NacosPropertySource进行加载。

@NacosPropertySource(dataId = NacosConstant.HREAD_POOL, groupId = NacosConstant.BIZ_GROUP, autoRefreshed = true, type = ConfigType.PROPERTIES)
public class Application {
    public static void main(String[] args) {
        SpringApplication.run(Application.class, args);
    }
}

然后需要在bootstrap.properties中关闭Spring Cloud Alibaba Nacos Config的自动配置。

spring.cloud.nacos.config.enabled=false

Apollo集成

Apollo的使用我们都是用它的client，依赖如下：

  <dependency>
      <groupId>com.ctrip.framework.apollo</groupId>
      <artifactId>apollo-client</artifactId>
      <version>1.4.0</version>
  </dependency>

集成Thread-Pool还是老的步骤，先添加Maven依赖：

<dependency>
    <groupId>com.cxytiandi</groupId>
    <artifactId>kitty-spring-cloud-starter-dynamic-thread-pool</artifactId>
</dependency>

然后配置线程池配置的namespace：

apollo.bootstrap.namespaces=thread-pool-config

Properties不用加后缀，如果是yaml文件那么需要加上后缀：

apollo.bootstrap.namespaces=thread-pool-config.yaml

如果你项目中用到了多个namespace的话，需要在线程池的namespace中指定，主要是监听配置修改需要用到。

kitty.threadpools.apolloNamespace=thread-pool-config.yaml

自研配置中心对接

如果你们项目使用的是自研的配置中心那该怎么使用动态线程池呢？

最好的方式是跟Nacos一样，将配置跟Spring进行集成，封装成PropertySource。

Apollo中集成Spring代码参考：https://github.com/ctripcorp/apollo/blob/master/apollo-client/src/main/java/com/ctrip/framework/apollo/spring/config/PropertySourcesProcessor.java

因为配置类是用的@ConfigurationProperties，这样就相当于无缝集成了。

如果没和Spring进行集成，那也是有办法的，可以在项目启动后获取你们的配置，然后修改

DynamicThreadPoolProperties配置类，再初始化线程池即可，具体步骤跟下面的无配置中心对接一致。DynamicThreadPoolManager提供了createThreadPoolExecutor()来创建线程池。

无配置中心对接

如果你的项目中没有使用配置中心怎么办？还是可以照样使用动态线程池的。

直接将线程池的配置信息放在项目的application配置文件中即可，但是这样的缺点就是无法动态修改配置信息了。

如果想有动态修改配置的能力，可以稍微扩展下，这边我提供下思路。

编写一个Rest API，参数就是整个线程池配置的内容，可以是Properties文件也可以是Yaml文件格式。

这个API的逻辑就是注入我们的DynamicThreadPoolProperties，调用refresh()刷新Properties文件，调用refreshYaml()刷新Yaml文件。

然后注入DynamicThreadPoolManager，调用refreshThreadPoolExecutor()刷新线程池参数。

实现源码分析

首先，我们要实现的需求是同时适配Nacos和Apollo两个主流的配置中心，一般有两种做法。

第一种：将跟Nacos和Apollo相关的代码独立成一个模块，使用者按需引入。

第二种：还是一个项目，内部做兼容。

我这边采取的是第二种，因为代码量不多，没必要拆分成两个。

需要在pom中同时增加两个配置中心的依赖，需要设置成可选（optional=true）。

<dependency>
    <groupId>com.alibaba.cloud</groupId>
    <artifactId>spring-cloud-alibaba-nacos-config</artifactId>
    <optional>true</optional>
</dependency>
<dependency>
    <groupId>com.ctrip.framework.apollo</groupId>
    <artifactId>apollo-client</artifactId>
    <version>1.4.0</version>
    <optional>true</optional>
</dependency>

然后内部将监听配置动态调整线程池参数的逻辑分开，ApolloConfigUpdateListener和NacosConfigUpdateListener。

在自动装配Bean的时候按需装配对应的Listener。

@ImportAutoConfiguration(DynamicThreadPoolProperties.class)
@Configuration
public class DynamicThreadPoolAutoConfiguration {
   @Bean
   @ConditionalOnClass(value = com.alibaba.nacos.api.config.ConfigService.class)
   public NacosConfigUpdateListener nacosConfigUpdateListener() { 
       return new NacosConfigUpdateListener();
   }
   @Bean
   @ConditionalOnClass(value = com.ctrip.framework.apollo.ConfigService.class)
   public ApolloConfigUpdateListener apolloConfigUpdateListener() {
       return new ApolloConfigUpdateListener();
   }
   
}

兼容Apollo和Nacos NoClassDefFoundError

通过@ConditionalOnClass来判断当前项目中使用的是哪种配置中心，然后装配对应的Listener。上面的代码看上去没问题，在实际使用的过程去报了下面的错误：

Caused by: java.lang.NoClassDefFoundError: Lcom/alibaba/nacos/api/config/ConfigService;
	at java.lang.Class.getDeclaredFields0(Native Method) ~[na:1.8.0_40]
	at java.lang.Class.privateGetDeclaredFields(Class.java:2583) ~[na:1.8.0_40]
	at java.lang.Class.getDeclaredFields(Class.java:1916) ~[na:1.8.0_40]
	at org.springframework.util.ReflectionUtils.getDeclaredFields(ReflectionUtils.java:755) ~[spring-core-5.1.8.RELEASE.jar:5.1.8.RELEASE]
	... 22 common frames omitted
Caused by: java.lang.ClassNotFoundException: com.alibaba.nacos.api.config.ConfigService
	at java.net.URLClassLoader.findClass(URLClassLoader.java:381) ~[na:1.8.0_40]
	at java.lang.ClassLoader.loadClass(ClassLoader.java:424) ~[na:1.8.0_40]
	at sun.misc.Launcher$AppClassLoader.loadClass(Launcher.java:331) ~[na:1.8.0_40]
	at java.lang.ClassLoader.loadClass(ClassLoader.java:357) ~[na:1.8.0_40]
	... 26 common frames omitted

比如我的项目是用的Apollo，然后我集成了动态线程池，在启动的时候就报上面的错误了，错误原因是找不到Nacos相关的类。

但其实我已经用了@ConditionalOnClass来判断，这个是因为你的DynamicThreadPoolAutoConfiguration类是生效的，Spring会去装载DynamicThreadPoolAutoConfiguration类，DynamicThreadPoolAutoConfiguration中有NacosConfigUpdateListener的实例化操作，而项目中又没有依赖Nacos，所以就报错了。

这种情况我们需要将装配的逻辑拆分的更细，直接用一个单独的类去配置，将@ConditionalOnClass放在类上。

这里我采用了静态内部类的方式，如果项目中没有依赖Nacos，那么NacosConfiguration就不会生效，也就不会去初始化NacosConfigUpdateListener。

@Configuration
@ConditionalOnClass(value = com.alibaba.nacos.api.config.ConfigService.class)
protected static class NacosConfiguration {
    @Bean
    public NacosConfigUpdateListener nacosConfigUpdateListener() {
        return new NacosConfigUpdateListener();
    }
}
@Configuration
@ConditionalOnClass(value = com.ctrip.framework.apollo.ConfigService.class)
protected static class ApolloConfiguration {
    @Bean
    public ApolloConfigUpdateListener apolloConfigUpdateListener() {
        return new ApolloConfigUpdateListener();
    }
}

这个地方我顺便提一个点，就是为什么我们平时要多去看看开源框架的源码。因为像这种适配多个框架的逻辑比较常见，那么一些开源框架中肯定也有类似的逻辑。如果你之前有看过其他的框架是怎么实现的，那么这里你就会直接采取那种方式。

比如Spring Cloud OpenFeign中对Http的客户端做了多个框架的适配，你可以用HttpClient也可以用Okhttp，这不就是跟我们这个一样的逻辑么。

我们看下源码就知道了，如下图：

Apollo自动刷新问题

在实现的过程中还遇到一个问题也跟大家分享下，就是Apollo中@ConfigurationProperties配置类，在配置信息变更后不会自动刷新，需要配合RefreshScope或者EnvironmentChangeEvent来实现。

下图是Apollo文档的原话：

Nacos刷新是没问题的，只不过在收到配置变更的消息时，配置信息还没刷新到Bean里面去，所以再刷新的时候单独起了一个线程去做，然后在这个线程中睡眠了1秒钟（可通过配置调整）。

如果按照Apollo文档中给的方式，肯定是可以实现的。但是不太好，因为需要依赖Spring Cloud Context。主要是考虑到使用者并不一定会用到Spring Cloud，我们的基础是Spring Boot。

万一使用者就是在Spring Boot项目中用了Apollo, 然后又用了我的动态线程池，这怎么搞？

最后我采用了手动刷新的方式，当配置发生变更的时候，我会通过Apollo的客户端，重新拉取整个配置文件的内容，然后手动刷新配置类。

config.addChangeListener(changeEvent -> {
    ConfigFileFormat configFileFormat = ConfigFileFormat.Properties;
    String getConfigNamespace = finalApolloNamespace;
    if (finalApolloNamespace.contains(ConfigFileFormat.YAML.getValue())) {
        configFileFormat = ConfigFileFormat.YAML;
        // 去除.yaml后缀，getConfigFile时候会根据类型自动追加
        getConfigNamespace = getConfigNamespace.replaceAll("." + ConfigFileFormat.YAML.getValue(), "");
    }
    ConfigFile configFile = ConfigService.getConfigFile(getConfigNamespace, configFileFormat);
    String content = configFile.getContent();
    if (finalApolloNamespace.contains(ConfigFileFormat.YAML.getValue())) {
        poolProperties.refreshYaml(content);
    } else {
        poolProperties.refresh(content);
    }
    dynamicThreadPoolManager.refreshThreadPoolExecutor(false);
    log.info("线程池配置有变化，刷新完成");
});

刷新逻辑：

public void refresh(String content) {
    Properties properties =  new Properties();
    try {
        properties.load(new ByteArrayInputStream(content.getBytes()));
    } catch (IOException e) {
        log.error("转换Properties异常", e);
    }
    doRefresh(properties);
}
public void refreshYaml(String content) {
    YamlPropertiesFactoryBean bean = new YamlPropertiesFactoryBean();
    bean.setResources(new ByteArrayResource(content.getBytes()));
    Properties properties = bean.getObject();
    doRefresh(properties);
}
private void doRefresh(Properties properties) {
    Map<String, String> dataMap = new HashMap<String, String>((Map) properties);
    ConfigurationPropertySource sources = new MapConfigurationPropertySource(dataMap);
    Binder binder = new Binder(sources);
    binder.bind("kitty.threadpools", Bindable.ofInstance(this)).get();
}

目前只支持Properties和Yaml文件配置格式。

感兴趣的Star下呗：https://github.com/yinjihuan/kitty