SpringCloud系列教程(十二):网关配置动态路由

除了token以外,还有一个很实用的功能就是把网关的路由配置放到nacos上,并且修改路由配置的时候,网关服务可以动态的更新,这样我们在调整网络配置的时候,就不用重启服务了。所以我们需要用到两个重要的类:NacosConfigManager和RouteDefinitionWriter,NacosConfigManager用来监听nacos上配置的变化,RouteDefinitionWriter会执行路由配置。

1、将原来写在application.yml中的路由配置信息迁移到nacos上,dataId定义为gateway-routes,这里要注意,原来在项目中yml文件解析的时候,会被SpringCloud层层解析后生成RouteDefinition,在这期间SpringCloud会去解析比如Path=/api/*这种形式,但是现在我们要自己解析,就要改成符合RouteDefinition的完全形式,这是一个小难点。

spring:
  cloud:
    gateway:
      routes:
        - id: nacos-client-demo
          uri: lb://nacos-client-demo
          predicates: #断言,匹配访问的路径
            - name: Path
              args:
                pattern: /nacos-client-demo/api/**
          filters: # 过滤器
            - name: AddRequestHeader
              args:
                _genkey_0: headername
                _genkey_1: I am a header!
            - name: My
              args:
                _genkey_0: zhangsan
                _genkey_1: lisi
                _genkey_2: wangwu
        - id: open-feign-demo
          uri: lb://open-feign-demo
          predicates:
            - name: Path
              args:
                pattern: /open-feign-demo/api/**

2、在application.yml中添加nacos的配置信息就可以了,并且我特意添加了两个参数,用来指明动态路由配置的dataId,这样以后改成其他的也方便。

spring:
  main:
    # gateway组件中的spring-boot-starter-webflux和springboot作为web项目启动必不可少的spring-boot-starter-web出现冲突
    web-application-type: reactive
  application:
    name: gateway-demo
  cloud:
    gateway:
      nacos-data-id: gateway-routes
      nacos-group: devops
    nacos:
      server-addr: 192.168.3.54:80
      username: nacos
      password: nacos
      discovery:
        group: devops
        namespace: sit
      config:
        namespace: sit
        group: devops
  config:
    import: nacos:${spring.application.name}?refresh=true
server:
  port: 8888

3、定义一个component类DynamicRouterLoader,继承ApplicationEventPublisherAware,网上很多教程没有继承,因为版本不同,我用新版的发现必须要用ApplicationEventPublisherAware发布一下才能让路由信息生效。

package com.mj.gateway.router;

import cn.hutool.core.lang.Dict;
import cn.hutool.json.JSONUtil;
import cn.hutool.setting.yaml.YamlUtil;
import com.alibaba.cloud.nacos.NacosConfigManager;
import com.alibaba.fastjson2.JSONObject;
import com.alibaba.nacos.api.config.listener.Listener;
import com.alibaba.nacos.api.exception.NacosException;
import jakarta.annotation.PostConstruct;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.cloud.gateway.event.RefreshRoutesEvent;
import org.springframework.cloud.gateway.route.RouteDefinition;
import org.springframework.cloud.gateway.route.RouteDefinitionWriter;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.context.ApplicationEventPublisherAware;
import org.springframework.stereotype.Component;
import reactor.core.publisher.Mono;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.concurrent.Executor;

@Slf4j
@Component
@RequiredArgsConstructor
public class DynamicRouterLoader implements ApplicationEventPublisherAware {
    //nacos配置管理器
    private final NacosConfigManager nacosConfigManager;
    //gateway路由规则写入器
    private final RouteDefinitionWriter writer;

    private ApplicationEventPublisher publisher;

    @Value("${spring.cloud.gateway.nacos-data-id:test}")
    private String dataId;
    @Value("${spring.cloud.gateway.nacos-group:test}")
    private String group;
    private final int timeout = 5000;

    //保存所有的路由id
    private final Set<String> routeIds = new HashSet<>();

    //PostConstruct表示实例化之后就会执行
    @PostConstruct
    public void initRouteConfigListener() throws NacosException {
        //获取配置,服务启动的时候就要加载一次配置
        String config = nacosConfigManager.getConfigService().getConfigAndSignListener(dataId, group, timeout, new Listener() {
            @Override
            public Executor getExecutor() {
                return null;
            }

            @Override
            public void receiveConfigInfo(String config) {
                //配置变化时,重新加载配置
                updateConfig(config);
            }
        });
        //服务启动,加载配置
        updateConfig(config);
    }

    private void updateConfig(String config) {
        log.info("更新路由信息");
        //读取nacos中yaml配置
        Dict dict = YamlUtil.load(new BufferedReader(new InputStreamReader(new ByteArrayInputStream(config.getBytes()))));
        //转换成Route对象
        List<RouteDefinition> routeDefinitions = JSONObject.parseObject(JSONUtil.toJsonStr(dict)).getJSONObject("spring").getJSONObject("cloud").getJSONObject("gateway").getJSONArray("routes").toList(RouteDefinition.class);
        //清理旧的路由规则
        routeIds.forEach(routeId -> writer.delete(Mono.just(routeId)).subscribe());
        //清空缓存
        routeIds.clear();
        // 遍历添加路由
        routeDefinitions.forEach(definition -> {
            //提交路由信息
            writer.save(Mono.just(definition)).subscribe();
            //缓存下记录,这样便于清除路由配置
            routeIds.add(definition.getId());
        });
        this.publisher.publishEvent(new RefreshRoutesEvent(this));
    }

    @Override
    public void setApplicationEventPublisher(ApplicationEventPublisher applicationEventPublisher) {
        this.publisher = applicationEventPublisher;
    }
}

DynamicRouterLoader在服务启动的时候就会实例化并且注入spring中,所以我们给方法initRouteConfigListener添加注解@PostConstruct,这样实例化一完成就会执行该方法。方法里使用NacosConfigManager获取到nacos的数据并做一次更新路由,这次更新是用来启动服务时初始化路由配置,再添加一个监听器,每次nacos修改都会触发这个监听器再次更新路由配置。更新路由其实有两个思路,思路1是能准确判断每次修改、删除的路由信息,然后分别执行update和delete,这就需要我们去做新旧版本配置的区分工作了,工作量比较大,思路2就是每次都清空所有路由,然后把最新版配置信息加载进去。思路2更简单一点,不过我没有测试过路由信息特别大的情况下会不会导致延迟,通常我们的微服务不至于多到会产生延迟,但是这里依旧要警惕,工作中如果发现数量真的很大了,就要多做一些压力测试。

4、重启服务,做测试:http://127.0.0.1:8888/nacos-client-demo/api/login?userid=123&username=zhangsan

这时候是正常的:

5、修改一下nacos上路由信息,把nacos-client-demo的路由直接删掉,保存服务一直正常运行,观察是否生效。

如果debug能发现监听器被调用了,日志也会打印,再调用接口就已经404了。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值