[工具类]AOP打印入参出参和响应时间

AOP打印入参出参和响应时间

用一个around代替before和after去打印入参出参,并且记录响应时间.

注意,重写入参出参类的toString,避免一些大字段被打印占位置.

package com.xx.xx.config;

import com.alibaba.fastjson.JSON;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.Signature;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.annotation.Configuration;

import java.util.StringJoiner;

@Aspect
@Configuration
public class AspectConfig {

    private static Logger logger = LoggerFactory.getLogger(AspectConfig.class);

    //AOP切面位置
    @Pointcut("execution(public * com.em.controller.*.*(..))")
    public void sentenceLog(){}

    /**
     * ※Desc   : 用around代替before和after,打印入参出参和时间
     * ※Author : zhangs@em.com
     * ※Date   : 2021/11/26
     */
    @Around("sentenceLog()")
    public Object around(ProceedingJoinPoint joinPoint) {
        Signature signature = joinPoint.getSignature();
        try {
            // 记录下请求内容
            logger.info("##[REQUEST]\t [CLASS_METHOD:]{}.{}; [ARGS:]{}", signature.getDeclaringTypeName(),signature.getName(), flatArgs(joinPoint.getArgs()));
            long startMs = System.currentTimeMillis();
            //调用 proceed() 方法才会真正的执行实际被代理的方法
            Object result = joinPoint.proceed();
            long consumedMs = System.currentTimeMillis() - startMs;
            logger.info("##[RESPONSE]\t [CONSUMED_MS:]{}; [RESPONSE:]{}", consumedMs, JSON.toJSONString(result));
            return result;
        } catch (Throwable te) {
            logger.error(te.getMessage(),te);
            throw new RuntimeException(te.getMessage());
        }
    }

    private String flatArgs(Object[] args) {
        try {
            return JSON.toJSONString(args);
        } catch (Exception e) {
            StringJoiner sj = new StringJoiner(", ");
            for (Object o : args) {
                if (o != null) {
                    sj.add(o.toString());
                }
            }
            return sj.toString();
        }
    }

    /*@Before("sentenceLog()")
    public void doBefore(JoinPoint joinPoint) throws Throwable {
        try {
            // 接收到请求，记录请求内容
            ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
            HttpServletRequest request = attributes.getRequest();
            // 记录下请求内容
            logger.info("[REQUEST:] \t [URL:]{} \t [HTTP_METHOD:]{} \t [IP:]{} \t [CLASS_METHOD:]{} \t [ARGS:]{}"
                    ,request.getRequestURL().toString(), request.getMethod(), request.getRemoteAddr()
                    ,joinPoint.getSignature().getDeclaringTypeName() + "." + joinPoint.getSignature().getName()
                    , flatArgs(joinPoint.getArgs()));
        } catch (Exception e) {
            logger.warn("aspect错误:",e);
        }
    }

    @AfterReturning(returning = "ret", pointcut = "sentenceLog()")
    public void doAfterReturning(Object ret) throws Throwable {
        // 处理完请求，返回内容
        logger.info("[RESPONSE:] \t " + JSON.toJSONString(ret));
    }*/

}