springboot切面寫法

自定義一個(gè)注解

@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Inherited
@Documented
public @interface PreschoolAuth {
    String value() default "#provinceCode";
}

定義增強(qiáng)

@Component
@Aspect
public class AuthAspect {

    @Pointcut("@annotation(PreschoolAuth)")
    public void authentication(){}

    @Before("authentication()")
    public void judge(JoinPoint joinPoint) {
        // 獲取被增強(qiáng)的方法注解的參數(shù)
        MethodSignature signature = (MethodSignature) joinPoint.getSignature();
        Method method = signature.getMethod();
        PreschoolAuth auth = method.getAnnotation(PreschoolAuth.class);
        String key = auth.value();

        // 根據(jù)注解的參數(shù)獲得被增強(qiáng)的方法的參數(shù)
        Object[] args = joinPoint.getArgs();
        List<String> codes = ParamUtil.parseExpression(key, method, args, List.class);
        if (CollectionUtils.isEmpty(codes)){
            throw new BizException(ErrorCode.PERMISSION_DENIED);
        }
        // 獲取區(qū)劃碼
        String code = codes.get(0);

        // 獲取請求頭
        RequestAttributes requestAttributes = RequestContextHolder.getRequestAttributes();
        ServletRequestAttributes sra = (ServletRequestAttributes)requestAttributes;
        HttpServletRequest request = sra.getRequest();
        String secret = request.getHeader("XXXX");
        if(StringUtils.isBlank(secret)) {
            throw new BizException(ErrorCode.XXXXXX);
        }
        // 做校驗(yàn)  do.......
    
    }
}

在需要增強(qiáng)的方法上使用注解

    @PreschoolAuth(value = "#provinceCode")
    public Object query(String provinceCode,.... ) {

        return ....;
    }

工具類

public class ParamUtil {

    public static <T> T parseExpression(String expression, Method method, Object[] args, Class<T> classType) {
        if (StringUtils.isBlank(expression)) {
            return null;
        } else if (!expression.trim().startsWith("#") && !expression.trim().startsWith("$")) {
            return null;
        } else {
            LocalVariableTableParameterNameDiscoverer discoverer = new LocalVariableTableParameterNameDiscoverer();
            String[] paramNames = discoverer.getParameterNames(method);
            if (ArrayUtils.isEmpty(paramNames)) {
                return null;
            } else {
                StandardEvaluationContext context = new StandardEvaluationContext();

                for (int i = 0; i < paramNames.length; ++i) {
                    context.setVariable(paramNames[i], args[i]);
                }
                return (new SpelExpressionParser()).parseExpression(expression).getValue(context, classType);
            }
        }
    }
}

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者