RequestLogAspect.java 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126
  1. package com.persagy.aspects;
  2. import cn.hutool.core.util.IdUtil;
  3. import cn.hutool.extra.servlet.ServletUtil;
  4. import com.persagy.dmp.common.model.response.CommonResult;
  5. import com.persagy.utils.JacksonMapper;
  6. import lombok.extern.slf4j.Slf4j;
  7. import org.aspectj.lang.JoinPoint;
  8. import org.aspectj.lang.annotation.*;
  9. import org.springframework.context.annotation.EnableAspectJAutoProxy;
  10. import org.springframework.stereotype.Component;
  11. import org.springframework.web.context.request.RequestContextHolder;
  12. import org.springframework.web.context.request.ServletRequestAttributes;
  13. import org.springframework.web.multipart.MultipartFile;
  14. import javax.servlet.ServletRequest;
  15. import javax.servlet.ServletResponse;
  16. import javax.servlet.http.HttpServletRequest;
  17. import java.util.LinkedList;
  18. import java.util.List;
  19. /**
  20. * @author: yaoll
  21. * @date: 2020-09-22
  22. * @verison: 1.0
  23. */
  24. @Slf4j
  25. @Aspect
  26. @Component
  27. @EnableAspectJAutoProxy
  28. public class RequestLogAspect {
  29. private static final ThreadLocal<LogData> dataStorage = new ThreadLocal<>();
  30. @Pointcut("execution(public * com.persagy.dmp..*.controller.*.*(..))")
  31. //@Pointcut("execution(* *(..))")
  32. public void controllerPoint() {
  33. }
  34. @Before("controllerPoint()")
  35. public void doBefore(JoinPoint joinPoint) {
  36. ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
  37. if (null == attributes) {
  38. log.warn("获取请求参数失败!!");
  39. return;
  40. }
  41. HttpServletRequest request = attributes.getRequest();
  42. LogData data = initFlag(request);
  43. dataStorage.set(data);
  44. log.info("{} REQUEST URL : {} {} {} {}", data.intFlag, data.method, data.url, data.remoteIp, data.remotePort);
  45. log.info("{} REQUEST ARGS : {} ", data.intFlag, JacksonMapper.toSimpleJson(prepare(joinPoint)));
  46. }
  47. public static List<Object> prepare(JoinPoint joinPoint) {
  48. Object[] args = joinPoint.getArgs();
  49. List<Object> list = new LinkedList<>();
  50. if (args != null && args.length > 0) {
  51. for (Object arg : args) {
  52. if (arg instanceof ServletRequest) {
  53. continue;
  54. }
  55. if (arg instanceof ServletResponse) {
  56. continue;
  57. }
  58. if (arg instanceof MultipartFile) {
  59. continue;
  60. }
  61. list.add(arg);
  62. }
  63. }
  64. return list;
  65. }
  66. @AfterReturning(returning = "ret", pointcut = "controllerPoint()")
  67. public void doAfterReturning(Object ret) {
  68. LogData data = dataStorage.get();
  69. if (ret != null) {
  70. if (ret instanceof CommonResult) {
  71. log.info("{} REQUEST DURATION : {} {} {}", data.intFlag, System.currentTimeMillis() - data.timestamp, ((CommonResult) ret).getResult(), data.url);
  72. } else {
  73. log.info("{} REQUEST DURATION : {} {} {}", data.intFlag, System.currentTimeMillis() - data.timestamp, ret.getClass().getName(), data.url);
  74. }
  75. } else {
  76. log.info("{} REQUEST DURATION : {} {} {}", data.intFlag, System.currentTimeMillis() - data.timestamp, "null", data.url);
  77. }
  78. dataStorage.remove();
  79. }
  80. @AfterThrowing(throwing = "ex", pointcut = "controllerPoint()")
  81. public void doAfterThrowing(Throwable ex) {
  82. LogData data = dataStorage.get();
  83. log.error("{} REQUEST EXCEPTION: {} {} {}", data.intFlag, System.currentTimeMillis() - data.timestamp, data.url, ex);
  84. dataStorage.remove();
  85. }
  86. private static LogData initFlag(HttpServletRequest request) {
  87. LogData data = new LogData();
  88. // 请求IP
  89. data.remoteIp = ServletUtil.getClientIP(request, null);
  90. // 请求端口
  91. data.remotePort = request.getRemotePort();
  92. data.timestamp = System.currentTimeMillis();
  93. data.method = request.getMethod();
  94. String requestURI = request.getRequestURI();
  95. String queryString = request.getQueryString();
  96. data.url = requestURI + (queryString == null ? "" : "?" + queryString);
  97. data.intFlag = IdUtil.fastSimpleUUID();
  98. dataStorage.set(data);
  99. return data;
  100. }
  101. /***
  102. * Description: 日志数据实体
  103. * @author : lijie
  104. * @date :2021/8/18 10:36
  105. * Update By lijie 2021/8/18 10:36
  106. */
  107. private static class LogData {
  108. private String intFlag;
  109. private Long timestamp;
  110. private String url;
  111. private String method;
  112. private String remoteIp;
  113. private Integer remotePort;
  114. }
  115. }