From a6dfc2fceab5de513608abaeb697b56e9d67b79a Mon Sep 17 00:00:00 2001
From: wenshao <szujobs@hotmail.com>
Date: Thu, 18 Jul 2013 09:38:44 +0800
Subject: [PATCH] WebStatFilter support realIpHeader

---
 .../druid/support/http/WebStatFilter.java     | 26 ++++++++++++++++++-
 1 file changed, 25 insertions(+), 1 deletion(-)

diff --git a/src/main/java/com/alibaba/druid/support/http/WebStatFilter.java b/src/main/java/com/alibaba/druid/support/http/WebStatFilter.java
index 4dea2225da..b77c1725e9 100644
--- a/src/main/java/com/alibaba/druid/support/http/WebStatFilter.java
+++ b/src/main/java/com/alibaba/druid/support/http/WebStatFilter.java
@@ -65,6 +65,7 @@ public class WebStatFilter implements Filter {
     public static final String           PARAM_NAME_EXCLUSIONS             = "exclusions";
     public static final String           PARAM_NAME_PRINCIPAL_SESSION_NAME = "principalSessionName";
     public static final String           PARAM_NAME_PRINCIPAL_COOKIE_NAME  = "principalCookieName";
+    public static final String           PARAM_NAME_REAL_IP_HEADER         = "realIpHeader";
 
     public final static int              DEFAULT_MAX_STAT_SESSION_COUNT    = 1000 * 100;
 
@@ -86,6 +87,7 @@ public class WebStatFilter implements Filter {
 
     private String                       principalSessionName;
     private String                       principalCookieName;
+    private String                       realIpHeader;
 
     @Override
     public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException,
@@ -232,7 +234,7 @@ public class WebStatFilter implements Filter {
 
             sessionStat.setUserAgent(userAgent);
 
-            String ip = DruidWebUtils.getRemoteAddr(request);
+            String ip = getRemoteAddress(request);
 
             sessionStat.addRemoteAddress(ip);
         }
@@ -240,6 +242,17 @@ public class WebStatFilter implements Filter {
         return sessionStat;
     }
 
+    protected String getRemoteAddress(HttpServletRequest request) {
+        String ip = null;
+        if (this.realIpHeader != null && this.realIpHeader.length() != 0) {
+            ip = request.getHeader(realIpHeader);
+        }
+        if (ip == null || ip.length() == 0) {
+            ip = DruidWebUtils.getRemoteAddr(request);
+        }
+        return ip;
+    }
+
     public String getSessionId(HttpServletRequest httpRequest) {
         String sessionId = null;
 
@@ -379,6 +392,17 @@ public class WebStatFilter implements Filter {
             }
         }
 
+        // realIpHeader
+        {
+            String param = config.getInitParameter(PARAM_NAME_REAL_IP_HEADER);
+            if (param != null) {
+                param = param.trim();
+                if (param.length() != 0) {
+                    this.realIpHeader = param;
+                }
+            }
+        }
+
         StatFilterContext.getInstance().addContextListener(statFilterContextListener);
 
         this.contextPath = DruidWebUtils.getContextPath(config.getServletContext());