自定义Ribbon负载均衡

需要在基包的上一级定义，不然会被扫到如：com.cn.me,要和me同级

然后自定义两个类

DshzsRandomRule类写自己定义的算法，DshzsRule写注入的bean

import com.netflix.loadbalancer.IRule;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
​
​
@Configuration
public class DshzsRule {
​
    @Bean
    public IRule myRule(){
        return new DshzsRandomRule();
    }
​
}

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;
import java.util.concurrent.ThreadLocalRandom;
​
​
public class DshzsRandomRule extends AbstractLoadBalancerRule {
​
​
    //定义次数
    private int total = 0;
​
    //定义当前的数量
    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;
            }
​
            //生成区间随机数
//            int index = chooseRandomInt(serverCount);
​
            //从可以用的服务中随机获取一个
//            server = upList.get(index);
​
            if (total<5){
                server = upList.get(currentIndex);
                total++;
            }else {
                total = 0;
                currentIndex++;
                if (currentIndex>=upList.size()){
                    currentIndex = 0;
                }
                server = upList.get(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
        
    }
}
​

定义完之后需要在启动类这里加一个注解才能找到你定义的ribbon

@RibbonClient(name = "服务名称", configuration = 你自定义的类（DshzsRule.class）)