com.quancheng.saluki.gateway.zuul.extend.StoreProxyRouteLocator.java Source code

Introduction

Here is the source code for com.quancheng.saluki.gateway.zuul.extend.StoreProxyRouteLocator.java

Source

/**
 * 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
 *
 * 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.
 */
package com.quancheng.saluki.gateway.zuul.extend;

import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.LinkedHashMap;
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;
import org.springframework.cloud.netflix.zuul.filters.discovery.DiscoveryClientRouteLocator;

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
 */
public class StoreProxyRouteLocator extends DiscoveryClientRouteLocator {

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

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

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

    /**
     * {@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)));

            }

        });
    }
}