<!--lombok-->
<dependency>
 <groupId>org.projectlombok</groupId>
 <artifactId>lombok</artifactId>
 <version>1.18.2</version>
 <optional>true</optional>
</dependency>

<!--Spring AOP-->
<dependency>
 <groupId>org.springframework.boot</groupId>
 <artifactId>spring-boot-starter-aop</artifactId>
</dependency>

/**
 * 用戶請(qǐng)求權(quán)限校驗(yàn)
 */
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface MyPermissionTag {
 String value() default "";
 String name() default "";
}

@Slf4j
@Aspect
@Component
public class AuthInterceptor {


 /**
 * 參數(shù)處理
 *
 * @param point
 */
 @Before("@annotation(com.luo.common.tag.MyPermissionTag)")
 public void beforeProReq(JoinPoint point) {
 log.info("前置攔截-開(kāi)始");
 Request req = getOperationRequest(point.getArgs());
 if (req != null) {
  //解密帳號(hào)
  log.info("前置攔截-開(kāi)始解密ACCOUNT:{}", req.getAccount());


  log.info("前置攔截-結(jié)束解密ACCOUNT:{}", req.getAccount());
 }
 log.info("前置攔截-結(jié)束");
 }


 @Around("@annotation(com.luo.common.tag.MyPermissionTag)")
 public Object authCheck(ProceedingJoinPoint pjp) throws Throwable {
 log.info("權(quán)限攔截-開(kāi)始");
 //請(qǐng)求方法
 ReqMethod reqMethod = getPermissionTag(pjp);


 MyPermissionTag myPermissionTag =reqMethod.perTag;
 log.info(myPermissionTag.value()); //獲取配置的值
 log.info("權(quán)限攔截-開(kāi)始-攔截到方法:{}", reqMethod.getMethodName());


 if("true".equals(myPermissionTag.value().toString())){
  //錯(cuò)誤返回
  Response notGoRes = new Response();
  Request req = getOperationRequest(pjp.getArgs());
  // 校驗(yàn)請(qǐng)求對(duì)象
  if (req == null) {
  notGoRes.setErrorMsg("(AUTH)未能獲得有效的請(qǐng)求參數(shù)！");
  log.info("(AUTH-NO)未能獲得有效的請(qǐng)求參數(shù)！");
  return notGoRes;
  }else {//可以在這里根據(jù)請(qǐng)求參數(shù)對(duì)請(qǐng)求做進(jìn)一步校驗(yàn)


  log.info("完成請(qǐng)求校驗(yàn):"+req);




  }
 }else {
  log.info("未開(kāi)啟權(quán)限校驗(yàn)");
 }


 return pjp.proceed();
 }




 /**
 * 獲取 request 接口中的請(qǐng)求參數(shù)
 * @param args
 * @return
 */
 private Request getOperationRequest(Object[] args) {
 if (args == null || args.length <= 0) {
  log.error("AUTH權(quán)限驗(yàn)證：攔截方法的請(qǐng)求參數(shù)為空！");
  return null;
 }
 Object obj = args[0];
 if (obj instanceof Request) {
  log.info("AUTH權(quán)限驗(yàn)證：請(qǐng)求對(duì)象為正確的OperationRequest對(duì)象");
  return (Request) obj;
 }
 return null;
 }




 /**
 * 獲取攔截的資源標(biāo)簽
 * 這里可以獲取方法名+注解信息(包括 key+value 等)
 * @param pjp
 * @return
 * @throws SecurityException
 * @throws NoSuchMethodException
 */
 private ReqMethod getPermissionTag(ProceedingJoinPoint pjp) throws NoSuchMethodException, SecurityException {
 Signature signature = pjp.getSignature();
 MethodSignature methodSignature = (MethodSignature) signature;
 Method targetMethod = methodSignature.getMethod();
 Method realMethod = pjp.getTarget().getClass().getDeclaredMethod(signature.getName(), targetMethod.getParameterTypes());
 MyPermissionTag permissionTag = realMethod.getAnnotation(MyPermissionTag.class);
 return new ReqMethod(permissionTag, realMethod.getName());
 }


 @Setter
 @Getter
 class ReqMethod {


 private MyPermissionTag perTag;
 private String methodName;


 public ReqMethod(MyPermissionTag perTag, String methodName) {
  this.perTag = perTag;
  this.methodName = methodName;
 }


 }
}

@PostMapping("/helloluo")
@MyPermissionTag(value = "true")
public String helloluo(UserPojoReq userPojoReq){
 return "Hello World";
}