SpringCloud LoadBalancer自定義負載均衡器使用解析

更新時間：2023年04月17日 08:34:41 作者：程序員李哈

LoadBalancerClient 是 SpringCloud 提供的一種負載均衡客戶端，Ribbon 負載均衡組件內部也是集成了 LoadBalancerClient 來實現負載均衡，本文給大家深入解析 LoadBalancerClient 接口源碼，感興趣的朋友跟隨小編一起看看吧

由于原有的負載均衡組件Ribbon停止維護，而完美的Spring生態(tài)怎能允許缺少負載均衡組件呢？Spring Cloud官方自己造出了Spring Cloud LoadBalancer來代替原有的Ribbon。由于是官方自己寫的組件，所以并沒有像eureka、Feign那樣抽出一個單獨的組件包出來。而放入到Spring Cloud Commons規(guī)范包中。

正文

Spring Cloud LoadBalancer 幫開發(fā)者已經實現了RoundRobinLoadBalancer、RandomLoadBalancer，分別是輪訓和隨機，默認實現為輪訓。

即讓是負載均衡組件，那必然有默認實現，也必然有擴展接口暴露給開發(fā)者。所以第一步肯定是介紹抽象接口。

// 負載均衡的標志性接口，繼承ReactorLoadBalancer接口
// 本接口無任何接口增強，僅僅作為一個標志性接口。
public interface ReactorServiceInstanceLoadBalancer extends ReactorLoadBalancer<ServiceInstance> 
// 在Spring Cloud高版本中大部分組件采用了Reacotr框架（有一說一，閱讀性極差?。。?
// 這里定義了一個選擇的抽象方法。
public interface ReactorLoadBalancer<T> extends ReactiveLoadBalancer<T> {
	Mono<Response<T>> choose(Request request);
	default Mono<Response<T>> choose() {
		return choose(REQUEST);
	}
}

這里有涉及到Reactor框架，不過完全可以當黑盒使用。接口非常的簡單就一個choose方法，所以接下來我們只需要實現此接口。

@LoadBalancerClient(value = "deptmanagecloud-provider", configuration = MyConfig.class)
public class MyConfig {
    // 配置負載均衡策略
    @Bean
    public ReactorLoadBalancer<ServiceInstance> myLB(Environment environment, LoadBalancerClientFactory loadBalancerClientFactory) {
        String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
        return new MyLB(
                loadBalancerClientFactory.getLazyProvider(name, ServiceInstanceListSupplier.class));
    }
}

這里有部分邏輯，讀者完全可以當作黑盒暫時不需要去理解。只需要明白從注冊中心拿到注冊表以后會執(zhí)行getInstanceResponse方法，只需要實現自定義負載均衡策略的邏輯即可，好比我這里的邏輯是永遠只調用一個服務（當然，這完全扯淡，僅僅是Demo無須太關心）

既然我們把自定義負載均衡策略寫好了，那么，怎么告訴Spring呢？肯定需要注入給Spring。

@LoadBalancerClient(value = "deptmanagecloud-provider", configuration = MyConfig.class)
public class MyConfig {
    // 配置負載均衡策略
    @Bean
    public ReactorLoadBalancer<ServiceInstance> myLB(Environment environment, LoadBalancerClientFactory loadBalancerClientFactory) {
        String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME);
        return new MyLB(
                loadBalancerClientFactory.getLazyProvider(name, ServiceInstanceListSupplier.class));
    }
}

這里需要?上@LoadBalancerClient注解，value為調用方的服務名，configuration為配置類