RestTemplate基于Ribbon负载均衡实现原理 上

都知道 RestTemplate 是基于 Ribbon实现的负载均衡,那么Ribbon又是如何做到的 ?

首先找到 spring-cloud-netflix-ribbon-2.1.0.RELEASE.jar 这个jar包

1.自动配置创建RibbonAutoConfiguration

查看META-INF下的spring.factories文件

org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
org.springframework.cloud.netflix.ribbon.RibbonAutoConfiguration

利用了SpringBoot自动配置机制,往容器中添加了一个自动配置类 RibbonAutoConfiguration

2. 加载 LoadBalancerAutoConfiguration

`RibbonAutoConfiguration 位于 org.springframework.cloud.netflix.ribbon 包下

注意这个@AutoConfigureAfter注解 在当前类加载完成后,将加载 LoadBalancerAutoConfiguration 自动配置类

跟进 LoadBalancerAutoConfiguration 自动配置类

3.找到需要定制的RestTemplate对象

来到 org.springframework.cloud.client.loadbalancer.LoadBalancerAutoConfiguration

首先能看到一个成员属性 restTemplates 类型是List 也就是 RestTemplate 的集合

头上标了一个 @Autowired 和一个 @LoadBalanced 注解 很迷惑对吧 什么时候 @Autowired 能和 @LoadBalanced 搭配使用了 ?

点进 @LoadBalanced 注解 可以看到有一个 @Qualifier 注解 恍然大悟了吧 原来这一步是在注入容器中 所有标记了 @LoadBalanced注解的 RestTempalte 对象

4.定制 RestTemplate 添加拦截器

往下 来到 静态内部类的LoadBalancerInterceptorConfig的ribbonInterceptor方法 可以看到往容器中添加了一个 LoadBalancerInterceptor 对象

继续往下看 来到 restTemplateCustomizer 方法

返回值是一个函数式接口

这里相当于往容器中存了一个lombad表达式(匿名实现类) 注意此时并未执行

然后我们回到上面的代码

找到 loadBalancedRestTemplateInitializerDeprecated 方法 打个断点

` LoadBalancerAutoConfiguration.this.restTemplates 也就是上面自动注入的所有标注了@LoadBalanced 注解的RestTemplate对象集合

进行遍历 然后遍历RestTemplateCustomizer接口的匿名实现类执行customize方法 此时将真正执行lombda表达式

回到 restTemplateCustomizer 方法 看这个lombad表达式都做了什么

首先拿到容器中的loadBalancerInterceptor拦截器,然后给restTemplate对象添加

而loadBalancerInterceptor拦截器 就是来完成负载均衡的