更改 spring 云负载均衡器默认循环算法

如何解决更改 spring 云负载均衡器默认循环算法

我正在尝试从 Ribbon 迁移到 Spring Cloud LoadBalancer，因为 Ribbon 现在处于维护模式，我无法使用默认的 spring-boot 2.4.2 版本在 spring 初始化中添加功能区依赖项。我目前正在使用 OpenFeign，我正在做一些测试，试图将默认的 RoundRobin 负载均衡器配置更改为随机规则。这就是我所拥有的

使用 Ribbon 时，我只需要更改默认配置

@Configuration
public class RibbonConfiguration {

    @Bean
    public IRule loadBalancingRule() {
//        return new RoundRobinRule();
        return new Randomrule();
    }

}

并在客户端添加注解@RibbonClient（也可以在MainApplication类中添加）

@FeignClient("first-service")
@RibbonClient(name = "first-service",configuration = RibbonConfiguration.class)
public interface HelloClient {
    @GetMapping("/hello")
    String hello();
}

但我无法使用 spring-cloud-loadbalancer 使其工作。我知道默认包含在 spring-cloud-starter-openfeign 依赖项中，所以我的 pom 是这样的

pom.xml

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
    xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <parent>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-parent</artifactId>
        <version>2.4.2</version>
        <relativePath/> <!-- lookup parent from repository -->
    </parent>
    <groupId>com.demo</groupId>
    <artifactId>edge-service</artifactId>
    <version>0.0.1-SNAPSHOT</version>
    <name>edge-service</name>
    <description>Demo project for Spring Boot</description>
    <properties>
        <java.version>11</java.version>
        <spring-cloud.version>2020.0.1</spring-cloud.version>
    </properties>
    <dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-starter-netflix-eureka-client</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.cloud</groupId>
            <artifactId>spring-cloud-starter-openfeign</artifactId>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
    </dependencies>
    <dependencyManagement>
        <dependencies>
            <dependency>
                <groupId>org.springframework.cloud</groupId>
                <artifactId>spring-cloud-dependencies</artifactId>
                <version>${spring-cloud.version}</version>
                <type>pom</type>
                <scope>import</scope>
            </dependency>
        </dependencies>
    </dependencyManagement>

    <build>
        <plugins>
            <plugin>
                <groupId>org.springframework.boot</groupId>
                <artifactId>spring-boot-maven-plugin</artifactId>
            </plugin>
        </plugins>
    </build>

</project>

我的 FeignClients 是这样的（没有任何 Ribbon 注释）

@FeignClient("first-service")
public interface HelloClient {
    @GetMapping("/hello")
    String hello();
}

@FeignClient("second-service")
public interface WorldClient {
    @GetMapping("/world")
    String world();
}

我的 RestController 就像这样

@RestController
public class HelloWorldController {
    @Autowired
    private HelloClient helloClient;
    @Autowired
    private WorldClient worldClient;

    private Logger logger = LoggerFactory.getLogger(this.getClass());

    
    @GetMapping("/hello-world")
    public String helloWorld() {
        logger.info("Calling the clients");
        return helloClient.hello() + " - " + worldClient.world();
    }
}

没有做任何其他事情，它似乎可以进行负载平衡（两个 hello-service 实例仍在运行并正在调用备用），但我想将默认的负载平衡器算法（RoundRobin）更改为另一个（如随机或任何人）自定义）来测试差异。我想我必须更改 FeignConfiguration 类中的某些内容，但我搜索没有成功。

希望你能帮助我，这样我就可以摆脱 Ribbon 并开始使用 Spring Cloud LoadBalancer 没有问题！ :C

解决方法

我认为您只需要创建 ReactorServiceInstanceLoadBalancer 类型的 bean，它将被用来代替默认的 RoundRobin bean。

@Configuration
public class LoadBalancingConfiguration {

    @Bean
    public ReactorLoadBalancer<ServiceInstance> reactorServiceInstanceLoadBalancer(Environment environment,LoadBalancerClientFactory loadBalancerClientFactory) {
      String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
      return new RandomLoadBalancer(
          loadBalancerClientFactory.getLazyProvider(name,ServiceInstanceListSupplier.class),name);
    }

}

以上示例复制自 spring-cloud-loadbalancer 项目中 LoadBalancerClientConfiguration 类中的默认 bean 创建

@Bean
@ConditionalOnMissingBean
public ReactorLoadBalancer<ServiceInstance> reactorServiceInstanceLoadBalancer(Environment environment,LoadBalancerClientFactory loadBalancerClientFactory) {
    String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
    return new RoundRobinLoadBalancer(
            loadBalancerClientFactory.getLazyProvider(name,name);
}

我正在尝试弄清楚目前所有这些是如何工作的，以便我可以让负载平衡代码使用线模拟服务器进行我的测试，而无需发现服务器。但是，我没有找到太多有用的文档来解释如何从功能区迁移到弹簧负载平衡器。

以下内容对我有用，可以让我在测试中使用wiremock。假设我在本地运行wiremock服务器并监听端口 9999。

@Bean
public RoundRobinLoadBalancer loadBalancer() {
  DefaultServiceInstance wiremock = new DefaultServiceInstance("","","localhost",9999,false);
  return new RoundRobinLoadBalancer(ServiceInstanceListSuppliers.toProvider("",wiremock),"");
}

Ribbon 配置不适用于 Spring Cloud LoadBalancer。如果您添加 spring-cloud-starter-loadbalancer，则会为您创建默认的 RoundRobin bean，因此您无需进行设置。但是，如果您想切换到 Random 算法，则需要添加 bean 以覆盖配置中的默认值，如下所示：

public class CustomLoadBalancerConfiguration {

@Bean
ReactorLoadBalancer<ServiceInstance> randomLoadBalancer(Environment environment,LoadBalancerClientFactory loadBalancerClientFactory) {
    String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
    return new RandomLoadBalancer(loadBalancerClientFactory
            .getLazyProvider(name,name); 
  }
}

您可以阅读更多相关信息here。

在添加了新 bean 的 LoadBalancer 配置类后，您可以设置客户端以使用 @LoadBalancerClient 或 @LoadBalancerClients 批注使用它。您可以详细阅读如何操作in the docs。