From 4e3f8104c44b6ff5388913997c758bf0ff588122 Mon Sep 17 00:00:00 2001
From: nili <nili@baicizhan.com>
Date: Thu, 14 Mar 2024 21:01:59 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8A=A0=E4=B8=8Bip?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../group/game/service/MatrixService.java     |  4 +-
 .../java/awesome/group/RequestContext.java    | 40 +++++++++++++++++++
 .../group/controller/MatrixController.java    |  3 +-
 3 files changed, 44 insertions(+), 3 deletions(-)

diff --git a/game-service/src/main/java/awesome/group/game/service/MatrixService.java b/game-service/src/main/java/awesome/group/game/service/MatrixService.java
index ef513cb..997cd9a 100644
--- a/game-service/src/main/java/awesome/group/game/service/MatrixService.java
+++ b/game-service/src/main/java/awesome/group/game/service/MatrixService.java
@@ -14,7 +14,7 @@ public class MatrixService {
     @Resource
     private MatrixAdvRecordMapper mapper;
 
-    public void saveRecord(MatrixAdvRecordBo bo) {
+    public void saveRecord(MatrixAdvRecordBo bo, String ip) {
         Assert.isTrue(StringUtils.hasText(bo.appCode), "appCode不能为空");
         Integer appId = mapper.queryAppId(bo.appCode);
         if (appId == null) {
@@ -28,7 +28,7 @@ public class MatrixService {
         record.setEcpm(bo.ecpm);
         record.setDeviceBrand(bo.deviceBrand);
         record.setDeviceName(bo.deviceName);
-        record.setIp(bo.ip);
+        record.setIp(ip);
         mapper.insert(record);
     }
 }
diff --git a/game-web/src/main/java/awesome/group/RequestContext.java b/game-web/src/main/java/awesome/group/RequestContext.java
index 175306d..aaf6429 100644
--- a/game-web/src/main/java/awesome/group/RequestContext.java
+++ b/game-web/src/main/java/awesome/group/RequestContext.java
@@ -3,6 +3,8 @@ package awesome.group;
 import jakarta.servlet.http.HttpServletRequest;
 import jakarta.servlet.http.HttpServletResponse;
 
+import java.net.InetAddress;
+import java.net.UnknownHostException;
 import java.util.HashMap;
 import java.util.Map;
 
@@ -47,4 +49,42 @@ public class RequestContext {
         THREAD_LOCAL.remove();
     }
 
+    public static String getIpAddr(HttpServletRequest request) {
+        String ipAddress = null;
+        try {
+            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();
+                if (ipAddress.equals("127.0.0.1")) {
+                    // 根据网卡取本机配置的IP
+                    InetAddress inet = null;
+                    try {
+                        inet = InetAddress.getLocalHost();
+                    } catch (UnknownHostException e) {
+                        e.printStackTrace();
+                    }
+                    ipAddress = inet.getHostAddress();
+                }
+            }
+            // 对于通过多个代理的情况，第一个IP为客户端真实IP,多个IP按照','分割
+            if (ipAddress != null && ipAddress.length() > 15) { // "***.***.***.***".length()
+                // = 15
+                if (ipAddress.indexOf(",") > 0) {
+                    ipAddress = ipAddress.substring(0, ipAddress.indexOf(","));
+                }
+            }
+        } catch (Exception e) {
+            ipAddress="";
+        }
+        // ipAddress = this.getRequest().getRemoteAddr();
+
+        return ipAddress;
+    }
+
 }
diff --git a/game-web/src/main/java/awesome/group/controller/MatrixController.java b/game-web/src/main/java/awesome/group/controller/MatrixController.java
index 4db863a..431a65a 100644
--- a/game-web/src/main/java/awesome/group/controller/MatrixController.java
+++ b/game-web/src/main/java/awesome/group/controller/MatrixController.java
@@ -1,5 +1,6 @@
 package awesome.group.controller;
 
+import awesome.group.RequestContext;
 import awesome.group.aop.RestApi;
 import awesome.group.game.service.MatrixService;
 import awesome.group.game.service.bo.MatrixAdvRecordBo;
@@ -19,7 +20,7 @@ public class MatrixController {
     @PostMapping("/saveAdvRecord")
     @RestApi
     public R<Void> saveAdvRecord(@RequestBody MatrixAdvRecordBo bo) {
-        matrixService.saveRecord(bo);
+        matrixService.saveRecord(bo, RequestContext.getIpAddr(RequestContext.getRequest()));
         return new R<>(R.CODE_SUCCESS, "ok", null);
     }
 }