<dependency>
   <groupId>org.springframework.cloud</groupId>
   <artifactId>spring-cloud-starter-gateway</artifactId>
</dependency>

<!-- 用于服務(wù)發(fā)現(xiàn)（如果灰度路由到注冊(cè)中心的服務(wù)） -->
<dependency>
   <groupId>com.alibaba.cloud</groupId>
   <artifactId>spring-cloud-starter-alibaba-nacos-discovery</artifactId>
</dependency>

@Configuration
publicclass TrafficDyeFilterConfig {

    // 定義全局過(guò)濾器，Order設(shè)為-1（確保比其他過(guò)濾器先執(zhí)行，早染色早用）
    @Bean
    @Order(-1)
    public GlobalFilter trafficDyeFilter() {
        return (exchange, chain) -> {
            // 1. 獲取請(qǐng)求中的用戶標(biāo)識(shí)（參數(shù)/Cookie）
            String userType = getUserTypeFromRequest(exchange);

            // 2. 根據(jù)用戶類型設(shè)置染色標(biāo)簽
            String trafficTag = getTrafficTagByUserType(userType);

            // 3. 將染色標(biāo)簽注入請(qǐng)求頭（傳遞給下游服務(wù)）
            exchange.getRequest().mutate()
                    .header("X-Traffic-Tag", trafficTag)
                    .build();
            
            // 4. 繼續(xù)執(zhí)行后續(xù)過(guò)濾器鏈
            return chain.filter(exchange);
        };
    }

    // 從請(qǐng)求參數(shù)或Cookie中獲取用戶類型

    private String getUserTypeFromRequest(ServerWebExchange exchange) {
        // 先查請(qǐng)求參數(shù)：比如 http://xxx?userType=vip
        List<String> userTypeParams = exchange.getRequest().getQueryParams().get("userType");
        if (userTypeParams != null && !userTypeParams.isEmpty()) {
            return userTypeParams.get(0);
        }
        // 默認(rèn)返回normal
        return"normal";
    }

    // 根據(jù)用戶類型映射染色標(biāo)簽
    private String getTrafficTagByUserType(String userType) {
        switch (userType) {
            case"vip":
                return"vip";
            case"test":
                return"test";
            default:
                return"normal";
        }
    }
}

// 自定義斷言工廠，命名格式：XXXRoutePredicateFactory（固定后綴）
@Configuration
publicclass GrayRoutePredicateFactory extends AbstractRoutePredicateFactory<GrayRoutePredicateFactory.Config> {

    // 染色標(biāo)簽的請(qǐng)求頭名（和第一步的X-Traffic-Tag對(duì)應(yīng)）
    privatestaticfinal String TRAFFIC_TAG_HEADER = "X-Traffic-Tag";

    // 構(gòu)造函數(shù)，指定配置類
    public GrayRoutePredicateFactory() {
        super(Config.class);
    }

    // 定義配置類：存儲(chǔ)斷言需要的參數(shù)（比如“需要匹配的染色標(biāo)簽”）
    @Validated
    publicstaticclass Config {
        // 允許的染色標(biāo)簽（比如["vip", "test"]）
        @NotEmpty
        private List<String> allowTags;

        public List<String> getAllowTags() {
            return allowTags;
        }

        public void setAllowTags(List<String> allowTags) {
            this.allowTags = allowTags;
        }
    }

    // 讀取配置參數(shù)的順序（和application.yml中配置的順序?qū)?yīng)）
    @Override
    public List<String> shortcutFieldOrder() {
        return Collections.singletonList("allowTags");
    }

    // 核心邏輯：判斷請(qǐng)求的染色標(biāo)簽是否在允許的列表中
    @Override
    public GatewayPredicate apply(Config config) {
        returnnew GatewayPredicate() {
            @Override
            public boolean test(ServerWebExchange exchange) {
                // 1. 獲取請(qǐng)求頭中的染色標(biāo)簽
                List<String> trafficTags = exchange.getRequest().getHeaders().get(TRAFFIC_TAG_HEADER);
                if (trafficTags == null || trafficTags.isEmpty()) {
                    returnfalse; // 沒(méi)有標(biāo)簽，不匹配灰度路由
                }

                String trafficTag = trafficTags.get(0);
                // 2. 判斷標(biāo)簽是否在允許的列表中（比如["vip", "test"]）
                return config.getAllowTags().contains(trafficTag);
            }

            // 用于日志打印，方便調(diào)試
            @Override
            public String toString() {
                return"GrayRoutePredicate{allowTags=" + config.getAllowTags() + "}";
            }
        };
    }
}

spring:
  cloud:
    gateway:
      routes:
        # 路由1：灰度流量（vip/test標(biāo)簽）→ 新版本服務(wù)（order-service-v2）
        -id:gray_route_v2
          uri:lb://order-service-v2# 服務(wù)注冊(cè)中心的新版本服務(wù)名
          predicates:
            # 自定義灰度斷言：允許的標(biāo)簽是["vip", "test"]
            -name:GrayRoute
              args:
                allowTags[0]:vip
                allowTags[1]:test
            # 匹配訂單接口的路徑（比如 /api/order/**）
            -Path=/api/order/**
          filters:
            # 路徑重寫（可選，根據(jù)實(shí)際業(yè)務(wù)調(diào)整）
            -RewritePath=/api/(?<segment>.*),/$\{segment}

        # 路由2：普通流量（normal標(biāo)簽）→ 舊版本服務(wù)（order-service-v1）
        -id:normal_route_v1
          uri:lb://order-service-v1# 舊版本服務(wù)名
          predicates:
            # 普通流量：不滿足灰度斷言，走這條路由
            -Path=/api/order/**
          filters:
            -RewritePath=/api/(?<segment>.*),/$\{segment}

@Component
public class FeignTrafficTagInterceptor implements RequestInterceptor {
    @Override
    public void apply(RequestTemplate template) {
        // 從當(dāng)前請(qǐng)求上下文獲取染色標(biāo)簽（需要用ThreadLocal存儲(chǔ)）
        String trafficTag = TrafficTagContextHolder.get();
        if (trafficTag != null) {
            template.header("X-Traffic-Tag", trafficTag);
        }
    }
}