NacosDynamicRouteService.java 4.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125
  1. /*
  2. package org.gateway.route;
  3. import com.alibaba.fastjson.JSONObject;
  4. import com.alibaba.nacos.api.NacosFactory;
  5. import com.alibaba.nacos.api.PropertyKeyConst;
  6. import com.alibaba.nacos.api.config.ConfigService;
  7. import com.alibaba.nacos.api.config.listener.Listener;
  8. import com.alibaba.nacos.api.exception.NacosException;
  9. import org.slf4j.Logger;
  10. import org.slf4j.LoggerFactory;
  11. import org.springframework.beans.factory.annotation.Autowired;
  12. import org.springframework.beans.factory.annotation.Value;
  13. import org.springframework.cloud.gateway.event.RefreshRoutesEvent;
  14. import org.springframework.cloud.gateway.route.RouteDefinition;
  15. import org.springframework.cloud.gateway.route.RouteDefinitionWriter;
  16. import org.springframework.context.ApplicationEventPublisher;
  17. import org.springframework.context.ApplicationEventPublisherAware;
  18. import org.springframework.stereotype.Component;
  19. import reactor.core.publisher.Mono;
  20. import javax.annotation.PostConstruct;
  21. import java.util.ArrayList;
  22. import java.util.List;
  23. import java.util.Properties;
  24. import java.util.concurrent.Executor;
  25. @Component
  26. public class NacosDynamicRouteService implements ApplicationEventPublisherAware {
  27. private final Logger logger = LoggerFactory.getLogger(getClass());
  28. private String dataId = "alibaba-gateway-router";
  29. private String group = "DEFAULT_GROUP";
  30. @Value("${spring.cloud.nacos.config.server-addr}")
  31. private String serverAddr;
  32. @Value("${spring.cloud.nacos.config.namespace}")
  33. private String namespace;
  34. @Autowired
  35. private RouteDefinitionWriter routeDefinitionWriter;
  36. private ApplicationEventPublisher applicationEventPublisher;
  37. private static final List<String> ROUTE_LIST = new ArrayList<>();
  38. @PostConstruct
  39. public void dynamicRouteByNacosListener() {
  40. try {
  41. logger.info("serverAddr:{}",serverAddr);
  42. logger.info("namespace:{}",namespace);
  43. logger.info("dataId:{}",dataId);
  44. logger.info("group:{}",group);
  45. Properties properties = new Properties();
  46. properties.put(PropertyKeyConst.SERVER_ADDR, serverAddr);
  47. properties.put(PropertyKeyConst.NAMESPACE, namespace);
  48. ConfigService configService = NacosFactory.createConfigService(properties);
  49. configService.getConfig(dataId, group, 5000);
  50. // 程序首次启动, 并加载初始化路由配置
  51. String initConfigInfo = configService.getConfig(dataId, group, 5000);
  52. addAndPublishBatchRoute(initConfigInfo);
  53. publish();
  54. configService.addListener(dataId, group, new Listener() {
  55. @Override
  56. public void receiveConfigInfo(String configInfo) {
  57. clearRoute();
  58. try {
  59. List<RouteDefinition> gatewayRouteDefinitions = JSONObject.parseArray(configInfo, RouteDefinition.class);
  60. for (RouteDefinition routeDefinition : gatewayRouteDefinitions) {
  61. logger.info("routeDefinition:{}",routeDefinition);
  62. addRoute(routeDefinition);
  63. }
  64. publish();
  65. } catch (Exception e) {
  66. e.printStackTrace();
  67. }
  68. }
  69. @Override
  70. public Executor getExecutor() {
  71. return null;
  72. }
  73. });
  74. } catch (NacosException e) {
  75. logger.error("error:{}",e);
  76. }
  77. }
  78. private void clearRoute() {
  79. for(String id : ROUTE_LIST) {
  80. this.routeDefinitionWriter.delete(Mono.just(id)).subscribe();
  81. }
  82. ROUTE_LIST.clear();
  83. }
  84. private void addAndPublishBatchRoute(String initConfigInfo){
  85. List<RouteDefinition> gatewayRouteDefinitions = JSONObject.parseArray(initConfigInfo, RouteDefinition.class);
  86. for (RouteDefinition routeDefinition : gatewayRouteDefinitions) {
  87. logger.info("init routeDefinition:{}",routeDefinition);
  88. addRoute(routeDefinition);
  89. }
  90. }
  91. private void addRoute(RouteDefinition definition) {
  92. try {
  93. routeDefinitionWriter.save(Mono.just(definition)).subscribe();
  94. ROUTE_LIST.add(definition.getId());
  95. } catch (Exception e) {
  96. e.printStackTrace();
  97. }
  98. }
  99. private void publish() {
  100. this.applicationEventPublisher.publishEvent(new RefreshRoutesEvent(this.routeDefinitionWriter));
  101. }
  102. @Override
  103. public void setApplicationEventPublisher(ApplicationEventPublisher applicationEventPublisher) {
  104. this.applicationEventPublisher = applicationEventPublisher;
  105. }
  106. }
  107. */