springcloud gateway动态路由

程序青

已于 2024-01-11 18:39:37 修改

阅读量802

点赞数 7

文章标签： spring cloud gateway spring boot

于 2024-01-11 17:05:36 首次发布

本文链接：https://blog.csdn.net/m0_56242825/article/details/135534175

版权

本文介绍了如何在SpringCloudGateway中实现实时从数据库动态加载路由规则，通过自定义DbRouteDefinitionRepository替换内存存储的InMemoryRouteDefinitionRepository，确保路由规则持久化。同时提及了从RouteLocator获取请求路由的过程。

摘要由CSDN通过智能技术生成

动态每秒调用数据库，加载路由信息到路由定义（RouteDefinition）中

主要是继承 RouteDefinitionRepository接口：

如果项目没有自定义的RouteDefinitionRepository 则会加载InMemoryRouteDefinitionRepository
默认将加载的规则存放在内存，加载application.properties文件中配置的路由规则，InMemory加载内存中配置的路由规则
Application启动的时候内存中是没有路由规则的且Application关闭的时候内存中保存的路由规则也会丢失
我们可以通过自定义RouteDefinitionRepository类替换掉InMemoryRouteDefinitionRepository
实现将路由规则保存到数据库中

@Component
@Slf4j
public class DbRouteDefinitionRepository implements RouteDefinitionRepository {

    @Resource
    private SysRouteConfMapper sys;


    @Override
    public Flux<RouteDefinition> getRouteDefinitions() {
        List<RouteDefinition> routeDefinitions = new ArrayList<>();
//获取数据库路由信息
        List<SysRouteConf> gatewayRouteList = sysRouteConfMapper.selectList();
//路由信息转为路由定义
        gatewayRouteList.forEach(gatewayRoute -> {
            RouteDefinition routeDefinition = JSONObject.parseObject(gatewayRoute.toString(), RouteDefinition.class);
            routeDefinitions.add(routeDefinition);
        });
//存入内存中
        return Flux.fromIterable(routeDefinitions);
    }

    @Override
    public Mono<Void> save(Mono<RouteDefinition> route) {
        return null;
    }

    @Override
    public Mono<Void> delete(Mono<String> routeId) {
        return null;
    }
}

获取请求的路由信息

主要从RouteLocator的三个实现类中获取

//获取路径

    protected String getUrl() {
        AtomicReference<String> url = new AtomicReference<>("");
        RouteLocator cachedCompositeRouteLocator = SpringUtils.getBean("cachedCompositeRouteLocator");
        Flux<Route> identification = cachedCompositeRouteLocator.getRoutes().filter(route -> {
            return route.getId().equals("identification");
        });
        identification.subscribe(result -> {
            URI uri = result.getUri();
            if (uri != null && ("lb".equals(uri.getScheme()))) {
                ServiceInstance router = loadBalancer.choose("identification");
                url.set(router.getHost() + ":" + router.getPort());
            }
            if (uri != null && ("http".equals(uri.getScheme()) || "https".equals(uri.getScheme()))) {
                url.set(uri.getHost() + ":" + uri.getPort());
            }
        });
        return url.get();
    }