spring cloud gateway 3.1.7版本 实现自定义负载均衡

在Spring Cloud Gateway 3.1.7版本中，自定义负载均衡策略可以通过实现`org.springframework.cloud.gateway.loadbalancer.LoadBalancer`接口来实现，该接口定义了选择服务实例的方法。

以下是一个示例，展示了如何定义一个基于特定请求头的自定义负载均衡策略：

public class CustomLoadBalancer implements LoadBalancer<ServiceInstance> {

    private final String headerName;

    public CustomLoadBalancer(String headerName) {
        this.headerName = headerName;
    }

    @Override
    public ServiceInstance choose(Object key) {
        if (key instanceof ServerWebExchange) {
            ServerWebExchange exchange = (ServerWebExchange) key;
            Object headerValue = exchange.getRequest().getHeaders().getFirst(headerName);
            String serviceName = "my-service";
            // 根据请求头的值选择服务实例
            ServiceInstance serviceInstance = ...;
            return serviceInstance;
        }
        return null;
    }
}

在上面的示例中，`CustomLoadBalancer`实现了`LoadBalancer`接口，并基于特定的请求头选择服务实例。

然后，你需要创建一个`org.springframework.cloud.gateway.filter.factory.rewrite.RewriteFunction`实例，用于将服务的URI重写为负载均衡的服务实例地址。最后，你需要将这个自定义的负载均衡策略应用到Spring Cloud Gateway的路由规则中。

以下是一个完整的示例，展示了如何定义和应用自定义的负载均衡策略：

@Bean
public LoadBalancer<ServiceInstance> customLoadBalancer() {
    return new CustomLoadBalancer("my-header");
}

@Bean
public RouteLocator customRouteLocator(RouteLocatorBuilder builder, LoadBalancer<ServiceInstance> customLoadBalancer) {
    return builder.routes()
            .route(r -> r.path("/my-service/**")
                    .uri("http://my-service")
                    .id("my-route")
                    .filter(new CustomLoadBalancerGatewayFilterFactory(customLoadBalancer)))
            .build();
}

public class CustomLoadBalancerGatewayFilterFactory extends AbstractGatewayFilterFactory<CustomLoadBalancerGatewayFilterFactory.Config> {

    private final LoadBalancer<ServiceInstance> loadBalancer;

    public CustomLoadBalancerGatewayFilterFactory(LoadBalancer<ServiceInstance> loadBalancer) {
        super(Config.class);
        this.loadBalancer = loadBalancer;
    }

    @Override
    public GatewayFilter apply(Config config) {
        RewriteFunction<String, String> rewriteFunction = uri -> {
            // 将URI重写为负载均衡的服务实例地址
            ServiceInstance serviceInstance = loadBalancer.choose(ServerWebExchangeUtils.GATEWAY_REQUEST_URL_ATTR).block();
            return "http://" + serviceInstance.getHost() + ":" + serviceInstance.getPort() + uri;
        };
        return new RewritePathGatewayFilterFactory().apply(new RewritePathGatewayFilterFactory.Config().setRewriteFunction(rewriteFunction));
    }

    public static class Config {
        // 配置属性
    }
}

在上面的示例中，`customLoadBalancer`方法定义了自定义的负载均衡策略`CustomLoadBalancer`。`customRouteLocator`方法创建了一个路由规则，将请求路径`/my-service/**`映射到`http://my-service`服务上，并使用自定义负载均衡策略。`CustomLoadBalancerGatewayFilterFactory`则将自定义负载均衡策略应用到路由规则中，并将服务的URI重写为负载均衡的服务实例地址。最后，你可以在路由规则中使用`CustomLoadBalancerGatewayFilterFactory`来定义自定义的负载均衡策略。

spring:
  cloud:
    gateway:
      discovery:
        locator:
          enabled: false
      routes:
        - id: my-route
          uri: http://my-service
          predicates:
            - Path=/my-service/**
          filters:
            - CustomLoadBalancer=my-header