当前位置: 代码迷 >> 综合 >> Spring Cloud Ribbon 客户端负载均衡~利用拦截器使RestTemplate对象拥有负载均衡功能
  详细解决方案

Spring Cloud Ribbon 客户端负载均衡~利用拦截器使RestTemplate对象拥有负载均衡功能

热度:22   发布时间:2023-12-08 17:13:51.0

负载均衡,英文名称为Load Balance,其含义就是指将负载(工作任务)进行平衡、分摊到多个操作单元上进行运行,例如FTP服务器、Web服务器、企业核心应用服务器和其它主要任务服务器等,从而协同完成工作任务。
负载均衡构建在原有网络结构之上,它提供了一种透明且廉价有效的方法扩展服务器和网络设备的带宽、加强网络数据处理能力、增加吞吐量、提高网络的可用性和灵活性。

在这里插入图片描述
本文中所关注的Spring Cloud Ribbon是客户端负载均衡组件,它依赖于一个被@LoadBalance注解所标识的RestTemplate对象实现负载均衡逻辑。在业务代码中发送请求时使用服务名替代真实的ip和port。

在这里插入图片描述

图(1)

在这里插入图片描述

图(2)
@AutoConfigureBefore({
    LoadBalancerAutoConfiguration.class, AsyncLoadBalancerAutoConfiguration.class})
public class RibbonAutoConfiguration {
    //...
}

根据两个配置类上的注解可知: RibbonAutoConfiguration配置类会优先于LoadBalancerAutoConfiguration配置类执行

RestTemplatespring-web包中的类,为什么在创建RestTemplate对象时使用@LoadBalanced注解标记后就支持了负载均衡?

LoadBalancerAutoConfiguration.java

	@LoadBalanced@Autowired(required = false)private List<RestTemplate> restTemplates = Collections.emptyList();

可以看到,restTemplates对象也有@LoadBalanced注解,被@LoadBalanced注解标记的RestTemplate对象会被注入到restTemplates对象中(实现逻辑见@Qualifier)

	@Beanpublic SmartInitializingSingleton loadBalancedRestTemplateInitializerDeprecated(final ObjectProvider<List<RestTemplateCustomizer>> restTemplateCustomizers) {
    return () -> restTemplateCustomizers.ifAvailable(customizers -> {
    for (RestTemplate restTemplate : LoadBalancerAutoConfiguration.this.restTemplates) {
    for (RestTemplateCustomizer customizer : customizers) {
    customizer.customize(restTemplate);}}});}@Configuration@ConditionalOnMissingClass("org.springframework.retry.support.RetryTemplate")static class LoadBalancerInterceptorConfig {
    @Beanpublic LoadBalancerInterceptor ribbonInterceptor(LoadBalancerClient loadBalancerClient,LoadBalancerRequestFactory requestFactory) {
    return new LoadBalancerInterceptor(loadBalancerClient, requestFactory);}@Bean@ConditionalOnMissingBeanpublic RestTemplateCustomizer restTemplateCustomizer(final LoadBalancerInterceptor loadBalancerInterceptor) {
    return restTemplate -> {
    List<ClientHttpRequestInterceptor> list = new ArrayList<>(restTemplate.getInterceptors());list.add(loadBalancerInterceptor);restTemplate.setInterceptors(list);};}}

上述代码为@LoadBalanced注解标记的RestTemplate对象添加LoadBalancerInterceptor拦截器

RibbonAutoConfiguration.java

	@Bean@ConditionalOnMissingBean(LoadBalancerClient.class)public LoadBalancerClient loadBalancerClient() {
    return new RibbonLoadBalancerClient(springClientFactory());}

由上述代码可知,创建LoadBalancerInterceptor对象时传入的LoadBalancerClient参数是RibbonLoadBalancerClient对象

如何将基于服务名的请求转换为基于ip:port的请求?
在这里插入图片描述
当找到实例后会调用LoadBalancerRequest#apply(ServiceInstance instance)
在这里插入图片描述
ServiceRequestWrapper通过重写HttpRequestWrapper中的getURI()实现根据实例信息中的ip、port 替换URL中的服务名,然后交由ClientHttpRequestExecution对象执行请求。

  相关解决方案