自定义注解之打印日志

/**
 * 自定义日志注解
 * @author Lynn
 * @date 2022/6/30 15:09
 */
@Documented
@Target({ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface Log {
    LogEnum logEnum() default LogEnum.RUNTIME;
}

/**
 * @author Lynn
 * @date 2022/6/30 16:46
 */
@Getter
@AllArgsConstructor
public enum LogEnum {

    /**
     * ACCESS
     */
    ACCESS("access", LogManager.getAccessLog()),

    /**
     * INTERFACE
     */
    INTERFACE("interface", LogManager.getInterfaceLog()),

    /**
     * SYSTEM
     */
    SYSTEM("system", LogManager.getSystemLog()),

    /**
     * RUNTIME
     */
    RUNTIME("runtime", LogManager.getRuntimeLog()),

    ;

    private String code;

    private ILog log;
}

/**
 * @author Lynn
 * @date 2022/6/30 15:12
 */
@Aspect
@Component
@Slf4j
public class LogAspect {

    /**
     * 切点
     */
    @Pointcut("@annotation(com.hahacloud.server.saas.service.iot.biz.service.annotation.Log)")
    public void pointCut() {
    }

    /**
     * 出入参打印
     *
     * @param joinPoint joinPoint
     * @return proceed
     * @throws Throwable Throwable
     */
    @Around("pointCut()")
    public Object around(ProceedingJoinPoint joinPoint) throws Throwable {
        Method method = ((MethodSignature) joinPoint.getSignature()).getMethod();
        Log annotation = method.getAnnotation(Log.class);
        if (null == annotation) {
            return joinPoint.getArgs();
        }

        StringBuilder params = new StringBuilder();
        //方法名
        String methodName = joinPoint.getSignature().getName();
        //参数值
        Object[] argValues = joinPoint.getArgs();
        //参数名称
        String[] argNames = ((MethodSignature) joinPoint.getSignature()).getParameterNames();
        if (argValues != null) {
            for (int i = 0; i < argValues.length; i++) {
                params.append(argNames[i]).append(":").append(argValues[i]);
            }
        }

        annotation.logEnum().getLog().info(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>{} 请求参数为: {}", methodName, JSON.toJSONString(params));
        Object proceed = joinPoint.proceed();
        annotation.logEnum().getLog().info(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>{} 响应参数为: {}", methodName, JSON.toJSONString(proceed));
        return proceed;
    }

    /**
     * 异常日志打印
     *
     * @param joinPoint joinPoint
     * @param e         e
     */
    @AfterThrowing(pointcut = "pointCut()", throwing = "e")
    public void throwIng(JoinPoint joinPoint, Throwable e) {
        String methodName = joinPoint.getSignature().getName();
        LogManager.getRuntimeLog().error(">>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>{} 抛出异常为: {}", methodName, e);
    }
}