Spring Cloud Gateway 负载均衡源码分析

Spring Cloud Gateway 负责将请求路由到后端服务，并提供一些内置的负载均衡策略。下面是如何使用Spring Cloud Gateway进行负载均衡的简单示例：

@Configuration
public class GatewayConfig {
 
    @Bean
    public RouteLocator customRouteLocator(RouteLocatorBuilder builder) {
        return builder.routes()
                .route("path_route", r -> r.path("/get")
                        .filters(f -> f.addRequestHeader("Hello", "World"))
                        .uri("lb://SERVICE-NAME"))
                .build();
    }
}

在这个配置中，lb://SERVICE-NAME 指定了Spring Cloud Gateway应该将请求通过负载均衡发送到名为 SERVICE-NAME 的服务。

Spring Cloud Gateway中负载均衡的实现依赖于Spring WebFlux中的ReactorLoadBalancerClient。在实际的请求处理过程中，Gateway中的LoadBalancerClientFilter会与ReactorLoadBalancerClient交互，从而根据提供的服务ID选择正确的服务实例，并进行负载均衡。

如果你想深入了解源码，可以查看LoadBalancerClientFilter的实现，它负责从LoadBalancer获取服务实例，并将请求委托给选定的服务实例。

public class LoadBalancerClientFilter implements GlobalFilter, Ordered {
    // ...
 
    @Override
    public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
        // ...
        ServiceInstance instance = loadBalancer.choose(serviceId, hints);
        if (instance == null) {
            log.trace("LoadBalancerClientFilter [{}]: no instances available for: {}",
                    serviceId, exchange.getRequest().getURI());
            return Mono.error(new NotFoundException("No instances available for "+ serviceId));
        }
 
        // Use the chosen instance to set the URL before filtering continues.
        URI requestUrl = loadBalancer.reconstructURI(instance, exchange.getRequest());
 
        // ...
 
        return chain.filter(exchange);
    }
    // ...
}

在这个代码片段中，loadBalancer.choose(serviceId, hints) 方法负责选择一个服务实例，而loadBalancer.reconstructURI(instance, exchange.getRequest()) 方法则用于构建新的请求URL，该URL将请求委托给选定的服务实例。

要进一步分析源码，你可以查看ReactorLoadBalancer接口及其实现类，如BlockingLoadBalancerClient或ReactiveLoadBalancer等。这些类负责处理服务发现和实例选择逻辑。