SpringCloud-Gateway路由动态配置Nacos实现

1. 编写配置类

   @Component
   public class NacosConfig {
       public static String GROUP_ID ;
       public static String DATA_ID ;
       @Value("${nacos.gateway.route.config.group}")
       public void setGroupId(String groupId){
           GROUP_ID = groupId ;
       }
       @Value("${nacos.gateway.route.config.data-id}")
       public void setDataId(String dataId){
           DATA_ID = dataId ;
       }
   }
   

2. properties添加配置

   nacos.gateway.route.config.group=DEFAULT_GROUP
   nacos.gateway.route.config.data-id=hello-gateway-router.json
   

3. 自定义RouteDefinitionLocator

   @Slf4j
   @Component
   public class NcosRouteDefinitionLocator implements RouteDefinitionLocator, InitializingBean {
       private volatile List<RouteDefinition> routeDefinitions = new CopyOnWriteArrayList<>() ;
       @Autowired
       private NacosConfigManager nacosConfigManager ;
       @Autowired
       private ObjectMapper objectMapper ;
       @Autowired
       private ApplicationEventPublisher eventPublisher ;
       private ConfigService configService ;
       @Override
       public Flux<RouteDefinition> getRouteDefinitions() {
           return Flux.fromIterable(routeDefinitions);
       }
       @Override
       public void afterPropertiesSet() throws Exception {
           // 从nacos获取routDefinition
           configService = nacosConfigManager.getConfigService();
           // 读取配置并装载
           this.initNacosConfig();
           // 添加listener，当数据变化时接收通知
           this.initNacosListener();
       }
       private void initNacosConfig(){
           try {
               String content = configService.getConfig(
                       NacosConfig.DATA_ID,
                       NacosConfig.GROUP_ID,
                       3000L
               );
               CollectionType collectionType = objectMapper.getTypeFactory()
                       .constructCollectionType(ArrayList.class, RouteDefinition.class);
               routeDefinitions = objectMapper.readValue(content, collectionType);
           }catch (NacosException e){
               log.info("nacos config NacosException ", e);
           } catch (JsonProcessingException e) {
               log.info("nacos config JsonProcessingException ", e);
           }
       }
       private void initNacosListener() throws NacosException {
           configService.addListener(
                   NacosConfig.DATA_ID,
                   NacosConfig.GROUP_ID,
                   new DataChangeListener());
       }
       class DataChangeListener implements Listener{
   
           @Override
           public Executor getExecutor() {
               return null;
           }
           @Override
           public void receiveConfigInfo(String content) {
               try {
                   log.info("gateway config change : {}", content);
                   CollectionType collectionType = objectMapper.getTypeFactory()
                           .constructCollectionType(ArrayList.class, RouteDefinition.class);
                   routeDefinitions = objectMapper.readValue(content, collectionType);
                   // 当数据发生变化后发布刷新事件,通知CachingRouteLocator需要重新加载route定义
                   eventPublisher.publishEvent(new RefreshRoutesEvent(content));
               }catch (JsonProcessingException e) {
                   log.info("nacos config JsonProcessingException ", e);
               }
           }
       }
   }
   

4. 编写GatewayDynamicConfiguration配置类

   @Configuration
   public class GatewayDynamicConfiguration {
       @Bean
       @ConditionalOnProperty(value = "spring.cloud.gateway.dynamic.config.enable", matchIfMissing = true)
       public NacosRouteDefinitionLocator routeDefinitionLocator(){
           return new NacosRouteDefinitionLocator() ;
       }
   }