diff --git a/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomEnvironmentPostProcessor.java b/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomEnvironmentPostProcessor.java new file mode 100644 index 00000000..37fe7b30 --- /dev/null +++ b/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomEnvironmentPostProcessor.java @@ -0,0 +1,25 @@ +package com.ruoyi.common.loadbalance.config; + +import org.springframework.boot.SpringApplication; +import org.springframework.boot.env.EnvironmentPostProcessor; +import org.springframework.core.Ordered; +import org.springframework.core.env.ConfigurableEnvironment; + +/** + * dubbo自定义负载均衡配置注入 + * + * @author Lion Li + */ +public class CustomEnvironmentPostProcessor implements EnvironmentPostProcessor, Ordered { + + @Override + public void postProcessEnvironment(ConfigurableEnvironment environment, SpringApplication application) { + System.setProperty("dubbo.consumer.loadbalance", "customDubboLoadBalancer"); + } + + @Override + public int getOrder() { + return Ordered.HIGHEST_PRECEDENCE; + } + +} diff --git a/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomLoadBalanceAutoConfiguration.java b/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomLoadBalanceAutoConfiguration.java index 9ad18b04..a134f136 100644 --- a/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomLoadBalanceAutoConfiguration.java +++ b/ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/config/CustomLoadBalanceAutoConfiguration.java @@ -1,14 +1,12 @@ package com.ruoyi.common.loadbalance.config; import org.springframework.cloud.loadbalancer.annotation.LoadBalancerClients; -import org.springframework.context.annotation.PropertySource; /** * 自定义负载均衡自动配置 * * @author Lion Li */ -@PropertySource(value = "classpath:loadbalance.properties", encoding = "UTF-8") @LoadBalancerClients(defaultConfiguration = CustomLoadBalanceClientConfiguration.class) public class CustomLoadBalanceAutoConfiguration { diff --git a/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/META-INF/spring.factories b/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/META-INF/spring.factories index 2146835e..50137508 100644 --- a/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/META-INF/spring.factories +++ b/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/META-INF/spring.factories @@ -1,2 +1,4 @@ org.springframework.boot.autoconfigure.EnableAutoConfiguration=\ com.ruoyi.common.loadbalance.config.CustomLoadBalanceAutoConfiguration +org.springframework.boot.env.EnvironmentPostProcessor=\ + com.ruoyi.common.loadbalance.config.CustomEnvironmentPostProcessor diff --git a/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/loadbalance.properties b/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/loadbalance.properties deleted file mode 100644 index 4e11f9e6..00000000 --- a/ruoyi-common/ruoyi-common-loadbalancer/src/main/resources/loadbalance.properties +++ /dev/null @@ -1 +0,0 @@ -dubbo.consumer.loadbalance=customDubboLoadBalancer