fix 获取用户ip

midjourney/src/main/java/com/yhlxj/util/IpUtils.java

@@ -0,0 +1,55 @@
+package com.yhlxj.util;
+
+import org.apache.commons.lang3.StringUtils;
+import org.springframework.util.ResourceUtils;
+
+import javax.servlet.http.HttpServletRequest;
+import java.io.File;
+import java.io.IOException;
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+
+/**
+ * @author chan
+ * @className IpUtils
+ * @description ip工具类
+ * @date 2020/11/12 1:39 下午
+ */
+public class IpUtils {
+
+    /**
+     * @Description：获取客户端的IP
+     * @author zrt
+     * @date 2018年9月22日 上午10:39:44
+     */
+    public static String getIpAddr(HttpServletRequest request) {
+        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();
+            if (ipAddress.equals("127.0.0.1") || ipAddress.equals("0:0:0:0:0: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(","));
+            }
+        }
+        return ipAddress;
+    }
+
+}

midjourney/src/main/java/com/yhlxj/web/mirror/MidjourneyController.java

@@ -29,6 +29,7 @@ import com.yhlxj.dao.model.views.MidjourneyUserPaintingRecordView;
 import com.yhlxj.redis.RedisService;
 import com.yhlxj.service.midjourney.MidjourneyService;
 import com.yhlxj.service.midjourney.MidjourneyUserSettingsService;
+import com.yhlxj.util.IpUtils;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
 import org.apache.commons.lang3.StringUtils;
@@ -84,7 +85,7 @@ public class MidjourneyController {
 
     public MidjourneyUser getUser() throws IOException {
         String userToken = request.getHeader("user-token");
-        String ip = ServletUtil.getClientIP(request);
+        String ip = IpUtils.getIpAddr(request);
         log.info("用户请求 ip:{}, userToken:{}", ip, userToken);
         if (StringUtils.isBlank(userToken)) {
             response.sendRedirect("https://nf.video");