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

Change spring cloud loadbalancer default RoundRobin Algorithm

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

使用功能区时,我只需更改默认配置

@Configuration
public class RibbonConfiguration {

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

}

并在客户端添加注解@RibbonClient(也可以在MainApplication中添加注解class)

@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 像这样(没有任何功能区注释)

@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 class 中的某些内容,但我没有成功搜索。

希望你能帮助我,这样我就可以摆脱 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 class 中的默认 bean 创建复制而来的
@Bean
@ConditionalOnMissingBean
public ReactorLoadBalancer<ServiceInstance> reactorServiceInstanceLoadBalancer(Environment environment,
        LoadBalancerClientFactory loadBalancerClientFactory) {
    String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
    return new RoundRobinLoadBalancer(
            loadBalancerClientFactory.getLazyProvider(name, ServiceInstanceListSupplier.class), name);
}

我目前正在尝试弄清楚这一切是如何工作的,这样我就可以获得负载平衡代码,以便在我的测试中使用 wiremock 服务器,而无需发现服务器。但是,我没有找到很多有用的文档来解释如何从功能区迁移到 spring 负载平衡器。

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

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

功能区配置不适用于 Spring Cloud LoadBalancer。如果你添加 spring-cloud-starter-loadbalancer,默认的 RoundRobin bean 会为你创建,所以你不需要设置它。但是,如果你想切换到随机算法,你需要添加 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, ServiceInstanceListSupplier.class),
            name); 
  }
}

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

在添加了新 bean 的 LoadBalancer 配置 class 之后,您可以将客户端设置为使用 @LoadBalancerClient@LoadBalancerClients 注释。您可以详细阅读操作方法 in the docs.