当前位置: 代码迷 >> 综合 >> Soul 网关源码阅读之 Nacos 数据同步方式
  详细解决方案

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

热度:58   发布时间:2023-12-15 21:33:28.0

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 文件中配置:

```javasoul :sync:nacos:url: localhost:8848namespace: 1c10d748-af86-43b9-8265-75f487d20c6cacm:enabled: falseendpoint: acm.aliyun.comnamespace:accessKey:secretKey:#url: 配置成你的nacos地址,集群环境请使用(,)分隔。# 其他参数配置,请参考naocs官网。

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

soul :sync:nacos:url: localhost:8848namespace: 1c10d748-af86-43b9-8265-75f487d20c6cacm:enabled: falseendpoint: acm.aliyun.comnamespace:accessKey:secretKey:

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

NacosDataChangedListener

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

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

updateRuleMap(getConfig(RULE_DATA_ID));
	@SneakyThrowsprivate 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:// 使用新的规则替换掉旧的 rulechanged.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;}

最终,调用如下方法将配置同步出去:

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

NacosCacheHandler

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

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

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

  相关解决方案