/**
|
* BladeX Commercial License Agreement
|
* Copyright (c) 2018-2099, https://bladex.cn. All rights reserved.
|
* <p>
|
* Use of this software is governed by the Commercial License Agreement
|
* obtained after purchasing a license from BladeX.
|
* <p>
|
* 1. This software is for development use only under a valid license
|
* from BladeX.
|
* <p>
|
* 2. Redistribution of this software's source code to any third party
|
* without a commercial license is strictly prohibited.
|
* <p>
|
* 3. Licensees may copyright their own code but cannot use segments
|
* from this software for such purposes. Copyright of this software
|
* remains with BladeX.
|
* <p>
|
* Using this software signifies agreement to this License, and the software
|
* must not be used for illegal purposes.
|
* <p>
|
* THIS SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY. The author is
|
* not liable for any claims arising from secondary or illegal development.
|
* <p>
|
* Author: Chill Zhuang (bladejava@qq.com)
|
*/
|
package org.springblade.gateway.dynamic;
|
|
import com.alibaba.cloud.nacos.NacosConfigProperties;
|
import com.alibaba.cloud.nacos.NacosDiscoveryProperties;
|
import com.alibaba.fastjson.JSON;
|
import com.alibaba.nacos.api.NacosFactory;
|
import com.alibaba.nacos.api.PropertyKeyConst;
|
import com.alibaba.nacos.api.config.ConfigService;
|
import com.alibaba.nacos.api.config.listener.Listener;
|
import com.alibaba.nacos.api.exception.NacosException;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springblade.core.launch.constant.NacosConstant;
|
import org.springblade.core.launch.props.BladeProperties;
|
import org.springframework.cloud.context.config.annotation.RefreshScope;
|
import org.springframework.cloud.gateway.route.RouteDefinition;
|
import org.springframework.core.annotation.Order;
|
import org.springframework.stereotype.Component;
|
|
import java.util.List;
|
import java.util.Properties;
|
import java.util.concurrent.Executor;
|
|
/**
|
* 动态路由监听器
|
*
|
* @author Chill
|
*/
|
@Order
|
@Slf4j
|
@Component
|
@RefreshScope
|
public class DynamicRouteServiceListener {
|
|
private final DynamicRouteService dynamicRouteService;
|
private final NacosDiscoveryProperties nacosDiscoveryProperties;
|
private final NacosConfigProperties nacosConfigProperties;
|
private final BladeProperties bladeProperties;
|
|
public DynamicRouteServiceListener(DynamicRouteService dynamicRouteService, NacosDiscoveryProperties nacosDiscoveryProperties, NacosConfigProperties nacosConfigProperties, BladeProperties bladeProperties) {
|
this.dynamicRouteService = dynamicRouteService;
|
this.nacosDiscoveryProperties = nacosDiscoveryProperties;
|
this.nacosConfigProperties = nacosConfigProperties;
|
this.bladeProperties = bladeProperties;
|
dynamicRouteServiceListener();
|
}
|
|
/**
|
* 监听Nacos下发的动态路由配置
|
*/
|
private void dynamicRouteServiceListener() {
|
try {
|
String dataId = NacosConstant.dataId(bladeProperties.getName(), bladeProperties.getEnv(), NacosConstant.NACOS_CONFIG_JSON_FORMAT);
|
String group = nacosConfigProperties.getGroup();
|
Properties properties = new Properties();
|
properties.setProperty(PropertyKeyConst.SERVER_ADDR, nacosDiscoveryProperties.getServerAddr());
|
properties.setProperty(PropertyKeyConst.NAMESPACE, nacosDiscoveryProperties.getNamespace());
|
properties.setProperty(PropertyKeyConst.USERNAME, nacosDiscoveryProperties.getUsername());
|
properties.setProperty(PropertyKeyConst.PASSWORD, nacosDiscoveryProperties.getPassword());
|
ConfigService configService = NacosFactory.createConfigService(properties);
|
configService.addListener(dataId, group, new Listener() {
|
@Override
|
public void receiveConfigInfo(String configInfo) {
|
List<RouteDefinition> routeDefinitions = JSON.parseArray(configInfo, RouteDefinition.class);
|
dynamicRouteService.updateList(routeDefinitions);
|
}
|
|
@Override
|
public Executor getExecutor() {
|
return null;
|
}
|
});
|
String configInfo = configService.getConfig(dataId, group, 5000);
|
if (configInfo != null) {
|
List<RouteDefinition> routeDefinitions = JSON.parseArray(configInfo, RouteDefinition.class);
|
dynamicRouteService.updateList(routeDefinitions);
|
}
|
} catch (NacosException ignored) {
|
ignored.printStackTrace();
|
}
|
}
|
|
}
|
