AOP & @Aspect 记录日志使用

骑猪看日落 2023-07-09 02:18 14阅读 0赞

package com.xxx.aspect;
    
    import com.github.pagehelper.PageHelper;
    import com.xxx.common.Result;
    import com.xxx.utils.SpringContextUtils;
    import org.aspectj.lang.JoinPoint;
    import org.aspectj.lang.annotation.*;
    import org.slf4j.Logger;
    import org.slf4j.LoggerFactory;
    import org.springframework.stereotype.Component;
    import org.springframework.web.context.request.RequestContextHolder;
    import org.springframework.web.context.request.ServletRequestAttributes;
    
    import javax.servlet.http.HttpServletRequest;
    import java.io.IOException;
    
    @Aspect
    @Component
    public class HttpControllerAspect {
    
        private final static Logger logger = LoggerFactory.getLogger(HttpControllerAspect.class);
    
        ThreadLocal<Long> startTime = new ThreadLocal();
    
        ThreadLocal<String> classMethod = new ThreadLocal();
    
        @Pointcut("execution(public * com.xxx.controller.*.*(..)) || execution(public * com.xxx.auth.controller.*.*(..))")
        public void log(){
        };
    
        @Before("log()")
        public void doBefore(JoinPoint point) throws IOException {
            ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
            HttpServletRequest request = attributes.getRequest();
            startTime.set(System.currentTimeMillis());
            classMethod.set(point.getSignature().getDeclaringTypeName() + "." + point.getSignature().getName());
            //url
            logger.info("class_method={}, url={}, requestUri={}, http_method={}, realIp={}"
                    ,point.getSignature().getDeclaringTypeName() + "." + point.getSignature().getName()
                    , request.getRequestURL(),request.getRemoteAddr(),request.getMethod(), SpringContextUtils.getRequestIp(request)
                   );
        }
    
        @After("log()")
        public void doAfter(){
            PageHelper.clearPage();
        }
    
        @AfterReturning(returning = "object",pointcut = "log()")
        public void doAfterReturning(Result object){
            logger.info("class_method={}，返回结果={}, 处理时间={}毫秒",classMethod.get(),object.toString(),System.currentTimeMillis()-startTime.get());
            startTime.remove();
            classMethod.remove();
        }
    }