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

编写配置类

java 复制代码

@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 ;
    }
}

properties添加配置

properties 复制代码

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

自定义RouteDefinitionLocator

java 复制代码

@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);
            }
        }
    }
}

编写GatewayDynamicConfiguration配置类

java 复制代码

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