随着业务体量的增大,业务复杂的同时,我们承载业务的机器也越来越多。日常、预发、线上每个环境都需要些许资源来运行服务。但当我们没有预发环境,而某些业务需要在线上验证时,我们能否将我们验证的请求固定的打到我们指定的某一台机器上呢?
Dubbo作为一款优秀的服务通信框架,为我们强大功能的同时也为我们提供了可插拔式扩展方式。通过Dubbo SPI机制,我们可以轻松实现我们自己的负载均衡机制。
下面通过简易代码来演示下如何基于Dubbo SPI 来实现我们的业务需求。
首先 描述下背景,项目采用SpringBoot 2.3.3 、Dubbo版本采用的是Apache 2.7.7 版本,虽然这个版本新增来tag特性可以更方便的来实现负载策略,但为了演示,还是采用了普通版本也可使用的SPI机制来实现。
<dependency> <groupId>org.apache.dubbogroupId> <artifactId>dubbo-spring-boot-starterartifactId> <version>2.7.7version>dependency>
首先,我们自定义一个类去实现 AbstractLoadBalance 抽象类,并实现其中的doSelect方法(Dubbo内置的四种负载也是实现该抽象类),便可在此处书写负载逻辑。
示例逻辑:如果Rpc上下文中的env变量=gary,则代表走灰度环境,否则 走随机策略。