【AOP系列】3.安全检查

在Java中，我们可以使用Spring AOP（面向切面编程）和自定义注解来做安全检查。以下是一个简单的示例：

首先，我们创建一个自定义注解，用于标记需要进行安全检查的方法：

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Target(ElementType.METHOD) //注解放置的目标位置，METHOD是可放在方法级别
@Retention(RetentionPolicy.RUNTIME) //注解在哪个阶段执行
public @interface SecurityCheck {
    String value() default ""; //注解的值
}

然后，我们创建一个切面，用于处理标记了@SecurityCheck的方法：

import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.springframework.stereotype.Component;

@Aspect
@Component
public class SecurityAspect {

    @Pointcut("@annotation(com.yourpackage.SecurityCheck)") //指定自定义注解的路径
    public void pointcut() {}

    @Around("pointcut()")
    public Object around(ProceedingJoinPoint joinPoint) {
        Object result = null;
        // 这里只是一个示例，你可以根据实际情况进行安全检查
        if (checkSecurity()) {
            try {
                result = joinPoint.proceed(); //执行方法
            } catch (Throwable e) {
                e.printStackTrace();
            }
        } else {
            System.out.println("安全检查未通过，方法执行被拒绝");
        }
        return result;
    }

    private boolean checkSecurity() {
        // 进行安全检查，返回true表示检查通过，返回false表示检查未通过
        return true;
    }
}

最后，我们在需要进行安全检查的方法上添加@SecurityCheck注解：

public class SomeService {

    @SecurityCheck("执行某个操作")
    public void someMethod() {
        //...
    }
}

这样，当someMethod方法被调用时，SecurityAspect中的around方法会被触发，从而实现安全检查的统一处理。