SpringCloud-Ribbon -- 负载均衡配置 及 自定义负载均衡算法

目录

Ribbon负载均衡策略

配置

1）通过配置文件配置

2）通过java注解配置

单独使用Ribbon

自定义负载均衡算法

自定义算法RandomRule_ZY

注入bean

添加ribbon注解

---

Ribbon负载均衡策略

配置

对调用的某个服务启用某种负载策略

1）通过配置文件配置

hello:
    ribbon:
     NFLoadBalancerRuleClassName:com.netflix.loadbalancer.RandomRule

2）通过java注解配置

// @Configuration -- 通过注解@RibbonClient为特定的服务配置负载均衡策略，
//                   这时候该注解需去掉，否则会执行两次RandomRule()
public class RibbonConfiguration{
      @Bean
      public IRule ribbonRule(){
          //随机负载
          return new RandomRule();
     }
}

通过注解@RibbonClient为特定的服务配置负载均衡策略

@Configuration
@RibbonClient(name="hello", configuration=RibbonConfiguration.class)
    public class TestRibbonConfiguration{
}

以上配置都是在服务消费者中配置。

单独使用Ribbon

      因为往往Ribbon配合Eureka使用的，往往也有第三方服务没有注册到Eureka Server，但也部署了多个实例，也需要进行负载均衡，这时可以在服务消费者的配置文件中进行如下方式配置，实现负载均衡

hello:
   ribbon:
      listOfServers:localhost:8010,localhost:8011

自定义负载均衡算法

官方文档指出：

这个自定义的类不能放在@ComponentScan所扫描的当前包以及子包下，否则我们自定义的这个配置类就会被所有的Ribbon客户端所共享，也就是我们达不到特殊化指定的目的了。

为何不能放在此位置：源码解析如下：

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@SpringBootConfiguration
@EnableAutoConfiguration
@ComponentScan(
    excludeFilters = {
      @Filter(
        type = FilterType.CUSTOM,
        classes = {TypeExcludeFilter.class}
      ), 
      @Filter(
        type = FilterType.CUSTOM,
        classes = {AutoConfigurationExcludeFilter.class}
      )
    }
)
public @interface SpringBootApplication {

因为主启动类已经有ComponentScan，这个注解了，所以说不能放在和主启动类同包下。

自定义算法RandomRule_ZY

自定义算法RandomRule_ZY 必须继承AbstractLoadBalanceRule类

/**自定义的负载均衡算法*/
public class RandomRule_ZY extends AbstractLoadBalancerRule {
    // total = 0 // 当total==5以后，我们指针才能往下走，
    // index = 0 // 当前对外提供服务的服务器地址，
    // total需要重新置为零，但是已经达到过一个5次，我们的index = 1
    // 分析：我们5次，但是微服务只有8001 8002 8003 三台，OK？
    private int total = 0;            // 总共被调用的次数，目前要求每台被调用5次
    private int currentIndex = 0;    // 当前提供服务的机器号
    public Server choose(ILoadBalancer lb, Object key) {
        if (lb == null) {
            return null;
        }
        Server server = null;
        while (server == null) {
            if (Thread.interrupted()) {
                return null;
            }
            List<Server> upList = lb.getReachableServers();
            List<Server> allList = lb.getAllServers();
            int serverCount = allList.size();
            if (serverCount == 0) {
                return null;
            }
            //private int total = 0; // 总共被调用的次数，目前要求每台被调用5次
            //private int currentIndex = 0; // 当前提供服务的机器号
            if (total < 5) {
                server = upList.get(currentIndex);
                total++;
            } else {
                total = 0;
                currentIndex++;
                if (currentIndex >= upList.size()) {
                    currentIndex = 0;
                }
            }
            if (server == null) {
                Thread.yield();
                continue;
            }
            if (server.isAlive()) {
                return (server);
            }
            server = null;
            Thread.yield();
        }
        return server;
    }
 
    @Override
    public Server choose(Object key) {
        return choose(getLoadBalancer(), key);
    }
    @Override
    public void initWithNiwsConfig(IClientConfig clientConfig) {
    }
}

注入bean

如果想使用某个自定义算法则注入到bean中即可。

/**
 * 自定义负载均衡算法
 */
// @Configuration -- 通过注解@RibbonClient为特定的服务配置负载均衡策略，
//                   这时候该注解需去掉，否则会执行两次RandomRule()
public class MySelfRule {
     @Bean
    public IRule myRule() {
        //return new RoundRobinRule();//轮询算法
        //return new RandomRule();//随机算法
        return new RandomRule_ZY();
    }
 
}

添加ribbon注解

在主启动类中添加ribbon注解

@SpringBootApplication
@EnableEurekaClient
@RibbonClient(name = "MICROSERVICECLOUD-DEPT", configuration = MySelfRule.class)
public class DeptConsumer80_App {
    public static void main(String[] args) {
        SpringApplication.run(DeptConsumer80_App.class, args);
    }
}

      特别注意的是@RibbonClient(name = "MICROSERVICECLOUD-DEPT", configuration = MySelfRule.class)

name指定针对哪个服务 进行负载均衡，而configuration指定负载均衡的算法具体实现类

       测试：启动80负载均衡微服务，然后启动一个7001微服务注册中心，然后启动8001,8002,8003，访问80负载均衡客户端，每次刷新的结果都是按照新写的负载均衡算法去执行，达到效果。

至此：自定义负载均衡算法开发完毕