springboot filter and interceptor实战之mdc日志打印

<property name="log.pattern" value="%d{yyyy-MM-dd'T'HH:mm:ss.SSSXXX}%level [%thread] [%logger{50}:%line] [uuid:%X{operation_id}] %msg%n"></property>

 1 @WebFilter(filterName="logFilter", urlPatterns="/*")
 2 
 3 public class LogFilter implements Filter {
 4 
 5     private static final Logger log = LoggerFactory.getLogger(LogInterceptor.class);
 6 
 7    
 8 
 9     @Override
10 
11     public void init(FilterConfig filterConfig) throws ServletException {
12 
13         log.info("---------log filter init");
14 
15     }
16 
17  
18 
19     @Override
20 
21     public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
22 
23             throws IOException, ServletException {
24 
25         log.info("log filter doFilter");
26 
27         ServletRequest requestWrapper = null;
28 
29         requestWrapper = new MyRequestWrapper((HttpServletRequest) request);
30 
31         chain.doFilter(requestWrapper, response);
32 
33     }
34 
35  
36 
37     @Override
38 
39     public void destroy() {
40 
41         log.info("-------------log filter destroy");
42 
43        
44 
45     }
46 
47 }

 1 @SpringBootApplication()
 2 
 3 @ServletComponentScan       //扫描过滤器
 4 
 5 public class LogApplication {
 6 
 7  
 8 
 9    public static void main(String[] args) {
10 
11        Tools.setMdc(Tools.getUuid());
12 
13       SpringApplication.run(LogApplication.class, args);
14 
15    }
16 
17 }

 1 public class MyRequestWrapper extends HttpServletRequestWrapper {
 2 
 3     private String body;
 4     private String requestMethod;
 5     
 6     public MyRequestWrapper(HttpServletRequest request) throws IOException {
 7         super(request);
 8         StringBuilder sb = new StringBuilder();
 9         String line = null;
10         BufferedReader reader = request.getReader();
11         requestMethod = request.getMethod();
12         
13         while((line = reader.readLine()) != null) {
14             sb.append(line);
15         }
16         body = sb.toString();
17     }
18     private boolean isPostOrPut() {
19         return "POST".equalsIgnoreCase(requestMethod) || "PUT".equalsIgnoreCase(requestMethod);
20     }
21     @Override
22     public String getQueryString() {
23         if(isPostOrPut()) {
24             return body;
25         } else {
26             return super.getQueryString();
27         }
28     }
29     @Override
30     public ServletInputStream getInputStream() throws IOException {
31         ByteArrayInputStream bais = new ByteArrayInputStream(body.getBytes(CommonVar.DEFAULT_CHARSET));
32         return new ServletInputStream() {
33             
34             @Override
35             public int read() throws IOException {
36                 return bais.read();
37             }
38 
39             @Override
40             public boolean isFinished() {
41                 return bais.available()==0;
42             }
43 
44             @Override
45             public boolean isReady() {
46                 return true;
47             }
48 
49             @Override
50             public void setReadListener(ReadListener listener) {
51                 
52             }
53         };
54     }
55 }

 1 public class LogInterceptor implements HandlerInterceptor{
 2     private static final Logger log = LoggerFactory.getLogger(LogInterceptor.class);
 3     
 4     private static final String REQUEST_START_TIME = "request_start_time";
 5     
 6     @Override
 7     public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
 8             throws Exception {
 9         setMdc(request);
10         logRequestMsg(request);
11         request.setAttribute(REQUEST_START_TIME, System.currentTimeMillis());
12         System.out.println("preHandle");
13         return true;
14     }
15 
16     @Override
17     public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler,
18             ModelAndView modelAndView) throws Exception {
19         HandlerInterceptor.super.postHandle(request, response, handler, modelAndView);
20         System.out.println("postHandle");
21     }
22 
23     @Override
24     public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex)
25             throws Exception {
26         logRequestEnd(request);
27         Tools.removeMdc();
28         System.out.println("afterCompletion");
29     }
30     private void setMdc(HttpServletRequest request) {
31         MDC.put("operation_id", Tools.getUuid());
32     }
33     /**
34      * log request url and request cost time
35      * @param request
36      */
37     private void logRequestEnd(HttpServletRequest request) {
38         long startTime = (long) request.getAttribute(REQUEST_START_TIME);
39         long endTime = System.currentTimeMillis();
40         long interval = endTime - startTime;
41         log.debug("request_end: {}, cost time:{}ms",request.getRequestURL(), interval);
42     }
43     /**
44      * log request url and param
45      * @param request 
46      * @throws IOException
47      */
48     private void logRequestMsg(HttpServletRequest request) throws IOException {
49         String url = request.getRequestURL().toString();
50         String method = request.getMethod();
51         String query = request.getQueryString();
52         log.debug("request_receive: url:{},method:{},query:{}", url, method, query);
53     }
54 }

 1 @Configuration
 2 public class LogWebAppConfig implements WebMvcConfigurer  {
 3 
 4     @Override
 5     public void addInterceptors(InterceptorRegistry registry) {
 6         registry.addInterceptor(new LogInterceptor()).addPathPatterns("/**");
 7         WebMvcConfigurer.super.addInterceptors(registry);
 8     }
 9     
10 }