详解Ribbon-蒲公英云

1.概述

2.使用

2.1.引入

2.2.启用

2.3.切换负载均衡算法

3.负载均衡源码分析

3.1.接口

3.2.抽象类

3.3.选择服务器

3.4.原子性

4.自定义负载均衡算法

1.概述

Ribbon是Netflix开源的一个客户端负载均衡库，也是Spring Cloud Netflix项目的核心组件之一。它主要用于在微服务架构中对服务进行负载均衡，以提高系统的可用性和性能。ribbon不是通信组件，而是服务调用者和通信组件之间的中间层，主要就是用来做负载均衡，选择出适合处理本次请求的服务节点。

现在整个spring cloud体系中的通信组件其实就是封装了负载均衡组件+HTTP通信组件，HTTP通信组件没什么好说的，就是用来发起HTTP请求的，市面上的HTTP通信组件也很多，诸如spring boot自带的RestTemplate，apache的Apache HttpClient等等。值得去探究一下的是负载均衡组件，其作为负载均很组件决定了请求的分发，可以说是整个通信组件的核心。

从Spring Cloud 2020.0版本开始，Spring Cloud官方已经将Ribbon标记为过时（deprecated），推荐使用Spring Cloud LoadBalancer作为替代方案。这样做是基于标准化的考虑，整个spring生态历来都是希望“包容万物”，所以社区自然不希望在负载均衡组件上直接就采用一个固定的实现，而是希望能让三方的解决方案可以平滑的接入、切换。

虽然ribbon以后不再是负载均衡组件的首选，但是作为最经典的负载均衡组件，其底层的一些思想仍然被后面的方案沿用，其实看明白ribbon的源码基本上也就明白负载均衡的原理了，万变不离其宗。

2.使用

2.1.引入

博主的上篇文章详细介绍了怎么使用eureka+通信组件来完成服务的注册、调用，其中也详细的介绍了ribbon的使用，可以移步：

详解Eureka服务注册和调用__BugMan的博客-CSDN博客

总的来说就是：

eureka集成了ribbon，导入eureka后不用单独导入ribbon，但是要注意的是一定要选对版本号，不要选到一个已经把ribbon移除的高版本，本文使用的依赖版本：

当然你也可以直接降级整个spring cloud的版本号，关于spring cloud版本号的问题，可以移步博主的另一篇文章，会有详细讲解：

详解Spring Cloud版本问题__BugMan的博客-CSDN博客

2.2.启用

引入依赖后在承载HTTP通信组件上开启负载均衡即可，此处以spring boot自带的RestTemplate为例，这样在每次客户端（消费者）发起http请求的时候都会在本端进行负载均衡运算后再进行服务访问。

2.3.切换负载均衡算法

负载均衡的核心接口Irule有多个实现类，每个实现类实现不同的负载均衡算法，

常用的有，轮询、随机、可获得、重试等几种：

RoundRobinRule，轮询
RandomRule，随机
AvailabilityFilteringRule，会过滤掉，跳闸，访问故障的服务，对剩下的进行轮询
RetryRule，会按照轮询获取服务，如果服务获取失败，则会在指定的时间内进行重试
需要切换的时候直接在@Configuration中注入@Bean即可：

3.负载均衡源码分析

3.1.接口

IRule接口：

choose：选举出一个服务器
get/setLoadBalancer：获取、修改均衡器

ILoadBalancer：

与服务器打交道，负责寻找、登记服务器

3.2.抽象类

AbstractLoadBalancer实现了Irule接口，重写了均衡器的get/set方法，只留下一个抽象方法——choose，待子类重写。

3.3.选择服务器

所有实现类都继承抽象类AbstractLoadBalancer。各自去重写choose方法，即各自实现不同的负载均衡规则（此处以ribbon默认的负载均衡规则，RoundRobinRule为例）：

choose方法即负载均衡策略，是各负载均衡类的核心方法（此处以ribbon默认的负载均衡规则，RoundRobinRule为例）：

均衡器会和注册中心交互，然后记录下当前整个系统中所有服务器的相关信息，包含服务器总数，可用总数等。

向均衡器所要服务器总数、服务器可用总数，然后根据这两个值进行运算，挑选出承载该此流量的服务器。

public Server choose(ILoadBalancer lb, Object key) {
        if (lb == null) {
            log.warn("no load balancer");
            return null;
        } else {
            Server server = null;
            int count = 0;
            while(true) {
                if (server == null && count++ < 10) {
                    List<Server> reachableServers = lb.getReachableServers();
                    List<Server> allServers = lb.getAllServers();
                    int upCount = reachableServers.size();
                    int serverCount = allServers.size();
                    if (upCount != 0 && serverCount != 0) {
                        int nextServerIndex = this.incrementAndGetModulo(serverCount);
                        server = (Server)allServers.get(nextServerIndex);
                        if (server == null) {
                            Thread.yield();
                        } else {
                            if (server.isAlive() && server.isReadyToServe()) {
                                return server;
                            }
                            server = null;
                        }
                        continue;
                    }
                    log.warn("No up servers available from load balancer: " + lb);
                    return null;
                }
                if (count >= 10) {
                    log.warn("No available alive servers after 10 tries from load balancer: " + lb);
                }
                return server;
            }
        }
    }

整个过程中封装服务器相关信息的Server类，其中有服务器的详细参数：

3.4.原子性

挑选过程中，为了保证原子性，使用了自旋锁（CAS），保证每次处理的只有一个访问线程，其余线程处于自选等待状态：

compareAndSet（期望值，修改值）

期望值，即版本号。

修改值，即要将版本号更新为的值。

判断期望值是否改变（前后是否相同），如果期望值未改变，则将期望值更新为修改值。

返回true，否则返回false。

4.自定义负载均衡算法

存在顶级接口并且可以切换负载均衡算法，那自然可以自定义负载均衡算法，以下是一个根据服务器权重进行负载均衡的一个负载均衡算法：

public class CustomWeightedRandomRule extends AbstractLoadBalancerRule {
    private AtomicInteger totalWeight = new AtomicInteger(0);
    @Override
    public Server choose(Object key) {
        ILoadBalancer lb = getLoadBalancer();
        if (lb == null) {
            return null;
        }
        List<Server> allServers = lb.getAllServers();
        int serverCount = allServers.size();
        if (serverCount == 0) {
            return null;
        }
        int randomWeight = ThreadLocalRandom.current().nextInt(totalWeight.get());
        int currentWeight = 0;
        for (Server server : allServers) {
            currentWeight += getWeight(server);
            if (randomWeight < currentWeight) {
                return server;
            }
        }
        // Fallback to the default server if no server is selected
        return super.choose(key);
    }
    private int getWeight(Server server) {
        // Return the weight of the server (custom logic)
        // Example: return server.getMetadata().getWeight();
        return 1; // Default weight is 1
    }
    @Override
    public void initWithNiwsConfig(IClientConfig clientConfig) {
        super.initWithNiwsConfig(clientConfig);
        // Calculate the total weight of all servers
        List<Server> allServers = getLoadBalancer().getAllServers();
        totalWeight.set(allServers.stream().mapToInt(this::getWeight).sum());
    }
}