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

Question

我正在尝试从Ribbon迁移到Spring，因为Ribbon现在处于维护模式，并且我不能使用默认的spring-boot 2.4.2版本在spring initialize中添加ribbon依赖项。我目前正在使用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像这样(没有任何功能区注释)

@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运行的实例仍在运行，并且正在调用alternate)，但我想将默认的负载均衡器算法(RoundRobin)更改为另一个(如随机或任何自定义的)来测试差异。我想我必须在FeignConfiguration类中修改一些东西，但是我已经搜索过了，但是没有成功。

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

Answer 1

Ribbon配置不适用于Spring Cloud LoadBalancer。如果您添加了spring-cloud-starter-loadbalancer，则会为您创建默认的RoundRobin bean，因此您不需要对其进行设置。但是，如果您希望切换到随机算法，则需要添加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配置类之后，可以将客户端设置为将其与@LoadBalancerClient或@LoadBalancerClients注释一起使用。您可以详细阅读如何操作in the docs。

Answer 2

我认为您只需要创建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, ServiceInstanceListSupplier.class), name);
}

我正在尝试弄清楚这一切是如何工作的，这样我就可以获得负载平衡代码，以便在不需要发现服务器的情况下使用wiremock服务器进行测试。但是，我没有找到太多有用的文档来解释如何从ribbon迁移到spring负载均衡器。

下面的代码可以让我在测试中使用wiremock。假设我在本地运行了wiremock服务器，并在端口9999上进行侦听。

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