com.jeesuite.gateway.CurrentSystemHolder Maven / Gradle / Ivy
The newest version!
package com.jeesuite.gateway;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Properties;
import java.util.Set;
import java.util.concurrent.atomic.AtomicReference;
import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.cloud.gateway.config.GatewayProperties;
import org.springframework.cloud.gateway.filter.FilterDefinition;
import org.springframework.cloud.gateway.handler.predicate.PredicateDefinition;
import org.springframework.cloud.gateway.route.RouteDefinition;
import com.jeesuite.common.GlobalRuntimeContext;
import com.jeesuite.common.JeesuiteBaseException;
import com.jeesuite.common.http.HttpRequestEntity;
import com.jeesuite.common.model.ApiInfo;
import com.jeesuite.common.util.ResourceUtils;
import com.jeesuite.gateway.api.SystemMgtApi;
import com.jeesuite.gateway.model.BizSystemModule;
import com.jeesuite.gateway.model.BizSystemPortal;
import com.jeesuite.spring.InstanceFactory;
import com.jeesuite.springweb.exporter.AppMetadataHolder;
import com.jeesuite.springweb.model.AppMetadata;
/**
*
* Class Name : CurrentSystemHolder
*
* @author jiangwei
* @version 1.0.0
* @date 2019年12月3日
*/
public class CurrentSystemHolder {
private static Logger log = LoggerFactory.getLogger("com.jeesuite.gateway");
private static AtomicReference> routeModuleMappings = new AtomicReference<>();
private static List localModules;
private static Set routeNames;
private static Map> moduleApiInfos = new HashMap<>();
private static int fetchApiMetaRound = 0;
public static BizSystemModule getModule(String route) {
BizSystemModule module = routeModuleMappings.get().get(route);
if (module == null) {
module = routeModuleMappings.get().get(GlobalRuntimeContext.APPID);
}
return module;
}
public static Map getRouteModuleMappings() {
if (routeModuleMappings.get() == null) {
load();
}
return Collections.unmodifiableMap(routeModuleMappings.get());
}
public static Set getRouteNames() {
if (routeNames == null) {
if (routeModuleMappings.get() == null) {
load();
}
synchronized(CurrentSystemHolder.class) {
if(routeNames != null)return routeNames;
routeNames = new HashSet<>(routeModuleMappings.get().keySet());
}
}
return routeNames;
}
public static Collection getModules() {
if (routeModuleMappings.get() == null)
return new ArrayList<>(0);
return Collections.unmodifiableCollection(routeModuleMappings.get().values());
}
public static BizSystemPortal getSystemPortal(String host) {
return null;
}
public static synchronized void load() {
//
loadLocalRouteModules();
List modules;
try {
modules = InstanceFactory.getInstance(SystemMgtApi.class).getSystemModules();
if (!localModules.isEmpty()) {
modules.addAll(localModules);
}
} catch (Exception e) {
modules = localModules;
}
//网关本身
if(!modules.stream().anyMatch(o -> GlobalRuntimeContext.APPID.equalsIgnoreCase(o.getServiceId()))) {
BizSystemModule module = new BizSystemModule();
module.setServiceId(GlobalRuntimeContext.APPID);
module.setRouteName(GlobalRuntimeContext.APPID);
module.setStripPrefix(0);
modules.add(module);
localModules.add(module);
}
Map _modules = new HashMap<>(modules.size());
for (BizSystemModule module : modules) {
boolean isGateway = GlobalRuntimeContext.APPID.equalsIgnoreCase(module.getServiceId());
if (!isGateway && StringUtils.isBlank(module.getRouteName())) {
continue;
}
module.finalCorrect();
//
if(moduleApiInfos.containsKey(module.getServiceId())) {
module.setApiInfos(moduleApiInfos.get(module.getServiceId()));
}
_modules.put(module.getRouteName(), module);
}
routeModuleMappings.set(_modules);
//
//查询api信息
if(fetchApiMetaRound == 0){
new Thread(() -> {
while(_modules.size() > moduleApiInfos.size() && fetchApiMetaRound < 360) {
for (BizSystemModule module : _modules.values()) {
if(moduleApiInfos.containsKey(module.getServiceId()))continue;
initModuleApiInfos(module);
}
fetchApiMetaRound++;
try {Thread.sleep(10000);} catch (Exception e) {}
}
}).start();
}
}
private static void loadLocalRouteModules() {
if(localModules != null)return;
localModules = new ArrayList<>();
List defaultRouteDefs = InstanceFactory.getInstance(GatewayProperties.class).getRoutes();
Properties properties = ResourceUtils.getAllProperties("spring.cloud.gateway.routes");
Set> entrySet = properties.entrySet();
BizSystemModule module;
String prefix;
for (Entry entry : entrySet) {
if(entry.getKey().toString().endsWith(".id")) {
prefix = entry.getKey().toString().replace(".id", "");
module = new BizSystemModule();
module.setDefaultRoute(true);
module.setServiceId(entry.getValue().toString());
module.setProxyUri(properties.getProperty(prefix + ".uri"));
module.setAnonymousUris(properties.getProperty(prefix + ".anonymousUris"));
//
updateModuleRouteInfos(module, defaultRouteDefs);
localModules.add(module);
}
}
}
/**
* @param module
* @param definition
*/
private static void updateModuleRouteInfos(BizSystemModule module, List defaultRouteDefs ) {
RouteDefinition routeDef = defaultRouteDefs.stream().filter(def -> StringUtils.equalsIgnoreCase(module.getServiceId(), def.getId())).findFirst().orElse(null);
if(routeDef == null)return;
FilterDefinition stripPrefixDef = routeDef.getFilters().stream().filter(p -> "StripPrefix".equals(p.getName())).findFirst().orElse(null);
int stripPrefix = 0;
if(stripPrefixDef != null) {
stripPrefix = Integer.parseInt(stripPrefixDef.getArgs().get("_genkey_0"));
}
module.setStripPrefix(stripPrefix);
PredicateDefinition pathDef = routeDef.getPredicates().stream().filter(p -> "Path".equals(p.getName())).findFirst().orElse(null);
if(pathDef != null) {
String pathPattern = pathDef.getArgs().get("_genkey_0");
if(!pathPattern.startsWith(GatewayConstants.PATH_PREFIX)) {
throw new JeesuiteBaseException("route path must startWith:" + GatewayConstants.PATH_PREFIX);
}
String[] parts = StringUtils.split(pathPattern, "/");
module.setRouteName(parts[1]);
}
}
private static void initModuleApiInfos(BizSystemModule module) {
try {
String url;
AppMetadata appMetadata;
if(GlobalRuntimeContext.APPID.equals(module.getRouteName())) {
appMetadata = AppMetadataHolder.getMetadata();
}else {
url = module.getMetadataUri();
appMetadata = HttpRequestEntity.get(url).backendInternalCall().execute().toObject(AppMetadata.class);
}
Map apiInfos = new HashMap<>(appMetadata.getApis().size());
for (ApiInfo api : appMetadata.getApis()) {
api.setUrl(BizSystemModule.resolveApiFinalUri(module, api.getUrl()));
apiInfos.put(api.getUrl(), api);
}
module.setApiInfos(apiInfos);
moduleApiInfos.put(module.getServiceId(), apiInfos);
log.info(">>initModuleApiInfos success -> serviceId:{},nums:{}",module.getServiceId(),apiInfos.size());
} catch (Exception e) {
boolean ignore = e instanceof ClassCastException;
if(!ignore && e instanceof JeesuiteBaseException) {
JeesuiteBaseException ex = (JeesuiteBaseException) e;
ignore = ex.getCode() == 404 || ex.getCode() == 401 || ex.getCode() == 403;
}
if(ignore) {
module.setApiInfos(new HashMap<>(0));
moduleApiInfos.put(module.getServiceId(), module.getApiInfos());
}else if(fetchApiMetaRound <= 1) {
log.error(">>initModuleApiInfos error -> serviceId:["+module.getServiceId()+"]",e);
}
}
}
}