自定义Ribbon负载均衡规则

方法一

总的来说一句话：继承IRule的实现类AbstractLoadBalancerRule ，实现后注册到bean

package com.springcloud;

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.Random;

/**
 * Created by Choisaaaa on 2018/7/9.
 * 自定义随机算法
 *
 * 一个服务实例server 用3次  在轮询选择下一个server
 */
public class MyRandomRule extends AbstractLoadBalancerRule {


    Random rand;

    public MyRandomRule() {
        rand = new Random();
    }

    /**
     * Randomly choose from all living servers
     */
    public Server choose(ILoadBalancer lb, Object key) {
    //重写该方法
    }

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

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

    }
}

 这里需要注意不能与启动类同一级或其子级包下，否则将不起作用

package com.mark.config;

import com.netflix.loadbalancer.IRule;
import com.springcloud.MyRandomRule;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * Created by Choisaaaa on 2018/7/9.
 * 自定义的轮询算法的配置类
 */
@Configuration
public class MyRuleConfig {

    @Bean //注入自定义的负载均衡的算法
    public IRule iRule(){
        return new MyRandomRule();
    }
}

启动类加注释 PROVIDERPRODUCT：要访问的服务，MyRuleConfig：自定义规则类；即指定服务遵循自定义规则类

@SpringBootApplication
@EnableEurekaClient
@EnableDiscoveryClient
//为服务PROVIDERPRODUCT指定自定义的负载均衡算法
@RibbonClient(name = "PROVIDERPRODUCT",configuration = MyRuleConfig.class)
public class ConsumerApplication {

代码参见：https://blog.csdn.net/ycd500756/article/details/80975410

方法二

1. 删除config中RestTemplate的Ribbon默认负载均衡注释@LoadBalanced

package com.atguigu.order.config;

import org.springframework.cloud.client.loadbalancer.LoadBalanced;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.client.RestTemplate;

@Configuration
public class ApplicationContextConfig
{
    @Bean
// 删除Ribbon默认负载均衡的注释
//    @LoadBalanced
    public RestTemplate getRestTemplate()
    {
        return new RestTemplate();
    }
}

2. 自定义接口及方法返回 ServiceInstance 

package com.atguigu.order.lb;

import org.springframework.cloud.client.ServiceInstance;

import java.util.List;

public interface LoadBalancer
{
// 仿照IRule的实现类,返回对象
    ServiceInstance instances(List<ServiceInstance> serviceInstances);
}

3. 实现该接口

package com.atguigu.order.lb;

import org.springframework.cloud.client.ServiceInstance;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.concurrent.atomic.AtomicInteger;

@Component
public class MyLB implements LoadBalancer
{

    private AtomicInteger atomicInteger = new AtomicInteger(0);

// 这里模仿轮训算法
    public final int getAndIncrement()
    {
        int current;
        int next;

        do {
            current = this.atomicInteger.get();
            next = current >= 2147483647 ? 0 : current + 1;
        }while(!this.atomicInteger.compareAndSet(current,next));
        System.out.println("*****第几次访问，次数next: "+next);
        return next;
    }

    //负载均衡算法：rest接口第几次请求数 % 服务器集群总数量 = 实际调用服务器位置下标  ，每次服务重启动后rest接口计数从1开始。
    @Override
    public ServiceInstance instances(List<ServiceInstance> serviceInstances)
    {
        int index = getAndIncrement() % serviceInstances.size();

        return serviceInstances.get(index);
    }
}

4. contrller测试接口

    @Resource
    private LoadBalancer loadBalancer;
    @Resource
    private DiscoveryClient discoveryClient;

    @GetMapping(value = "/consumer/payment/lb")
    public String getPaymentLB()
    {
// 获取服务对象集合信息(需要访问的项目在eureka中的信息)
        List<ServiceInstance> instances = discoveryClient.getInstances("CLOUD-PAYMENT-SERVICE");

        if(instances == null || instances.size() <= 0)
        {
            return null;
        }

        ServiceInstance serviceInstance = loadBalancer.instances(instances);
        URI uri = serviceInstance.getUri();

        return restTemplate.getForObject(uri+"/payment/lb",String.class);

    }