LoggerMDCFilter.java

package com.fpcms.common.web.filter;

import java.io.IOException;
import java.util.Map;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang.StringUtils;
import org.apache.log4j.MDC;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.web.filter.OncePerRequestFilter;

import com.github.rapid.common.util.LogTraceUtil;
import com.github.rapid.common.util.Profiler;
import com.fpcms.common.util.Constants;
import com.fpcms.common.util.IpUtil;
import com.fpcms.common.util.SpiderUtil;

/**
 * 存放在MDC中的数据，log4j可以直接引用并作为日志信息打印出来.
 * 
 * <pre>
 * 示例使用:
 * log4j.appender.stdout.layout.conversionPattern=%d [%X{loginUserId}/%X{req.remoteAddr}/%X{traceId} - %X{req.requestURI}?%X{req.queryString}] %-5p %c{2} - %m%n
 * </pre>
 * @author badqiu
 */
public class LoggerMDCFilter extends OncePerRequestFilter implements Filter{
	private static Logger log = LoggerFactory.getLogger(LoggerMDCFilter.class);
    protected void doFilterInternal(HttpServletRequest request,HttpServletResponse response, FilterChain chain)throws ServletException,IOException {
        try {
            //示例为一个固定的登陆用户,请直接修改代码
        	if(SpiderUtil.isSpider(request)) {
        		MDC.put("loginUserId",  SpiderUtil.getSpiderName(request.getHeader("User-Agent")));
        	}
            
            MDC.put("req.requestURI", StringUtils.defaultString(request.getRequestURL().toString()));
            MDC.put("req.queryString", StringUtils.defaultString(request.getQueryString()));
            MDC.put("req.requestURIWithQueryString", request.getRequestURI() + (request.getQueryString() == null ? "" : "?"+request.getQueryString()));
            String clientIp = IpUtil.getIpAddr(request);
			MDC.put("req.remoteAddr", StringUtils.defaultString(clientIp));
            
            Profiler.start(request.getServletPath());
            
            //为每一个请求创建一个traceId，方便查找日志时可以根据ID查找出一个http请求所有相关日志
            // LogTraceUtils完成的功能是: MDC.put("traceId",StringUtils.remove(UUID.randomUUID().toString(),"-"))
            LogTraceUtil.beginTrace();
            chain.doFilter(request, response);
            
			logSpiderUserAgent(request, clientIp);
        }finally {
        	Profiler.release();
        	Constants.LOGGER_DUMP_PROFILER.info(Profiler.dump());
            clearMDC();
        }
    }

	private void logSpiderUserAgent(HttpServletRequest request, String clientIp) {
		if(SpiderUtil.isSpider(request)) {
			String userAgent = request.getHeader("User-Agent");
			String spiderName = SpiderUtil.getSpiderName(userAgent);
			Constants.LOGGER_SPIDER.info(request.getRequestURL()+"\t"+clientIp+"\t"+spiderName+"\t");
		}
	}


    private static void clearMDC() {
        Map map = MDC.getContext();
        if(map != null) {
            map.clear();
        }
    }
}