Springboot Aop日志妙用

最新推荐文章于 2025-01-06 16:01:20 发布

原创最新推荐文章于 2025-01-06 16:01:20 发布 · 164 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#spring boot #aop

笔记专栏收录该内容

14 篇文章

订阅专栏

本文介绍了一个基于AOP（面向切面编程）的日志记录和权限验证的具体实现。通过定义切入点和切面，实现了对指定方法的日志记录及参数检查，并通过自定义注解方式对控制器的方法进行权限验证。

这个是自定义注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface AuthChecker {

}

@Component  //声明组件
@Aspect //  声明切面
public class LogAop extends BaseController {

    /**
     * 自定义日志
     */
    private Logger logger = LoggerFactory.getLogger(LogAop.class);

    /**
     * 功能描述 service切面
     *
     * @param * @param
     * @return void
     * @author
     * @date 2021/6/9
     */
	//定义一个切入点
    @Pointcut("execution(public * com.esp.espplay.play.service.impl.*.*(..))")
    public void webLog() {

    }

    long startTime = System.currentTimeMillis();

    @Before("webLog()")
    public void doBefore(JoinPoint point) {
        logger.info("方法" + point.getSignature().getDeclaringTypeName() + "." + point.getSignature().getName() + "执行开始");
        logger.info("入参为"+ JSON.toJSONString(((MethodSignature)point.getSignature()).getParameterNames())+" : "+ JSON.toJSONString(point.getArgs()));
    }
//    @After("webLog()")
//    public void doAfter(JoinPoint point){
//        logger.info(System.currentTimeMillis()-startTime/1000f +"秒！"+"----------------------");
//    }

    /**
     * 功能描述 controller切面
     *这个是通过注解实现的
     * @param * @param
     * @return void
     * @author
     * @date 2021/6/9
     */
    @Pointcut("@annotation(com.esp.espplay.play.aop.AuthChecker)")
    public void webContro() {

    }
     /**
     * 功能描述 controller切面
     *
     * @param * @param
     * @return void
     * @author
     * @date 2021/6/9
     */
    @Pointcut("execution(public * com.esp.espplay.play.controller.*.*(..))")
    public void webController() {

    }

    // 定义 advise
    @Around("webContro()")
    public ResponseEntity<ResultModel> checkAuth(ProceedingJoinPoint joinPoint) throws Throwable {
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes())
                .getRequest();

        // 检查用户所传递的 token 是否合法
        String token = getUserToken(request);
        if (!token.equalsIgnoreCase("123456")) {
            return responseEntity("错误, 权限不合法!");
        }

        return responseEntity(joinPoint.proceed());
    }

    private String getUserToken(HttpServletRequest request) {
        Cookie[] cookies = request.getCookies();
        if (cookies == null) {
            return "";
        }
        for (Cookie cookie : cookies) {
            if (cookie.getName().equalsIgnoreCase("token")) {
                return cookie.getValue();
            }
        }
        return "";
    }
   @Before("webController()")
   public void doBeforeController(JoinPoint point) {
       logger.info("方法" + point.getSignature().getDeclaringTypeName() + "." + point.getSignature().getName() + "执行开始");
       logger.info("入参为"+ JSON.toJSONString(((MethodSignature)point.getSignature()).getParameterNames())+" : "+ JSON.toJSONString(point.getArgs()));
   }

}

效果图
在这里插入图片描述