spring cloud zuul 實現動態路由_每天變化一次

前期工作

服務提供者的eureka添加配置

# 配置 server.servlet.context-path                必須配置,不可單獨爲反斜槓 /
server.servlet.context-path=/user/
# 在eurek的metadata-map 中存入 server.servlet.context-path
eureka.instance.metadata-map.context-path=${server.servlet.context-path} 

改造zuul 自定義網關路由規則

package cn.zhangfusheng.zuul.server.config;

import com.netflix.appinfo.InstanceInfo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.web.ServerProperties;
import org.springframework.cloud.client.discovery.DiscoveryClient;
import org.springframework.cloud.netflix.eureka.EurekaServiceInstance;
import org.springframework.cloud.netflix.zuul.filters.SimpleRouteLocator;
import org.springframework.cloud.netflix.zuul.filters.ZuulProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import java.time.LocalDate;
import java.time.format.DateTimeFormatter;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @author fusheng.zhang
 * @Description 網關動態路由
 * @create 2020-04-17 20:04:00
 */
@Slf4j
@Configuration
public class ZuulRoutesConfig {

    /**
     * 注入 discoveryClient 用於獲取註冊中心的服務提供者列表
     */
    @Autowired
    private DiscoveryClient discoveryClient;
    @Autowired
    ZuulProperties zuulProperties;
    @Autowired
    ServerProperties serverProperties;

    @Bean
    public DynamicRoutes dynamicRoutes() {
        return new DynamicRoutes(this.serverProperties.getServlet().getContextPath(), this.zuulProperties);
    }

    /**
     * 動態路由
     * 路由規則: /${eureka.instance.metadata-map.context-path}/2020-04-20/**
     * 目的: 路由一天一變化__未測試
     */
    class DynamicRoutes extends SimpleRouteLocator {

        private final Map<String, Map<String, ZuulProperties.ZuulRoute>> dayZuulRoureMap = new ConcurrentHashMap<>(1);

        public DynamicRoutes(String servletPath, ZuulProperties properties) {
            super(servletPath, properties);
        }

        /**
         * 重寫 SimpleRouteLocator.locateRoutes()方法
         * 自定義路由生成方式
         * @return
         */
        @Override
        protected Map<String, ZuulProperties.ZuulRoute> locateRoutes() {
            Map<String, ZuulProperties.ZuulRoute> routes = new LinkedHashMap<>();
            String localDate = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd"));
            // 獲取服務名稱列表[serviceId]
            discoveryClient.getServices().forEach(serviceId -> {
                // 獲取服務名稱對應的實例
                discoveryClient.getInstances(serviceId).forEach(instance -> {
                    InstanceInfo instanceInfo = ((EurekaServiceInstance) instance).getInstanceInfo();
                    // 從 eureka.instance.metadata-map中獲取context-path,既服務提供者的server.servlet.context-path
                    // 如果獲取不到,則採用服務名稱'-'切割的方式作爲路由
                    String key = instanceInfo.getMetadata().getOrDefault(
                            "context-path", String.format("/%s/", instanceInfo.getSecureVipAddress().split("-")[0]));
                    // 生成 /${eureka.instance.metadata-map.context-path}/2020-04-20/** 格式的key
                    key = String.format("%s%s/**", key, localDate);
                    ZuulProperties.ZuulRoute zuulRoute = new ZuulProperties.ZuulRoute();
                    zuulRoute.setLocation(instanceInfo.getSecureVipAddress());
                    zuulRoute.setId(instanceInfo.getSecureVipAddress());
                    zuulRoute.setPath(key);
                    zuulRoute.setServiceId(instanceInfo.getSecureVipAddress());
                    routes.put(key, zuulRoute);
                });
            });
            return routes;
        }

        @Override
        protected Map<String, ZuulProperties.ZuulRoute> getRoutesMap() {
            String routesDate = LocalDate.now().format(DateTimeFormatter.ofPattern("yyyy/MM/dd"));
            if (!dayZuulRoureMap.containsKey(routesDate)) {
                dayZuulRoureMap.clear();
                Map<String, ZuulProperties.ZuulRoute> zuulRouteMap = locateRoutes();
                dayZuulRoureMap.put(routesDate, zuulRouteMap);
            }
            return dayZuulRoureMap.get(routesDate);
        }
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章