Nacos 中配置 Map 类型,不香!

2022-09-23 09:21:42 浏览数 (1)

大家好,我是君哥。

最近在使用 Nacos 过程中遇到一个场景,配置的字符串可以解析成 Map 类型使用,有一个配置如下:

代码语言:javascript复制
map:
  test: key1:value1,key2:value2,key3:value3

后来有同事建议 Nacos 可以直接配置成 Map 类型,后台使用 Java Map 类型获取就可以。配置如下:

代码语言:javascript复制
map:
  test:
    key1: value1
    key2: value2
    key3: value3

下面就来分享一下配置 Map 类型的过程中遇到的问题。

1 使用 Bean 方式获取配置

1.1 使用方式

参考网上的一些案例,第一个方式是把读取到的 Map 作为一个 Spring 的 Bean,一看代码就明白了。

代码语言:javascript复制
@Bean
@ConfigurationProperties(prefix = "map.test")
public Map<String, String> mapping() {
    return new HashMap<>();
}

1.2 槽点

这样确实可以把 Nacos 中读取到的配置转换成 Map 类型,但一个致命的槽点就是 mapping 这个 bean 不能自动刷新。这样如果修改了 Nacos 中配置,要想让配置生效,就必须重启应用服务,这怎么能接受呢?

2 ConfigurationProperties

2.1 使用方式

直接使用 @Value 和 @NacosValue 是获取不到值的。下面的这种方式,类的定义上加注解 @ConfigurationProperties,再定义一个变量,名称跟 Nacos 中配置的后缀一样,这样是可以获取到 Map 类型的配置的。

代码语言:javascript复制
@Component
@RefreshScope
@ConfigurationProperties(prefix = "map")
public class NacosRefresh {

    private Logger logger = LoggerFactory.getLogger(getClass());

    public void setTest(Map<String, String> test) {
        this.test = test;
    }

    private Map<String, String> test;
}

注意:上面的 setTest 方法是必须要的,不然 test 变量取不到值。

2.2 槽点

这样确实可以把 Nacos 中读取到的配置转换成 Map 类型,但是跟第一种方式一样,定义的 Map 类型变量不能自动刷新。

3 使用监听

Nacos API 提供了监听功能,可以监听配置的变化,对变化进行处理,只要在监听方法上增加 @NacosConfigListener 这个注解就可以生效。见下面代码:

代码语言:javascript复制
@Service
public class NacosListener {

    private Logger logger = LoggerFactory.getLogger(getClass());

    private Map<String, String> map = new HashMap<>();

    @NacosConfigListener(dataId = "maptest.yaml",groupId = "DEFAULT_GROUP")
    public void listener(String context){
        logger.info("================listener context:{}", context);
        if (StringUtils.isBlank(context)){
            return;
        }
        Yaml yaml = new Yaml();
        Map<String, Object> contextMap = yaml.load(context);
        Map<String, Object> map = (Map<String, Object>)contextMap.get("map");
        if (CollectionUtils.isEmpty(map)){
            return;
        }
        Map<String, String> test = (Map<String, String>) map.get("test");
        if (CollectionUtils.isEmpty(test)){
            return;
        }
        map.clear();
        map.putAll(test);
        map.forEach((k,v) -> logger.info("Entry in map, key:{},value:{}", k, v));
    }
}

这段代码是从 Nacos 配置中解析出 Map 类型的配置,然后把配置 put 到本地变量 map。这个也可以完成我们的需求,但是有几点需要注意。

3.1 服务重启

如果服务重启了,本地变量 map 拉不到值。因为上面监听的逻辑并没有走,即使在 Nacos 上重新发布一下,也不行。

上面的监听方法,只有在 Nacos 配置发生变化并且发布后才会触发,比如 map.test 配置改变如下:

代码语言:javascript复制
map:
  test:
    key1: value1
    key2: value2
    key3: value3
    key4: value4

3.2 并发问题

上面监听的代码里面,需要把本地变量 map 先 clear 然后再 putAll,如果这两个方法调用中间发生了线程上下文切换,读取线程可能会因为从 map 中取不到值而发生异常。

4 改进

上面讲解了使用 Nacos 配置 Map 类型的坑,不过使用 Nacos 配置 Map 类型也有个好处,不用解析字符串,直接可以转成 Map 类型。

4.1 使用字符串

完全不使用 Map 类型了,改成配置字符串,配置如下:

代码语言:javascript复制
map:
  test: key1:value1,key2:value2,key3:value3

解析代码如下:

代码语言:javascript复制
@NacosValue(value = "${map.test}", autoRefreshed = true)
private String mapTest;

public String get(String key){
    String[] keys = mapTest.split(",");
    for (String item : keys){
        if (!item.contains(key)){
            continue;
        }
        return item.split(":")[1];
    }
    return null;
}

这种写法的好处是不用监听 Nacos,配置改变后 mapTest 变量自动刷新,缺点是每次调用 get 方法都需要解析 mapTest 这个字符串。

4.2 刷新本地 Map

把解析字符串的结果放到本地变量 map 上,考虑到 Nacos 中配置可能会发生变化,用定时线程池每 1 秒刷新一次,代码如下:

代码语言:javascript复制
private Map<String, String> map = new HashMap<>();

@NacosValue(value = "${map.test}", autoRefreshed = true)
private String mapTest;

@PostConstruct
public void refreshLocalMap(){
    ScheduledThreadPoolExecutor scheduled = new ScheduledThreadPoolExecutor(1);
    scheduled.scheduleAtFixedRate(() -> refresh(), 0, 1000, TimeUnit.MILLISECONDS);
}

public void refresh(){
    String[] keys = mapTest.split(",");
    for (String item : keys){
        String[] kv = item.split(":");
        map.put(kv[0], kv[1]);
    }
}

这个写法的好处是不用每次调用都解析字符串,而是由异步线程每秒钟刷新。但是也有两个问题:

  • 需要一个定时线程池,会消耗 CPU 资源。
  • refresh 方法是每秒执行一次,会有短暂的本地变量和 Nacos 配置不一致的问题。

5 总结

Nacos 中配置 Map 类型确实不香,主要原因是刷新不方便。但是对于配置不需要刷新的场景,还是很有好处的,尤其是 key 比较多的时候,比解析字符串方便很多,而且 Hash 的时间复杂度是 o(1) ,在数据结构中是最优秀的。

对于需要刷新的场景,无论使用哪种方案,都有优缺点,没有最好的,只有最适合的,要根据系统的业务场景来做选择。

0 人点赞