/* * Copyright (c) 2018-2028, Chill Zhuang All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright * notice, this list of conditions and the following disclaimer in the * documentation and/or other materials provided with the distribution. * Neither the name of the dreamlu.net developer nor the names of its * contributors may be used to endorse or promote products derived from * this software without specific prior written permission. * Author: Chill 庄骞 (smallchill@163.com) */ package org.springblade.core.loadbalancer.config; import org.springblade.core.loadbalancer.props.BladeLoadBalancerProperties; import org.springblade.core.loadbalancer.rule.GrayscaleLoadBalancer; import org.springframework.boot.autoconfigure.AutoConfiguration; import org.springframework.boot.autoconfigure.condition.ConditionalOnProperty; import org.springframework.boot.context.properties.EnableConfigurationProperties; import org.springframework.cloud.client.ServiceInstance; import org.springframework.cloud.loadbalancer.annotation.LoadBalancerClientConfiguration; import org.springframework.cloud.loadbalancer.annotation.LoadBalancerClientSpecification; import org.springframework.cloud.loadbalancer.core.ReactorLoadBalancer; import org.springframework.cloud.loadbalancer.core.ServiceInstanceListSupplier; import org.springframework.cloud.loadbalancer.support.LoadBalancerClientFactory; import org.springframework.context.annotation.Bean; import org.springframework.core.annotation.Order; import org.springframework.core.env.Environment; /** * blade 负载均衡策略 * * @author Chill */ @AutoConfiguration(before = LoadBalancerClientConfiguration.class) @EnableConfigurationProperties(BladeLoadBalancerProperties.class) @ConditionalOnProperty(value = BladeLoadBalancerProperties.PROPERTIES_PREFIX + ".enabled", matchIfMissing = true) @Order(BladeLoadBalancerConfiguration.REACTIVE_SERVICE_INSTANCE_SUPPLIER_ORDER) public class BladeLoadBalancerConfiguration { public static final int REACTIVE_SERVICE_INSTANCE_SUPPLIER_ORDER = 193827465; @Bean public ReactorLoadBalancer reactorServiceInstanceLoadBalancer(Environment environment, LoadBalancerClientFactory loadBalancerClientFactory, BladeLoadBalancerProperties bladeLoadBalancerProperties) { String name = environment.getProperty(LoadBalancerClientFactory.PROPERTY_NAME); return new GrayscaleLoadBalancer(loadBalancerClientFactory.getLazyProvider(name, ServiceInstanceListSupplier.class), bladeLoadBalancerProperties); } @Bean public LoadBalancerClientSpecification loadBalancerClientSpecification() { return new LoadBalancerClientSpecification("default.bladeLoadBalancerConfiguration", new Class[]{BladeLoadBalancerConfiguration.class}); } }