@LoadBalanced 实现原理代码解释

@LoadBalanced 的实现原理主要是通过为 RestTemplate 添加 LoadBalancerInterceptor 拦截器实现的。下面是代码解释：

首先，我们需要在 Spring Boot 应用的配置类中添加注解 @LoadBalanced，如下所示：

@Configuration
public class MyConfig {

    @Bean
    @LoadBalanced
    public RestTemplate restTemplate() {
        return new RestTemplate();
    }
}

这样，我们就为 RestTemplate 添加了 @LoadBalanced 注解，并将其注入到 Spring 容器中。

然后，在使用 RestTemplate 发送请求时，@LoadBalanced 注解会自动为我们添加 LoadBalancerInterceptor 拦截器。这个拦截器的作用是将请求转发给 Ribbon 进行负载均衡，选择可用的服务实例。

LoadBalancerInterceptor 的实现代码如下：

public class LoadBalancerInterceptor implements ClientHttpRequestInterceptor {

    private LoadBalancerClient loadBalancer;

    public LoadBalancerInterceptor(LoadBalancerClient loadBalancer) {
        this.loadBalancer = loadBalancer;
    }

    @Override
    public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
        URI originalUri = request.getURI();
        String serviceName = originalUri.getHost();
        ServiceInstance serviceInstance = loadBalancer.choose(serviceName);
        URI uri = serviceInstance.getUri();
        uri = UriComponentsBuilder.fromUri(uri).path(originalUri.getPath()).build().toUri();
        return execution.execute(new ServiceRequestWrapper(request, uri), body);
    }

    private static class ServiceRequestWrapper extends HttpRequestWrapper {

        private final URI uri;

        public ServiceRequestWrapper(HttpRequest request, URI uri) {
            super(request);
            this.uri = uri;
        }

        @Override
        public URI getURI() {
            return uri;
        }
    }
}

在这个拦截器中，我们首先获取请求的原始 URI，然后从中获取服务名。接着，通过 LoadBalancerClient 选择一个可用的服务实例，得到该实例的 URI。然后，我们将请求的目标地址和端口替换为服务实例的地址和端口，最后将请求转发给该实例。

综上所述，@LoadBalanced 实际上是将负载均衡的逻辑封装在了 LoadBalancerInterceptor 拦截器中，而这个拦截器会自动添加到 RestTemplate 中。这样，我们就可以通过简单的注解来实现客户端负载均衡。