自定义负载均衡策略

说明

• 在自定义开发前，请先自定义搭建好网关环境，请参考: 自定义部署

• 本文介绍如何对 org.apache.shenyu.plugin.divide.balance.LoadBalance 进行自定义扩展。

扩展实现

• 新增一个类 ${you class}，实现 org.apache.shenyu.plugin.divide.balance.LoadBalance

public class ${you class} implements LoadBalance {       /**     * 从上游列表中选择一个服务返回     *     * @param upstreamList 上游列表     * @param ip 请求的远程地址     * @return divide upstream     */    @Override    public DivideUpstream doSelect(final List<DivideUpstream> upstreamList,                                    final String ip) {        //开发你的选择逻辑，返回结果    }}

• 在项目 resources 目录下，新建 META-INF/shenyu 目录， 并且新增文件名为 : org.apache.shenyu.plugin.divide.balance.LoadBalance. 内容新增 ${you spi name} = ${you class path}:

${you spi name} = ${you class path}

• 在 Admin 后台 ---> 基础管理 ---> 字典管理 , 找到字典编码为 LOAD_BALANCE，新增一条数据，注意字典名称要为: ${you spi name}。

• 或者执行以下自定义SQL语句：

INSERT IGNORE INTO `shenyu_dict` (        `id`,        `type`,        `dict_code`,        `dict_name`,        `dict_value`,        `desc`,        `sort`,        `enabled`,        `date_created`,        `date_updated`    )VALUES (        'you id',        'matchMode',        'MATCH_MODE',        'you spi name',        'you value',        'you spi name',        0,        1,        '2021-08-30 19:29:10',        '2021-08-30 20:15:23'    );