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拦截器 就是来完成负载均衡的