Skip to main content
Version: 2.4.3

自定义负载均衡策略

本文介绍如何对 org.apache.shenyu.loadbalancer.spi.LoadBalancer 进行自定义扩展。

  • 新建一个工程,引入如下依赖:
<dependencies>    <dependency>        <groupId>org.apache.shenyu</groupId>        <artifactId>shenyu-plugin-base</artifactId>        <version>${project.version}</version>    </dependency></dependencies>
  • 新增一个类 CustomLoadBalancer,继承org.apache.shenyu.loadbalancer.spi.AbstractLoadBalancer
public class CustomLoadBalancer extends AbstractLoadBalancer {
    @Override    public Upstream doSelect(final List<Upstream> upstreamList, final String ip) {        // 自定义负载均衡实现逻辑    }}
  • 在工程的META-INF/services目录创建 org.apache.shenyu.loadbalancer.spi.LoadBalancer文件中添加如下内容:
script
${spi name}=${custom class path}

${spi name}表示spi的名称,${custom class path}表示该类的全限定名。比如:

script
custom=xxx.xxx.xxx.CustomLoadBalancer
  • 将工程打包,拷贝到网关 (bootstrap-bin) 的 libext-lib 目录。

  • Apache ShenYu网关管理系统 --> 基础配置 --> 字典管理, 找到字典编码为 LOAD_BALANCE,新增一条数据,注意字典名称要为: ${spi name},图中的示例是custom

字典类型:loadBalance

字典编码:LOAD_BALANCE

字典名称:${spi name},填写自定义spi的名称;

字典值:使用时,下拉框的值,不要和现有的重复;

字典描述或备注信息:描述信息;

排序: 排序;

  • 在添加选择器或规则时,就可以使用自定义的匹配方式: