Skip to content

Commit

Permalink
up
Browse files Browse the repository at this point in the history
  • Loading branch information
shiming.liu committed Aug 14, 2017
1 parent 4c507ee commit 70116be
Showing 1 changed file with 70 additions and 83 deletions.
153 changes: 70 additions & 83 deletions ...ateway/src/main/java/com/quancheng/saluki/gateway/zuul/extend/StoreProxyRouteLocator.java
View file
Original file line number Diff line number Diff line change
@@ -1,17 +1,15 @@
/**
* Copyright (c) 2015 the original author or authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except
* in compliance with the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
* Unless required by applicable law or agreed to in writing, software distributed under the License
* is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
* or implied. See the License for the specific language governing permissions and limitations under
* the License.
*/
package com.quancheng.saluki.gateway.zuul.extend;

Expand All @@ -22,12 +20,8 @@
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.cloud.client.discovery.DiscoveryClient;
import org.springframework.cloud.netflix.zuul.filters.ZuulProperties;
import org.springframework.cloud.netflix.zuul.filters.ZuulProperties.ZuulRoute;
Expand All @@ -36,94 +30,87 @@
import com.google.common.base.Function;
import com.google.common.collect.Lists;
import com.google.common.collect.Sets;
import com.quancheng.saluki.core.common.NamedThreadFactory;
import com.quancheng.saluki.gateway.zuul.dto.ZuulRouteDto;
import com.quancheng.saluki.gateway.zuul.service.ZuulRouteService;

/**
* A simple {@link org.springframework.cloud.netflix.zuul.filters.RouteLocator} that is being populated from configured
* {@link ZuulRouteStore}.
*
* @author Jakub Narloch
*
* @author liushiming
* @version StoreProxyRouteLocator.java, v 0.0.1 2017年8月14日 下午4:52:01 liushiming
* @since JDK 1.8
*/
public class StoreProxyRouteLocator extends DiscoveryClientRouteLocator {

private final Logger log = LoggerFactory.getLogger(StoreProxyRouteLocator.class);

private final ZuulRouteService store;
private final ZuulRouteService store;

private volatile Boolean hasLoaded = false;

/**
* Creates new instance of {@link StoreProxyRouteLocator}
*
* @param servletPath the application servlet path
* @param discovery the discovery service
* @param properties the zuul properties
* @param store the route store
*/
public StoreProxyRouteLocator(String servletPath, DiscoveryClient discovery, ZuulProperties properties,
ZuulRouteService store){
super(servletPath, discovery, properties);
this.store = store;
}
/**
* Creates new instance of {@link StoreProxyRouteLocator}
*
* @param servletPath the application servlet path
* @param discovery the discovery service
* @param properties the zuul properties
* @param store the route store
*/
public StoreProxyRouteLocator(String servletPath, DiscoveryClient discovery,
ZuulProperties properties, ZuulRouteService store) {
super(servletPath, discovery, properties);
this.store = store;
}

@Override
public Collection<String> getIgnoredPaths() {
List<String> ignoredPath = Lists.newArrayList();
ignoredPath.add("/oauth/**");
return ignoredPath;
}
@Override
public Collection<String> getIgnoredPaths() {
List<String> ignoredPath = Lists.newArrayList();
ignoredPath.add("/oauth/**");
return ignoredPath;
}

@Override
protected LinkedHashMap<String, ZuulRoute> locateRoutes() {
LinkedHashMap<String, ZuulRoute> routesMap = new LinkedHashMap<String, ZuulRoute>();
routesMap.putAll(super.locateRoutes());
for (ZuulProperties.ZuulRoute route : findAll()) {
routesMap.put(route.getPath(), route);
}
return routesMap;
@Override
protected LinkedHashMap<String, ZuulRoute> locateRoutes() {
LinkedHashMap<String, ZuulRoute> routesMap = new LinkedHashMap<String, ZuulRoute>();
routesMap.putAll(super.locateRoutes());
for (ZuulProperties.ZuulRoute route : findAll()) {
routesMap.put(route.getPath(), route);
}
return routesMap;
}

/**
* {@inheritDoc}
*/
@Override
protected void addConfiguredRoutes(Map<String, ZuulProperties.ZuulRoute> routes) {
super.addConfiguredRoutes(routes);
}
/**
* {@inheritDoc}
*/
@Override
protected void addConfiguredRoutes(Map<String, ZuulProperties.ZuulRoute> routes) {
super.addConfiguredRoutes(routes);
}

private List<ZuulProperties.ZuulRoute> findAll() {
Set<ZuulRouteDto> routers = Sets.newHashSet();
if (!hasLoaded) {
routers.addAll(store.loadAllRoute());
hasLoaded = true;
} else {
routers.addAll(store.loadTop10Route());
}
RouterLocalCache.getInstance().putAllRouters(routers);
return Lists.transform(Lists.newArrayList(routers), new Function<ZuulRouteDto, ZuulProperties.ZuulRoute>() {

@Override
public ZuulRoute apply(ZuulRouteDto input) {
String id = input.getRouteId();
String path = input.getRoutePath();
String service_id = input.getServiceId();
String url = input.getRouteUrl();
boolean strip_prefix = input.getStripPrefix() != null ? input.getStripPrefix() : true;
Boolean retryable = input.getRetryAble();
String sensitiveHeader = input.getSensitiveHeaders();
String[] sensitiveHeaders = null;
if (sensitiveHeader != null) {
sensitiveHeaders = StringUtils.split(sensitiveHeader, ",");
} else {
sensitiveHeaders = new String[] {};
}
return new ZuulRoute(id, path, service_id, url, strip_prefix, retryable,
new HashSet<String>(Arrays.asList(sensitiveHeaders)));
private List<ZuulProperties.ZuulRoute> findAll() {
Set<ZuulRouteDto> routers = Sets.newHashSet();
routers.addAll(store.loadAllRoute());
RouterLocalCache.getInstance().putAllRouters(routers);
return Lists.transform(Lists.newArrayList(routers),
new Function<ZuulRouteDto, ZuulProperties.ZuulRoute>() {

@Override
public ZuulRoute apply(ZuulRouteDto input) {
String id = input.getRouteId();
String path = input.getRoutePath();
String service_id = input.getServiceId();
String url = input.getRouteUrl();
boolean strip_prefix = input.getStripPrefix() != null ? input.getStripPrefix() : true;
Boolean retryable = input.getRetryAble();
String sensitiveHeader = input.getSensitiveHeaders();
String[] sensitiveHeaders = null;
if (sensitiveHeader != null) {
sensitiveHeaders = StringUtils.split(sensitiveHeader, ",");
} else {
sensitiveHeaders = new String[] {};
}
return new ZuulRoute(id, path, service_id, url, strip_prefix, retryable,
new HashSet<String>(Arrays.asList(sensitiveHeaders)));

}

});
}
}
}

0 comments on commit 70116be

Please sign in to comment.