聊聊Ribbon源码解读之负载均衡

271 阅读2分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第14天,点击查看活动详情

聊聊Ribbon源码解读之负载均衡

上篇文章我们说到RibbonLoadBalancerClient的execute()方法需要获取负载均衡器,这篇文章我们就看一下到底是怎么获取负载均衡器的。

负载均衡器

深入getLoadBalancer()方法我们发现它就是调用SpringClientFactory的getInstance()方法获取的是ILoadBalancer的实例,那么ILoadBalancer实例是在哪里定义的呢?

我们看一下RibbonAutoConfiguration类

@Bean
  public SpringClientFactory springClientFactory() {
    SpringClientFactory factory = new SpringClientFactory();
    factory.setConfigurations(this.configurations);
    return factory;
  }

这里定义了SpringClientFactory类,它的构造方法中使用了RibbonClientConfiguration类,RibbonClientConfiguration中定义了ILoadBalancer的实现类ZoneAwareLoadBalancer,它是在选择服务器时可以避开一个区域的负载均衡器。通过看这一段代码就水落石出了,ILoadBalancer实例是在RibbonAutoConfiguration类中定义的。

通过负载均衡器选择服务

getServer()方法中调用ILoadBalancer实例的chooseServer(); ILoadBalancer实例我们分析是ZoneAwareLoadBalancer

这里的判断getLoadBalancerStats().getAvailableZones().size()默认是1,所以会执行父类BaseLoadBalancer的chooseServer()方法,这里用到的IRule来进行选择服务,通过RibbonClientConfiguration中的IRule接口的注入,可以知道是ZoneAvoidanceRule,而ZoneAvoidanceRule没有重写choose()方法,而是使用父类PredicateBasedRule的choose()方法,从过滤后的服务列表中选择一个服务,具体是incrementAndGetModulo()方法:

private int incrementAndGetModulo(int modulo) {
        for (;;) {
            int current = nextIndex.get();
            int next = (current + 1) % modulo;
            if (nextIndex.compareAndSet(current, next) && current < modulo)
                return current;
        }
    }
  1. 获取当前服务实例的索引值
  2. 通过求余数的方法记录下一个索引值
  3. 使用CAS,nextIndex和current相等的时候,设置值nextIndex为next
  4. 返回当前服务实例的索引值

这其实就是轮询操作的逻辑,轮询操作是常用的也就是比较简单的一种负载均衡策略。

执行RibbonLoadBalancerClient的execute()方法

execute()执行方法中的关键代码是request.apply(serviceInstance);

request是LoadBalancerRequest的实例,而通过拦截器LoadBalancerInterceptor调用this.loadBalancer.execute(),LoadBalancerRequest实例的生成是通过LoadBalancerRequestFactory 的createRequest()生成的 ,最终调用了AbstractClientHttpRequest的execute()方法,

而我们使用restTemplate.getForObject()进行远程调用的时候,也是调用了AbstractClientHttpRequest的execute()方法

总结

这篇文章主要解读了RibbonLoadBalancerClient的execute()方法中的三个重要步骤:获取负载均衡器、选取服务、执行execute()方法,其中负载均衡器是ZoneAwareLoadBalancer,选取服务是通过轮询机制,然后执行执行RibbonLoadBalancerClient的execute()方法