【Spring Cloud 源码解读】之 【这也太神奇了,RestTemplate加上一个@LoadBalanced注解就能实现负载均衡!】 (2)

我们会先找拦截器相关的代码,因为此时我们都知道负载均衡主要靠的是拦截器,所以,上代码:

@Bean @ConditionalOnMissingBean public RestTemplateCustomizer restTemplateCustomizer(final LoadBalancerInterceptor loadBalancerInterceptor) { return (restTemplate) -> { // 我们可以看到,如果我们没有自己实现`RestTemplateCustomizer`,就会执行下面的逻辑,而最突兀的就是,它给每一个`RestTemplate`添加了`LoadBalancerInterceptor`,也就是实现负载均衡的重点所在。 List<ClientHttpRequestInterceptor> list = new ArrayList(restTemplate.getInterceptors()); list.add(loadBalancerInterceptor); restTemplate.setInterceptors(list); }; } ④ 何时利用RestTemplateCustomizer来给RestTemplate加拦截器

还有另外一段很重要的代码,需要来解读一下:

首先我们得先了解SmartInitializingSingleton是干嘛的,它的afterSingletonsInstantiated()方法会在所有的单例Bean初始化完成之后,再去一个一个的去处理。

public interface SmartInitializingSingleton { void afterSingletonsInstantiated(); }

那么我们就知道了,接下来要解读的代码就是为了处理一个个带有@LoadBalanced的RestTemplate们,利用RestTemplateCustomizer给RestTemplate们加上拦截器LoadBalancerInterceptor。

@Bean public SmartInitializingSingleton loadBalancedRestTemplateInitializerDeprecated(final ObjectProvider<List<RestTemplateCustomizer>> restTemplateCustomizers) { return () -> { restTemplateCustomizers.ifAvailable((customizers) -> { // 遍历上面提及的成员变量,带@LoadBalanced的RestTemplate们 Iterator var2 = this.restTemplates.iterator(); while(var2.hasNext()) { RestTemplate restTemplate = (RestTemplate)var2.next(); Iterator var4 = customizers.iterator(); while(var4.hasNext()) { // 利用上面的RestTemplateCustomizer给RestTemplate们加拦截器 RestTemplateCustomizer customizer = (RestTemplateCustomizer)var4.next(); customizer.customize(restTemplate); } } }); }; }

所以最后,我们可以给第三个问题一个答案:在带有@LoadBalanced注解的RestTemplate们完成Bean初始化之后,利用RestTemplateCustomizer给RestTemplate们加上拦截器LoadBalancerInterceptor,来实现负载均衡。

3、非负载均衡探索 @Autowrite private RestTemplate restTemplate; @GetMapping("restTemplate-hello") public String sayHello(){ return myRestTemplate.getForObject("http://10.172.29.666:8887/hello",String.class); }

首先可以看到,RestTemplate不再带有拦截器

RestTemplate不带拦截器


而且,我们可以看到,最后接口走的是SimpleBufferingClientHttpRequest,而不是RibbonLoadBalancerClient:

SimpleBufferingClientHttpRequest

到此,关于为什么添加了@LoadBalanced就能进行负载均衡的分析已经结束。而如果大家对Ribbon如何进行负载均衡的也很感兴趣,有空再大家一起研究研究

内容版权声明:除非注明,否则皆为本站原创文章。

转载注明出处:https://www.heiqu.com/zwypsd.html