Alawn/Druid
1
0
Fork 0
mirror of https://github.com/alibaba/druid.git synced 2023-03-04 19:17:16 +00:00
Code Issues Releases Wiki Activity

WebStatFilter support realIpHeader

Browse Source
This commit is contained in:
wenshao
2013-07-18 09:38:44 +08:00
parent 34d07a9c26
commit a6dfc2fcea
1 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/main/java/com/alibaba/druid/support/http/WebStatFilter.java
+25 -1
View File
@@ -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());