LoggingFilter.java 1.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445
  1. package com.example.logging;
  2. import javax.servlet.*;
  3. import javax.servlet.http.HttpServletRequest;
  4. import java.io.IOException;
  5. import org.slf4j.MDC;
  6. public class LoggingFilter implements Filter {
  7. @Override
  8. public void init(FilterConfig filterConfig) throws ServletException {}
  9. @Override
  10. public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)
  11. throws IOException, ServletException {
  12. HttpServletRequest req = (HttpServletRequest) request;
  13. String traceId = extractOrGenerate(req);
  14. try {
  15. MDC.put("traceId", traceId);
  16. MDC.put("uri", req.getRequestURI());
  17. // 如果能从认证/上下文拿到 userId,填入 MDC
  18. String userId = req.getHeader("X-User-Id");
  19. if (userId != null) MDC.put("userId", userId);
  20. long start = System.currentTimeMillis();
  21. chain.doFilter(request, response);
  22. long duration = System.currentTimeMillis() - start;
  23. MDC.put("duration", String.valueOf(duration));
  24. } finally {
  25. MDC.remove("traceId");
  26. MDC.remove("uri");
  27. MDC.remove("userId");
  28. MDC.remove("duration");
  29. }
  30. }
  31. @Override
  32. public void destroy() {}
  33. private String extractOrGenerate(HttpServletRequest req) {
  34. String t = req.getHeader("X-Trace-Id");
  35. if (t != null && !t.isEmpty()) return t;
  36. return java.util.UUID.randomUUID().toString();
  37. }
  38. }