Merge pull request #914 from weibocom/feature/getExportIpFormEnv

support setting export ip from env
weibocom · Sep 3, 2020 · 94c6a23 · 94c6a23
2 parents 453d5db + 8f662f0
commit 94c6a23
Showing 1 changed file with 36 additions and 18 deletions.
diff --git a/motan-core/src/main/java/com/weibo/api/motan/util/NetUtils.java b/motan-core/src/main/java/com/weibo/api/motan/util/NetUtils.java
@@ -16,20 +16,16 @@
 
 package com.weibo.api.motan.util;
 
-import java.net.InetAddress;
-import java.net.InetSocketAddress;
-import java.net.NetworkInterface;
-import java.net.Socket;
-import java.net.SocketAddress;
+import org.apache.commons.lang3.StringUtils;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.net.*;
 import java.util.Enumeration;
 import java.util.Map;
 import java.util.regex.Pattern;
 
-import org.slf4j.Logger;
-import org.slf4j.LoggerFactory;
-
 /**
- * 
  * 网络工具类
  *
  * @author fishermen
@@ -44,6 +40,8 @@ public class NetUtils {
 
     public static final String ANYHOST = "0.0.0.0";
 
+    public static final String MOTAN_IP_PREFIX = "MOTAN_IP_PREFIX";
+
     private static volatile InetAddress LOCAL_ADDRESS = null;
 
     private static final Pattern LOCAL_IP_PATTERN = Pattern.compile("127(\\.\\d{1,3}){3}$");
@@ -63,7 +61,7 @@ public static boolean isValidLocalHost(String host) {
 
     /**
      * {@link #getLocalAddress(Map)}
-     * 
+     *
      * @return
      */
     public static InetAddress getLocalAddress() {
@@ -72,23 +70,35 @@ public static InetAddress getLocalAddress() {
 
     /**
      * <pre>
-     * 查找策略：首先看是否已经查到ip --> hostname对应的ip --> 根据连接目标端口得到的本地ip --> 轮询网卡
+     * 查找策略：首先看是否已经查到ip --> 环境变量中指定的ip --> hostname对应的ip --> 根据连接目标端口得到的本地ip --> 轮询网卡
      * </pre>
-     * 
+     *
      * @return loca ip
      */
     public static InetAddress getLocalAddress(Map<String, Integer> destHostPorts) {
         if (LOCAL_ADDRESS != null) {
             return LOCAL_ADDRESS;
         }
+        InetAddress localAddress = null;
+        String ipPrefix = System.getenv(MOTAN_IP_PREFIX);
+        if (StringUtils.isNotBlank(ipPrefix)) { // 环境变量中如果指定了motan使用的ip前缀，则使用与该前缀匹配的网卡ip作为本机ip。
+            localAddress = getLocalAddressByNetworkInterface(ipPrefix);
+            LoggerUtil.info("get local address by ip prefix: " + ipPrefix + ", address:" + localAddress);
+        }
+
+        if (!isValidAddress(localAddress)) {
+            localAddress = getLocalAddressByHostname();
+            LoggerUtil.info("get local address by hostname, address:" + localAddress);
+        }
 
-        InetAddress localAddress = getLocalAddressByHostname();
         if (!isValidAddress(localAddress)) {
             localAddress = getLocalAddressBySocket(destHostPorts);
+            LoggerUtil.info("get local address by remote host. address:" + localAddress);
         }
 
         if (!isValidAddress(localAddress)) {
-            localAddress = getLocalAddressByNetworkInterface();
+            localAddress = getLocalAddressByNetworkInterface(null);
+            LoggerUtil.info("get local address from network interface. address:" + localAddress);
         }
 
         if (isValidAddress(localAddress)) {
@@ -123,11 +133,13 @@ private static InetAddress getLocalAddressBySocket(Map<String, Integer> destHost
                 try {
                     SocketAddress addr = new InetSocketAddress(host, port);
                     socket.connect(addr, 1000);
+                    LoggerUtil.info("get local address from socket. remote host:" + host + ", port:" + port);
                     return socket.getLocalAddress();
                 } finally {
                     try {
                         socket.close();
-                    } catch (Throwable e) {}
+                    } catch (Throwable e) {
+                    }
                 }
             } catch (Exception e) {
                 LoggerUtil.warn(String.format("Failed to retriving local address by connecting to dest host:port(%s:%s) false, e=%s", host,
@@ -137,7 +149,7 @@ private static InetAddress getLocalAddressBySocket(Map<String, Integer> destHost
         return null;
     }
 
-    private static InetAddress getLocalAddressByNetworkInterface() {
+    private static InetAddress getLocalAddressByNetworkInterface(String prefix) {
         try {
             Enumeration<NetworkInterface> interfaces = NetworkInterface.getNetworkInterfaces();
             if (interfaces != null) {
@@ -149,7 +161,12 @@ private static InetAddress getLocalAddressByNetworkInterface() {
                             try {
                                 InetAddress address = addresses.nextElement();
                                 if (isValidAddress(address)) {
-                                    return address;
+                                    if (StringUtils.isBlank(prefix)) {
+                                        return address;
+                                    }
+                                    if (address.getHostAddress().startsWith(prefix)) {
+                                        return address;
+                                    }
                                 }
                             } catch (Throwable e) {
                                 logger.warn("Failed to retriving ip address, " + e.getMessage(), e);
@@ -175,6 +192,7 @@ public static boolean isValidAddress(InetAddress address) {
         String name = address.getHostAddress();
         return (name != null && !ANYHOST.equals(name) && !LOCALHOST.equals(name) && IP_PATTERN.matcher(name).matches());
     }
+
     //return ip to avoid lookup dns
     public static String getHostName(SocketAddress socketAddress) {
         if (socketAddress == null) {
@@ -183,7 +201,7 @@ public static String getHostName(SocketAddress socketAddress) {
 
         if (socketAddress instanceof InetSocketAddress) {
             InetAddress addr = ((InetSocketAddress) socketAddress).getAddress();
-            if(addr != null){
+            if (addr != null) {
                 return addr.getHostAddress();
             }
         }