fix: 反代后IP获取错误

Author: gadfly3173

src/main/java/io/github/talelin/latticy/common/interceptor/RequestLogInterceptor.java

@@ -1,5 +1,6 @@
 package io.github.talelin.latticy.common.interceptor;
 
+import io.github.talelin.latticy.common.util.IPUtil;
 import lombok.extern.slf4j.Slf4j;
 import org.springframework.web.servlet.AsyncHandlerInterceptor;
 
@@ -27,7 +28,7 @@ public void afterCompletion(HttpServletRequest request, HttpServletResponse resp
         log.info("[{}] -> [{}] from: {} costs: {}ms",
                 request.getMethod(),
                 request.getServletPath(),
-                request.getRemoteAddr(),
+                IPUtil.getIPFromRequest(request),
                 System.currentTimeMillis() - startTime.get()
         );
     }

src/main/java/io/github/talelin/latticy/common/util/IPUtil.java

@@ -0,0 +1,48 @@
+package io.github.talelin.latticy.common.util;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.servlet.http.HttpServletRequest;
+
+/**
+ * IP工具类
+ */
+@Slf4j
+public class IPUtil {
+
+    private static final String[] IP_HEADER_CANDIDATES = {
+            "X-Forwarded-For",
+            "Proxy-Client-IP",
+            "WL-Proxy-Client-IP",
+            "HTTP_X_FORWARDED_FOR",
+            "HTTP_X_FORWARDED",
+            "HTTP_X_CLUSTER_CLIENT_IP",
+            "HTTP_CLIENT_IP",
+            "HTTP_FORWARDED_FOR",
+            "HTTP_FORWARDED",
+            "HTTP_VIA",
+            "REMOTE_ADDR"
+    };
+
+    public static String getIPFromRequest(HttpServletRequest request) {
+        if (request == null) {
+            if (null == RequestContextHolder.getRequestAttributes()) {
+                return null;
+            }
+            request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
+        }
+
+        for (String header : IP_HEADER_CANDIDATES) {
+            String ipList = request.getHeader(header);
+            if (ipList != null && ipList.length() != 0 && !"unknown".equalsIgnoreCase(ipList)) {
+                log.debug("ipList.split(\",\")[0]::{}", ipList.split(",")[0]);
+                return ipList.split(",")[0];
+            }
+        }
+
+        log.debug("request.getRemoteAddr()::{}", request.getRemoteAddr());
+        return request.getRemoteAddr();
+    }
+}

src/main/java/io/github/talelin/latticy/module/log/MDCAccessServletFilter.java

@@ -1,5 +1,6 @@
 package io.github.talelin.latticy.module.log;
 
+import io.github.talelin.latticy.common.util.IPUtil;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.slf4j.MDC;
@@ -55,14 +56,14 @@ public void putResponseMDC(ServletResponse servletResponse) {
      */
     public void putRequestMDC(ServletRequest servletRequest) throws IOException {
         MDC.put(MDCAccessConstant.REQUEST_REMOTE_HOST_MDC_KEY, servletRequest.getRemoteHost());
-        MDC.put(MDCAccessConstant.REQUEST_REMOTE_ADDR_MDC_KEY, servletRequest.getRemoteAddr());
-        MDC.put(MDCAccessConstant.REQUEST_PROTOCOL_MDC_KEY, servletRequest.getProtocol());
         MDC.put(MDCAccessConstant.REQUEST_PROTOCOL_MDC_KEY, servletRequest.getProtocol());
+        MDC.put(MDCAccessConstant.REQUEST_REMOTE_ADDR_MDC_KEY, servletRequest.getRemoteAddr());
         MDC.put(MDCAccessConstant.REQUEST_REMOTE_PORT_MDC_KEY, String.valueOf(servletRequest.getRemotePort()));
         MDC.put(MDCAccessConstant.REQUEST_BODY_BYTES_SENT_MDC_KEY, String.valueOf(servletRequest.getContentLength()));
 
         if (servletRequest instanceof HttpServletRequest) {
             HttpServletRequest httpServletRequest = (HttpServletRequest) servletRequest;
+            MDC.put(MDCAccessConstant.REQUEST_REMOTE_ADDR_MDC_KEY, IPUtil.getIPFromRequest(httpServletRequest));
             MDC.put(MDCAccessConstant.REQUEST_REQUEST_URI_MDC_KEY, httpServletRequest.getRequestURI());
             StringBuffer requestUrl = httpServletRequest.getRequestURL();
             if (requestUrl != null) {