《RPC手撸专栏》第55章：基于SPI扩展增强型Zookeeper一致性Hash算法负载均衡策略

作者：冰河
星球：http://m6z.cn/6aeFbs (opens new window)
博客：https://binghe.gitcode.host (opens new window)
文章汇总：https://binghe.gitcode.host/md/all/all.html (opens new window)

沉淀，成长，突破，帮助他人，成就自我。

大家好，我是冰河~~

在前面的章节中，我们基于SPI扩展了JDK、Json、Hessian2、FST、Kryo和Protostuff序列化与反序列化机制，在服务消费者端基于SPI扩展了JDK、CGLib、Javassist、ByteBuddy和ASM动态代理机制。在服务提供者端，基于SPI扩展了JDK、CGLib、Javassist、ByteBuddy和ASM反射机制调用真实方法的功能。

一、前言

实现增强型负载均衡策略，我们继续肝...

在前面的文章中，我们一起实现了基于SPI扩展随机算法、加权随机算法、轮询算法、加权轮询算法、Hash算法、加权Hash算法、基于源IP地址Hash算法、基于源IP地址加权Hash算法和基于Zookeeper的一致性Hash算法负载均衡策略，服务消费者可以根据这些算法实现的负载均衡策略，从多个服务提供者中选择一个，与其进行连接从而实现数据交互。

在增强型负载均衡策略中，我们一起实现了基于SPI扩展增强型加权随机算法、增强型加权轮询算法、增强型加权Hash算法和增强型加权源IP地址Hash算法负载均衡策略，本章，我们继续往下肝，小伙伴们别掉队，一起加油！

二、目标

目标很明确：基于SPI扩展增强型Zookeeper一致性Hash算法负载均衡策略！

增强型负载均衡策略最核心的实现逻辑就是：根据权重重新生成服务元数据列表，权重越高的元数据，会在最终的列表中出现的次数越多。例如，权重为1，最终出现1次，权重为2，最终出现2次，权重为3，最终出现3次，依此类推。此时，不管是使用何种算法，出现次数越多的服务元数据，被选中的概率越大，一个服务元数据就代表了一个服务提供者实例。

这样，使用增强型Zookeeper一致性Hash算法负载均衡策略时，权重越高的服务提供者实例，被选中的概率越大。这样，使用RPC框架时，可以根据服务提供者所在的服务器资源情况，为当前服务器上部署的RPC服务提供者实例配置权重信息。权重越大，使用Hash算法被选中的概率越大。

这一章，我们就一起手撸实现基于SPI扩展增强型Zookeeper一致性Hash算法负载均衡策略。

三、设计

如果让你设计基于SPI扩展增强型Zookeeper一致性Hash算法的负载均衡策略，你会怎么设计呢？

基于SPI扩展增强型Zookeeper一致性Hash算法的负载均衡策略的流程如图55-1所示。

由图55-1可以看出如下信息：

（1）服务提供者会通过自定义类扫描器整合注册中心，将服务注册到注册中心。

（2）服务注册到注册中心的元数据，例如服务的名称、服务的版本号、服务地址、服务端口和服务分组等信息，元数据会贯穿整个服务的注册与发现流程。

（3）服务注册与发现接口对外提供服务注册与发现的方法，服务提供者通过自定义扫描器会调用服务注册与发现接口的方法实现服务注册功能。

（4）基于服务注册与发现的接口，服务提供者会实现多个服务注册与发现的实现类，每个实现类对应着一种注册中心服务。

（5）服务消费者会通过服务注册与发现接口订阅注册中心的服务，会从注册中心获取到服务提供者发布的服务信息，实现服务发现的功能。

（7）服务消费者从注册中心获取到服务提供者发布的服务信息后，会基于SPI机制动态加载普通算法（我们将第42章~第50章实现的负载均衡算法统称为普通算法）、基于增强型加权随机算法、基于增强型加权轮询算法、基于增强型加权Hash算法、基于增强型加权源IP地址Hash算法和基于增强型Zookeeper一致性Hash算法的负载均衡策略，从多个服务中选择一个进行远程网络连接。

（8）服务消费者会直接与根据基于SPI机制动态加载的负载均衡策略选择出的服务提供者建立连接，实现数据交互。也就是说，后续服务消费者会与服务提供者直接实现数据交互。

四、实现

说了这么多，具体要怎么实现呢？

核心类实现关系

基于SPI扩展增强型Zookeeper一致性Hash算法负载均衡策略的核心类关系如图55-2所示。

查看完整文章

加入冰河技术 (opens new window)知识星球，解锁完整技术文章与完整代码