Soul 网关源码阅读之 Nacos 数据同步方式

Nacos 是阿里推出的分布式注册中心，提供诸如服务注册发现，动态配置等功能，Soul 支持使用这些特性进行数据同步。

开启 Nocos 同步功能

和 zookeeper 相同，先在 bootstrap 的 pom 文件中加入如下配置：

	<dependency>
           <groupId>org.dromara</groupId>
           <artifactId>soul-spring-boot-starter-sync-data-nacos</artifactId>
           <version>${last.version}</version>
     </dependency>

然后在 yml 文件中配置：

```java
 soul :
     sync:
        nacos:
             url: localhost:8848
             namespace: 1c10d748-af86-43b9-8265-75f487d20c6c
             acm:
               enabled: false
               endpoint: acm.aliyun.com
               namespace:
               accessKey:
               secretKey:
 #url: 配置成你的nacos地址，集群环境请使用（,）分隔。
 # 其他参数配置，请参考naocs官网。

接下来在 admin 的 yml 文件中配置：

soul :
      sync:
         nacos:
              url: localhost:8848
              namespace: 1c10d748-af86-43b9-8265-75f487d20c6c
              acm:
                enabled: false
                endpoint: acm.aliyun.com
                namespace:
                accessKey:
                secretKey:

即可使用本地的 Nacos 进行数据同步。

NacosDataChangedListener

同之前介绍的同步策略类似，org.dromara.soul.admin.listener.nacos.NacosDataChangedListener 也实现了 DataChangedListener 接口，不过，在 Nacos 中，还多了一些 ConcurrentHashMap 来作为数据处理的前置工具。

仍然使用选择器规则的修改举例，在 onRuleChanged 方法中，首先调用了两个方法：

updateRuleMap(getConfig(RULE_DATA_ID));

	@SneakyThrows
    private String getConfig(final String dataId) {
        String config = configService.getConfig(dataId, GROUP, 6000);
        return StringUtils.hasLength(config) ? config : EMPTY_CONFIG_DEFAULT_VALUE;
    }
    
	private void updateRuleMap(final String configInfo) {
        JsonObject jo = GsonUtils.getInstance().fromJson(configInfo, JsonObject.class);
        Set<String> set = new HashSet<>(RULE_MAP.keySet());
        for (Entry<String, JsonElement> e : jo.entrySet()) {
            set.remove(e.getKey());
            List<RuleData> ls = new ArrayList<>();
            e.getValue().getAsJsonArray().forEach(je -> ls.add(GsonUtils.getInstance().fromJson(je, RuleData.class)));
            RULE_MAP.put(e.getKey(), ls);
        }
        RULE_MAP.keySet().removeAll(set);
    }

其中 @SneakyThrows 注解是 lombok 提供的功能，用于异常处理的简化。可以看到，此逻辑先使用 getConfig 方法获取最新的配置内容，再同本地的 JVM 内存去重合并。

接下来，针对用户在界面上不同的操作，对 ConcurrentHashMap 中的缓存实施不同的操作：

		switch (eventType) {
            case DELETE:
            	// 如果是删除操作，就将此规则从 map 中删除
                changed.forEach(rule -> {
                    List<RuleData> ls = RULE_MAP
                            .getOrDefault(rule.getSelectorId(), new ArrayList<>())
                            .stream()
                            .filter(s -> !s.getId().equals(rule.getId()))
                            .sorted(RULE_DATA_COMPARATOR)
                            .collect(Collectors.toList());
                    RULE_MAP.put(rule.getSelectorId(), ls);
                });
                break;
            case REFRESH:
            case MYSELF:
            	// 构建一个空的规则放入 map 中
                Set<String> set = new HashSet<>(RULE_MAP.keySet());
                changed.forEach(rule -> {
                    set.remove(rule.getSelectorId());
                    List<RuleData> ls = RULE_MAP
                            .getOrDefault(rule.getSelectorId(), new ArrayList<>())
                            .stream()
                            .sorted(RULE_DATA_COMPARATOR)
                            .collect(Collectors.toList());
                    RULE_MAP.put(rule.getSelectorId(), ls);
                });
                RULE_MAP.keySet().removeAll(set);
                break;
            default:
            	// 使用新的规则替换掉旧的 rule
                changed.forEach(rule -> {
                    List<RuleData> ls = RULE_MAP
                            .getOrDefault(rule.getSelectorId(), new ArrayList<>())
                            .stream()
                            .filter(s -> !s.getId().equals(rule.getSelectorId()))
                            .sorted(RULE_DATA_COMPARATOR)
                            .collect(Collectors.toList());
                    ls.add(rule);
                    RULE_MAP.put(rule.getSelectorId(), ls);
                });
                break;
        }

最终，调用如下方法将配置同步出去：

	@SneakyThrows
    private void publishConfig(final String dataId, final Object data) {
        configService.publishConfig(dataId, GROUP, GsonUtils.getInstance().toJson(data));
    }

NacosCacheHandler

接收端使用 Nacos 的 ConfigService 来进行监听器的创建：

	@SneakyThrows
    private String getConfigAndSignListener(final String dataId, final Listener listener) {
        return configService.getConfigAndSignListener(dataId, GROUP, 6000, listener);
    }

同之前的同步方案类似，最终同步到缓存模块中。