欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2022-06-14 14:37:36
...

目录

  • 回顾昨日
  • 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,这不就是跟我们这个一样的逻辑么。

我们看下源码就知道了,如下图:

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

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

Apollo自动刷新问题

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

下图是Apollo文档的原话:

Kitty中的动态线程池支持Nacos,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

关于作者:尹吉欢,简单的技术爱好者,《Spring Cloud微服务-全栈技术与案例解析》, 《Spring Cloud微服务 入门 实战与进阶》作者, 公众号 猿天地 发起人。个人微信 jihuan900,欢迎勾搭。