SpringBoot-dubbo自定义负载均衡实现简单灰度
本文介绍如何利用dubbo自定义负载实现简单灰度(用户纬度,部分用户访问一个服务,其余访问剩余服务)。
其实在这之前,对dubbo了解的也不是很多,只是简单的使用过,跑了几个demo而已,但是得知接下来可能要用dubbo进行开发,还是赶紧补了一下相关的知识,看了看官网,另外买了一本书《深入理解Apache Dubbo实战》,看了一大半,感觉还是很不错的。

1.dubbo负载均衡介绍
因为官网介绍的很详细了,这里只简单的说一下。dubbo负载均衡包含如下四种:
- RandomLoadBalance:默认的负载策略,随机负载。
- ConsistentHashLoadBalance:一致性 Hash负载。
- LeastActiveLoadBalance:最少活跃数负载。
- RoundRobinLoadBalance:轮询负载。
可以查看官方:http://dubbo.apache.org/zh-cn/docs/user/demos/loadbalance.html
这四个类都继承了AbstractLoadBalance抽象类,源码相关分析可以查看官方:http://dubbo.apache.org/zh-cn/docs/source_code_guide/loadbalance.html
2.springboot-dubbo实现自定义负载方法
springboot-dubbo使用自定义负载其实很简单,大致分为如下几步:
- 1.创建自定义负载类,继承AbstractLoadBalance,重写doSelect方法,这个方法就是定义算法规则的地方。
- 2.添加dubbo负载扩展点,在src/main/resources目录下创建META-INFO/dubbo目录,在目录下创建org.apache.dubbo.rpc.cluster.LoadBalance文件,里面配置对应的负载算法类,如下:
gray=com.dalaoyang.balance.GrayLoadBalance
- 3.配置文件中使用,如下:
dubbo.provider.loadbalance=gray
3.模拟灰度方案及具体实现
3.1 灰度场景
现在模拟一个这样的方案,比如有4个服务提供者,端口分别是9001,9002,9003,9004,将其中9002端口的服务设置为灰度服务,当请求消费者接口testUser的userid为1-10时,强制转发到到灰度状态的提供者去,其余的还是请求到正常的服务,如图所示。

3.2 代码实现
接下来使用代码简单实现如上场景。
3.2.1 服务提供者
首先看一下pom文件,都是一些springboot-dubbo的依赖,如下:
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.0.3.RELEASE</version>
<relativePath/> <!-- lookup parent from repository -->
</parent>
<groupId>com.dalaoyang</groupId>
<artifactId>springboot_dubbo_provider</artifactId>
<version>0.0.1-SNAPSHOT</version>
<name>springboot_dubbo_provider</name>
<description>springboot_dubbo_provider</description>
<properties>
<java.version>1.8</java.version>
</properties>
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
<!-- Aapche Dubbo -->
<dependency>
<groupId>org.apache.dubbo</groupId>
<artifactId>dubbo-spring-boot-starter</artifactId>
<version>2.7.1</version>
</dependency>
<dependency>
<groupId>org.apache.dubbo</groupId>
<artifactId>dubbo</artifactId>
<version>2.7.2</version>
<exclusions>
<exclusion>
<groupId>org.springframework</groupId>
<artifactId>spring</artifactId>
</exclusion>
<exclusion>
<groupId>javax.servlet</groupId>
<artifactId>servlet-api</artifactId>
</exclusion>
<exclusion>
<groupId>log4j</groupId>
<artifactId>log4j</artifactId>
</exclusion>
</exclusions>
</dependency>
<dependency>
<groupId>org.apache.curator</groupId>
<artifactId>curator-framework</artifactId>
<version>2.8.0</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.apache.curator/curator-recipes -->
<dependency>
<groupId>org.apache.curator</groupId>
<artifactId>curator-recipes</artifactId>
<version>2.8.0</version>
</dependency>
<dependency>
<groupId>org.apache.zookeeper</groupId>
<artifactId>zookeeper</artifactId>
<exclusions>
<exclusion>
<artifactId>log4j</artifactId>
<groupId>log4j</groupId>
</exclusion>
<exclusion>
<artifactId>slf4j-log4j12</artifactId>
<groupId>org.slf4j