Ribbon：自定义负载均衡算法

在springcloud同级，与启动类所在包不同级创建myrule包，写MyWTYRule配置

 

 

Spring Cloud还允许您通过使用@RibbonClient声明其他配置（位于RibbonClientConfiguration之上）来完全控制客户端。例：

@Configuration
@RibbonClient(name = "foo", configuration = FooConfiguration.class)
public class TestConfiguration {
}

在这种情况下，客户端由RibbonClientConfiguration中已经存在的组件与FooConfiguration中的任何组件组成（后者通常会覆盖前者）。

注意：

警告	FooConfiguration必须是@Configuration，但请注意，它不在主应用程序上下文的@ComponentScan中，否则将由所有@RibbonClients共享。如果您使用@ComponentScan（或@SpringBootApplication），则需要采取措施避免包含（例如将其放在一个单独的，不重叠的包中，或者指定要在@ComponentScan）。

 

再写自定义负载均衡配置RandRule.java

package com.company.myrule;


import com.netflix.client.config.IClientConfig;
import com.netflix.loadbalancer.AbstractLoadBalancerRule;
import com.netflix.loadbalancer.ILoadBalancer;
import com.netflix.loadbalancer.Server;

import java.util.List;


public class RandRule extends AbstractLoadBalancerRule {


    //每个服务，访问5次后换下一个服务（3个）

    private int total = 0; //被调用的次数
    private int currentIndex = 0; //当前是谁在提供服务
    public Server choose(ILoadBalancer lb, Object key) {
        System.out.println("=========================================");
        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;
            }

            //int index = chooseRandomInt(serverCount);
            //server = upList.get(index);

            if (total<5){
                server = upList.get(currentIndex);
                total++;
                System.out.println("<5"+total+":"+currentIndex);
            }else {
                total=0;
                currentIndex++;
                if (currentIndex>upList.size()-1){
                    currentIndex = 0;
                    System.out.println(">size"+total+":"+currentIndex);
                }
                server = upList.get(currentIndex);
                System.out.println(">5"+total+":"+currentIndex);
            }

            if (server == null) {

                Thread.yield();
                continue;
            }

            if (server.isAlive()) {
                return (server);
            }


            server = null;
            Thread.yield();
        }

        return server;

    }

    //protected int chooseRandomInt(int serverCount) {
    //    return ThreadLocalRandom.current().nextInt(serverCount);
    //}

    @Override
    public Server choose(Object key) {
        return choose(getLoadBalancer(), key);
    }

    @Override
    public void initWithNiwsConfig(IClientConfig clientConfig) {
        // TODO Auto-generated method stub

    }
}

主启动类添加注解@RibbonClient(name="XXXX",configuration=XXXRule.class)