在SpringCloud基础1.3章节中介绍过,使用SpringCloud的ribbon作为客户端负载均衡非常简单,只需要两步操作即可:1.注册消费者;2.通过调用被@LoadBalanced注解修饰的RestTemplate来实现面向服务的接口调用。
为什么使用被@LoadBalanced注解修饰的RestTemplate发起调用就能实现ribbon客户端负载均衡呢?本文将结合源码进行详细解释。
LoadBalancerAutoConfiguration
在LoadBalanced注解的注释里面有这么一句话:
Annotation to mark a RestTemplate bean to be configured to use a LoadBalancerClient
意思就是说:该注解用于标记一个使用LoadBalancerClient进行配置的RestTemplate。
这说明通过RestTemplate进行接口调用时,使用了LoadBalancerClient。而在LoadBalancerClient类的同级目录下,很容易就找到了LoadBalancerAutoConfiguration。
继续找LoadBalancerAutoConfiguration的useage,发现在spring.factories中有入口:
到这里就比较清晰了。在SpringBootApplication启动的时候,执行自动装配逻辑,找到所有的META-INF路径下的spring.factories文件,初始化里面的自动配置类,因此LoadBalancerAutoConfiguration也得到初始化。
在该类的初始化过程中,主要做了三件事:
- 创建了ribbonInterceptor的bean,用于对客户端请求进行拦截,实现负载均衡。
- 创建了一个RestTemplateCustomizer的bean,用于给RestTemplate增加loadBalancerInterceptor。
- 维护了一个被@LoadBalanced注解修饰的RestTemplate对象列表,并在这里进行初始化,通过调用RestTemplateCustomizer的实例来给需要客户端负载均衡的RestTemplate增加LoadBalancerInterceptor拦截器。
LoadBalancerInterceptor
通过上文的分析可以知道,LoadBalancerAutoConfiguration类在初始化的时候对所有被@LoadBalanced修饰的restTemplate加一个负载均衡拦截器。通过restTemplate发起服务调用时,调用请求会被interceptor类进行拦截。
拦截器方法如下:
这里的loadBalancer是上文创建拦截器时候传入的RibbonLoadBalancerClient实例,execute方法如下:
首先根据服务名获取ribbon的负载均衡器。那么这里取到的负载均衡器是哪一个呢?
事实上,ribbon的负载均衡器类体系结构如下图所示:
其中,BaseloadBalancer类实现了基础的负载均衡,而DynamicServerListLoadBalancer和ZoneAwareLoadBalancer在负载均衡的策略上做了一些功能的扩展。
在RibbonClientConfiguration类中,默认返回的负载均衡器是ZoneAwareLoadBalancer。
因此在RibbonLoadBalancerClient的execute方法中,获取到的ribbon负载均衡器就是ZoneAwareLoadBalancer。
然后通过ZoneAwareLoadBalancer来获取实际的服务实例,将其包装成RibbonServer对象。
接着使用RibbonServer对象回调LoadBalancerRequest的apply函数,向RibbonServer对象中的具体服务实例发起请求。
参考资料
[1].https://www.jianshu.com/p/1bd66db5dc46