nacos实现指定ribbon负载均衡策略的三种方式

本文简单介绍下为nacos指定ribbon中相应的负载均衡策略的三种方式

一、通过配置类方式指定负载均衡策略

1、定义配置类

/**
 * @description 通过配置文件的方式为消费者也就是客户端设置负载均衡策略
 * 需要注意，该配置文件需要放在ComponentScan所设置的文件夹之外（没设置ComponentScan，默认是application启动类所在的包），
 * 否则会被其他客户端共享
 * 最后还需要在启动类上指定哪些服务提供商需要引用该负载均衡策略，并加上当前配置类
 * @Date 2023/12/27 10:37
 * @return: 返回一个随机策略
 */

@Configuration
public class RibbonRandomRuleConfig {

    // 方法名一定需要叫 iRule
    @Bean
    public IRule iRule(){
        return new RandomRule();
    }
}

注意：该类的路径不要和启动类放在同一包下，本人目录结构如下

2、启动类上加相应注解

@SpringBootApplication
@RibbonClients(value = {
        @RibbonClient(name = "stock-service",configuration = RibbonRandomRuleConfig.class)
})
public class OrderRibbonApplication {
    public static void main(String[] args) {
        SpringApplication.run(OrderRibbonApplication.class,args);
    }
}

启动类上的注解 stock-service 为服务提供者，RibbonRandomRuleConfig 为我们定义的配置类，表明客户端即消费者（order)采用了RibbonRandomRuleConfig 中定义的负载均衡策略来调用服务提供者。

二、通过配置文件指定负载均衡策略

通过配置文件方式来指定nacos对应的负载均衡策略简单方便，只需在yml文件中填写如下配置即可，无需在启动类上加注解

stock-service:
  ribbon:
    #  基于权重的负载均衡策略
    NFLoadBalancerRuleClassName: com.alibaba.cloud.nacos.ribbon.NacosRule

其中 stock-service 为服务提供者，com.alibaba.cloud.nacos.ribbon.NacosRule 为权重策略类的全路径

三、自定义负载均衡策略

除了使用ribbon中现有的几种负载均衡策略之外，还可以自定义自己的负载均衡策略

1、实现choose接口

public class CustomRule extends AbstractLoadBalancerRule {
    @Override
    public void initWithNiwsConfig(IClientConfig iClientConfig) {

    }

    @Override
    public Server choose(Object o) {
        // 获取当前可用的服务实例列表
        ILoadBalancer loadBalancer = this.getLoadBalancer();
        List<Server> reachableServers = loadBalancer.getReachableServers();
        // 采用线程安全的方式生成随机数
        int random = ThreadLocalRandom.current().nextInt(reachableServers.size());
        // 返回获取的随机服务实例
        return reachableServers.get(random);
    }
}

通过继承 AbstractLoadBalancerRule 方法，实现choose类来自定义我们的负载均衡策略，其中choose方法是ribbon中返回所选则的服务提供者的一个接口，我们通过实现该接口来自定义负载均衡策略，这里定义一个随机的负载均衡策略。

2、修改yml配置文件

stock-service:
  ribbon:
    # 使用自定义的负载均衡策略
    NFLoadBalancerRuleClassName:  com.tuling.ribbon.CustomRule