From a25181b16d36a73be223588aa2f0db0c097cc1c1 Mon Sep 17 00:00:00 2001
From: macro <zhenghong1991@126.com>
Date: Thu, 8 Oct 2020 15:04:13 +0800
Subject: [PATCH] =?UTF-8?q?=E8=8E=B7=E5=8F=96=E8=AE=BF=E9=97=AE=E5=9C=B0?=
 =?UTF-8?q?=E5=9D=80=E6=96=B9=E5=BC=8F=E4=BF=AE=E6=94=B9?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/UmsAdminServiceImpl.java     |  3 +-
 .../macro/mall/common/util/RequestUtil.java   | 47 +++++++++++++++++++
 2 files changed, 49 insertions(+), 1 deletion(-)
 create mode 100644 mall-common/src/main/java/com/macro/mall/common/util/RequestUtil.java

diff --git a/mall-admin/src/main/java/com/macro/mall/service/impl/UmsAdminServiceImpl.java b/mall-admin/src/main/java/com/macro/mall/service/impl/UmsAdminServiceImpl.java
index 37b9ae6..6d1193b 100644
--- a/mall-admin/src/main/java/com/macro/mall/service/impl/UmsAdminServiceImpl.java
+++ b/mall-admin/src/main/java/com/macro/mall/service/impl/UmsAdminServiceImpl.java
@@ -5,6 +5,7 @@ import cn.hutool.core.util.StrUtil;
 import com.github.pagehelper.PageHelper;
 import com.macro.mall.bo.AdminUserDetails;
 import com.macro.mall.common.exception.Asserts;
+import com.macro.mall.common.util.RequestUtil;
 import com.macro.mall.dao.UmsAdminRoleRelationDao;
 import com.macro.mall.dto.UmsAdminParam;
 import com.macro.mall.dto.UpdateAdminPasswordParam;
@@ -128,7 +129,7 @@ public class UmsAdminServiceImpl implements UmsAdminService {
         loginLog.setCreateTime(new Date());
         ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
         HttpServletRequest request = attributes.getRequest();
-        loginLog.setIp(request.getRemoteAddr());
+        loginLog.setIp(RequestUtil.getRequestIp(request));
         loginLogMapper.insert(loginLog);
     }
 
diff --git a/mall-common/src/main/java/com/macro/mall/common/util/RequestUtil.java b/mall-common/src/main/java/com/macro/mall/common/util/RequestUtil.java
new file mode 100644
index 0000000..730e131
--- /dev/null
+++ b/mall-common/src/main/java/com/macro/mall/common/util/RequestUtil.java
@@ -0,0 +1,47 @@
+package com.macro.mall.common.util;
+
+import javax.servlet.http.HttpServletRequest;
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+
+/**
+ * 请求工具类
+ * Created by macro on 2020/10/8.
+ */
+public class RequestUtil {
+
+    /**
+     * 获取请求真实IP地址
+     */
+    public static String getRequestIp(HttpServletRequest request) {
+        //通过HTTP代理服务器转发时添加
+        String ipAddress = request.getHeader("x-forwarded-for");
+        if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
+            ipAddress = request.getHeader("Proxy-Client-IP");
+        }
+        if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
+            ipAddress = request.getHeader("WL-Proxy-Client-IP");
+        }
+        if (ipAddress == null || ipAddress.length() == 0 || "unknown".equalsIgnoreCase(ipAddress)) {
+            ipAddress = request.getRemoteAddr();
+            // 从本地访问时根据网卡取本机配置的IP
+            if (ipAddress.equals("127.0.0.1") || ipAddress.equals("0:0:0:0:0:0:0:1")) {
+                InetAddress inetAddress = null;
+                try {
+                    inetAddress = InetAddress.getLocalHost();
+                } catch (UnknownHostException e) {
+                    e.printStackTrace();
+                }
+                ipAddress = inetAddress.getHostAddress();
+            }
+        }
+        // 通过多个代理转发的情况，第一个IP为客户端真实IP，多个IP会按照','分割
+        if (ipAddress != null && ipAddress.length() > 15) {
+            if (ipAddress.indexOf(",") > 0) {
+                ipAddress = ipAddress.substring(0, ipAddress.indexOf(","));
+            }
+        }
+        return ipAddress;
+    }
+
+}