image.png

servlet的requestbody以及response的body一旦流被讀取了，就無法再次消費了，因此這對于有要攔截請求，記錄相關信息的時候，帶來一個潛在的坑。那么如何處理這個呢，利用filter，wrapper一層，然后proceed，最后response完之后在把cached的body設置回原始響應。如果你的項目應用到了Spring，那么Spring已幫你實現(xiàn)了此類需求。直接上代碼(基于Spring BOOT)

1、攔截器

/**
 * Filter配置
 *
 * @author lengyu
 * @date 2017-04-21 21:56
 */
@Configuration
public class FilterConfig {
    @Bean
    public FilterRegistrationBean streamFilterRegistration() {
        FilterRegistrationBean registration = new FilterRegistrationBean();
        registration.setDispatcherTypes(DispatcherType.REQUEST);
        registration.setFilter(new StreamFilter());
        registration.addUrlPatterns("/*");
        registration.setName("streamFilter");
        registration.setOrder(3);
        return registration;
    }
}

2、攔截器實現(xiàn)類

/**
 * 使用ContentCachingRequestWrapper來包裝HttpServletRequest，
 * 使用ContentCachingResponseWrapper來包裝HttpServletResponse，
 *
 * @author lengyu
 * @date
 */
public class StreamFilter implements Filter {
    private static final Logger log = LoggerFactory.getLogger(StreamFilter.class);

    @Override
    public void init(FilterConfig config) throws ServletException {
    }

    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
        HttpServletRequest req = (HttpServletRequest) request;
        String contentType = request.getContentType();
        HttpServletResponse resp = (HttpServletResponse) response;
        String method = req.getMethod();

        boolean hasApiUri = req.getRequestURI().startsWith("/api/");
        boolean auth = StringUtils.isBlank(HttpContextUtils.getRequestToken(req, Const.AUTHORIZATION_KEY));
        if ((!(request instanceof HttpServletRequest) || auth) && !hasApiUri) {
            chain.doFilter(request, response);
            return;
        }

        if ((contentType != null) && (contentType.startsWith("multipart/form-data")) && ("POST".equalsIgnoreCase(method))) {
            log.info("==> 當前請求為文件上傳，不作請求日志收集");
        } else {
            Date startTime = new Date();
            ContentCachingRequestWrapper requestWrapper = new ContentCachingRequestWrapper(req);
            ContentCachingResponseWrapper responseWrapper = new ContentCachingResponseWrapper(resp);
            try {
                chain.doFilter(requestWrapper, responseWrapper);
            } finally {
                String requestBody = req.getQueryString();
                if (StringUtils.isEmpty(requestBody)) {
                    requestBody = new String(requestWrapper.getContentAsByteArray());
                }
                String responseBody = new String(responseWrapper.getContentAsByteArray());
                LogInfo logInfo = LogInfo.build(req, startTime, requestBody, responseBody);
                String json = logInfo.toJson();

                VhscProperties vhscProperties = (VhscProperties) SpringHelper.getBean("vhscProperties");
                boolean isNotice = logInfo.getWasteTime() > vhscProperties.getRespNoticeTime();
                log.info("[日志歸檔] - {}", json);
                if (isNotice) {
                    log.warn("[性能告警] - {}", json);
                }
                responseWrapper.copyBodyToResponse();
            }
        }
    }
    @Override
    public void destroy() {
    }
}

3、日志BEAN

/**
 * 請求日志信息收集
 *
 * @author lengyu
 * @version 2.0
 * @date 2017/9/2 22:45
 */
public class LogInfo implements Serializable {
    private String threadToken;//請求線程ID
    private String scheme;//協(xié)議
    private String host;//主機
    private int port;//端口
    private String method;//請求方法
    private String path;//請求路徑
    private Object params;//請求參數(shù)
    private Object result;//返回值
    private String clientIp;//客戶端IP
    private String serverIp;//當前服務節(jié)點IP
    private Map<String, String> headers;//請求頭
    private String remoteUser;//當前登陸人

    @JSONField(format = "yyyy-MM-dd HH:mm:ss.SSS")
    private Date startTime = new Date();//請求開始時間
    @JSONField(format = "yyyy-MM-dd HH:mm:ss.SSS")
    private Date endTime;//請求結束時間

    public String toJson() {
        return JSONObject.toJSONString(this);
    }


    /**
     * 消耗時長=請求開始時間-請求結束時間
     *
     * @return
     */
    public Long getWasteTime() {
        return this.endTime.getTime() - this.startTime.getTime();
    }

    /**
     * 生成日志對象
     *
     * @param req
     */
    public static LogInfo build(HttpServletRequest req, Date startTime, String requestBody, String responseBody) {
        LogInfo info = new LogInfo();
        info.threadToken = LogThreadTokenConvert.getThreadToken();
        info.scheme = req.getScheme();
        info.host = req.getServerName();
        info.port = req.getServerPort();
        info.path = req.getRequestURI();
        info.method = req.getMethod();
        info.clientIp = IPUtils.getIpAddr(req);
        info.serverIp = req.getLocalAddr();
        info.startTime = startTime;
        info.endTime = new Date();

        try {
            SysUserEntity sysUserEntity = ShiroUtils.getUserEntity();
            if (sysUserEntity != null) {
                info.remoteUser = sysUserEntity.getUserId() + ":" + sysUserEntity.getUsername();
            }
        } catch (Exception e) {
            info.remoteUser = req.getRemoteUser();
            e.printStackTrace();
        }

        VhscProperties vp = (VhscProperties) SpringHelper.getBean("vhscProperties");
        if (vp.getHasHeaders()) {
            info.headers = getHeaderMap(req);
        }
        if (vp.getHasParams()) {
            try {
                info.params = JSONObject.parseObject(requestBody);
            } catch (JSONException e) {
                info.params = requestBody;
            }
        }
        if (vp.getHasResult()) {
            try {
                info.result = JSONObject.parseObject(responseBody);
            } catch (JSONException e) {
                info.result = responseBody;
            }
        }
        return info;
    }

    public static Map<String, String> getHeaderMap(HttpServletRequest req) {
        Map<String, String> map = new HashMap<>();
        Enumeration enu = req.getHeaderNames();//取得全部頭信息
        while (enu.hasMoreElements()) {//以此取出頭Name
            String headerName = (String) enu.nextElement();
            String headerValue = req.getHeader(headerName);//取出頭信息Value
            map.put(headerName, headerValue);
        }
        return map;
    }
/***省略GET/SET***/
}