ly
/
Stwzhj-Cloud-Plus
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!77 修改LoadBalancer模块，支持`多网卡` 

Merge pull request !77 from Yjoioooo/dev
2.X
疯狂的狮子Li 2023-03-31 01:28:17 +00:00 committed by Gitee
parent 0b810c0b8d 5e03a2c1f1
commit a1f1a4565b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
4 changed files with 9 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ruoyi-common/ruoyi-common-loadbalancer/pom.xml
View File

@ -31,5 +31,10 @@
<artifactId>lombok</artifactId>
</dependency>
<dependency>
<groupId>com.ruoyi</groupId>
<artifactId>ruoyi-common-core</artifactId>
</dependency>
</dependencies>
</project>

34
ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/constant/LoadBalancerConstant.java
View File

@ -1,34 +0,0 @@
package com.ruoyi.common.loadbalance.constant;
import lombok.extern.slf4j.Slf4j;
import java.net.InetAddress;
import java.net.UnknownHostException;
/**
* 常量类
*
* @author Lion Li
*/
@Slf4j
public class LoadBalancerConstant {
/**
* 获取服务host
* 默认自动获取
*/
public static String getHost() {
String host = "127.0.0.1";
try {
// 如需自定义ip可修改此处
String address = InetAddress.getLocalHost().getHostAddress();
if (address != null) {
host = address;
}
} catch (UnknownHostException e) {
e.printStackTrace();
}
log.debug("[LoadBalancer] - 本机IP地址: {}", host);
return host;
}
}

4
ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/core/CustomDubboLoadBalancer.java
View File

@ -1,6 +1,6 @@
package com.ruoyi.common.loadbalance.core;
import com.ruoyi.common.loadbalance.constant.LoadBalancerConstant;
import cn.hutool.core.net.NetUtil;
import lombok.extern.slf4j.Slf4j;
import org.apache.dubbo.common.URL;
import org.apache.dubbo.rpc.Invocation;
@ -21,7 +21,7 @@ public class CustomDubboLoadBalancer extends AbstractLoadBalance {
@Override
protected <T> Invoker<T> doSelect(List<Invoker<T>> invokers, URL url, Invocation invocation) {
for (Invoker<T> invoker : invokers) {
if (invoker.getUrl().getHost().equals(LoadBalancerConstant.getHost())) {
if (NetUtil.localIpv4s().contains(invoker.getUrl().getHost())) {
return invoker;
}
}

4
ruoyi-common/ruoyi-common-loadbalancer/src/main/java/com/ruoyi/common/loadbalance/core/CustomSpringCloudLoadBalancer.java
View File

@ -1,6 +1,6 @@
package com.ruoyi.common.loadbalance.core;
import com.ruoyi.common.loadbalance.constant.LoadBalancerConstant;
import cn.hutool.core.net.NetUtil;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.ObjectProvider;
@ -54,7 +54,7 @@ public class CustomSpringCloudLoadBalancer implements ReactorServiceInstanceLoad
return new EmptyResponse();
}
for (ServiceInstance instance : instances) {
if (instance.getHost().equals(LoadBalancerConstant.getHost())) {
if (NetUtil.localIpv4s().contains(instance.getHost())) {
return new DefaultResponse(instance);
}
}