Elgato_dark/sol-cloud
1
0
Fork 0
mirror of https://github.com/BgaSol/sol-cloud synced 2026-05-23 00:58:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(gateway): Add dynamic route registration and improve route addition logging

Browse source
This commit is contained in:
sol 2025-07-01 11:01:46 +08:00
parent a107b6085d
commit d5be623aa9
1 changed files with 31 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

47
cloud/gateway-9527/src/main/java/com/bgasol/gateway/config/GatewayConfig.java
View file

@ -41,6 +41,32 @@ public class GatewayConfig {
routeIdList.add(GatewayConfigValues.SERVICE_NAME);
}
public Flux<RouteDefinition> getAllRoutes() {
return routeDefinitionLocator.getRouteDefinitions();
}
private boolean isRegistered = false;
@EventListener(InstanceRegisteredEvent.class)
public void onInstanceRegistered() {
isRegistered = true;
}
@Scheduled(cron = "0/5 * * * * ?")
public void customRouteLocator() {
if (!isRegistered) {
log.info("Gateway 尚未注册，跳过路由添加");
return;
}
for (String service : discoveryClient.getServices()) {
if (!routeIdList.contains(service)) {
routeIdList.add(service);
addRoute(service, "lb://" + service, "/" + service + "/**");
log.info("添加路由：{}", service);
}
}
}
public void addRoute(String id, String uri, String path) {
RouteDefinition definition = new RouteDefinition();
definition.setId(id);
@ -53,22 +79,11 @@ public class GatewayConfig {
predicateDefinition.setName("Path");
predicateDefinition.setArgs(Map.of("pattern", path));
definition.getPredicates().add(predicateDefinition);
routeDefinitionWriter.save(Mono.just(definition)).subscribe();
routeDefinitionWriter.save(Mono.just(definition))
.doOnSuccess(unused -> log.info("路由 [{}] 添加成功，path: [{}]", id, path))
.doOnError(error -> log.error("路由 [{}] 添加失败：{}", id, error.getMessage()))
.block();
}
public Flux<RouteDefinition> getAllRoutes() {
return routeDefinitionLocator.getRouteDefinitions();
}
@EventListener(InstanceRegisteredEvent.class)
@Scheduled(cron = "0/5 * * * * ?")
public void customRouteLocator() {
for (String service : discoveryClient.getServices()) {
if (!routeIdList.contains(service)) {
routeIdList.add(service);
addRoute(service, "lb://" + service, "/" + service + "/**");
log.info("添加路由：{}", service);
}
}
}
}