【SpringCloud】Ribbon定制化配置

使用Ribbon自带负载均衡算法

添加负载均衡算法Configuration

//注意package位置假设启动器为@ComponentScan("com.test.springcloud")
package com.test.config
import com.netflix.loadbalancer.IRule;
import com.netflix.loadbalancer.RandomRule;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class MySelfRule
{
    @Bean
    public IRule myRule()
    {
        return new RandomRule();//定义为随机
    }
}

注意：
官方文档明确给出了警告：
这个自定义配置类不能放在@ComponentScan所扫描的当前包下以及子包下，
否则我们自定义的这个配置类就会被所有的Ribbon客户端所共享，达不到特殊化定制的目的了。

RestTemplate使用上面负载均衡算法

package com.test.springcloud.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
    @LoadBalanced //添加负载均衡配置注解，如果自定义的话不需要使用该注解否则报错
    public RestTemplate getRestTemplate()
    {
        return new RestTemplate();
    }
}

自定义负载均衡算法

负载均衡算法实现

定义负载均衡算法接口

public interface LoadBalancer
{
    ServiceInstance instances(List<ServiceInstance> serviceInstances);
}

实现负载均衡算法

@Component
public class MyLB implements LoadBalancer
{
	//通过CAS（乐观锁）的方式进行，本质是通过CompareAndSet，先比较后赋值的方式，只有当前值和期望值相同时才可以进行赋值
    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);
    }
}

以上实现负载均衡算法的instance可自定义

RestTemplate

package com.test.springcloud.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
    //@LoadBalanced //去掉这个注解
    public RestTemplate getRestTemplate()
    {
        return new RestTemplate();
    }
}

在Controller中使用该负载均衡算法

public class Controller{
	@GetMapping(value = "/consumer/payment/lb")
    public String getPaymentLB()
    {
        List<ServiceInstance> instances = discoveryClient.getInstances("CLOUD-PAYMENT-SERVICE");
        if(instances == null || instances.size() <= 0)
        {
            return null;
        }
        // 在下面有解释ServiceInstance
        ServiceInstance serviceInstance = loadBalancer.instances(instances);
        URI uri = serviceInstance.getUri();
        return restTemplate.getForObject(uri+"/payment/lb",String.class);

    }

}

ServiceIInstance解释

ServiceInstance 是 Spring Cloud 提供的一个接口，用于表示服务的实例信息。它提供了获取服务实例的相关信息和元数据的方法。
ServiceInstance 接口定义了以下常用方法：

• String getServiceId(): 获取服务名称（serviceId）。
• String getHost(): 获取服务实例的主机名。
• int getPort(): 获取服务实例的端口号。
• boolean isSecure(): 返回服务实例是否使用安全协议（如 HTTPS）进行通信。
• URI getUri(): 获取服务实例的完整 URI。
• Map getMetadata(): 返回服务实例的元数据，这些元数据可以由服务提供方自定义。
• 通过使用 DiscoveryClient（服务发现客户端）获取到的服务实例列表，可以进一步使用 ServiceInstance 接口来获取每个服务实例的具体信息。
  示例：

@Autowired
private DiscoveryClient discoveryClient;

public void getServiceInstances(String serviceName) {
    List<ServiceInstance> instances = discoveryClient.getInstances(serviceName);
    for (ServiceInstance instance : instances) {
        String serviceId = instance.getServiceId();
        String host = instance.getHost();
        int port = instance.getPort();
        boolean isSecure = instance.isSecure();
        URI uri = instance.getUri();
        Map<String, String> metadata = instance.getMetadata();
        
        // 打印服务实例信息
        System.out.println("Service ID: " + serviceId);
        System.out.println("Host: " + host);
        System.out.println("Port: " + port);
        System.out.println("Is Secure: " + isSecure);
        System.out.println("URI: " + uri);
        System.out.println("Metadata: " + metadata);
    }
}