快捷導(dǎo)航

spring boot+自定義 AOP 實現(xiàn)全局校驗的實例代碼

更新時間：2019年04月09日 16:07:06 作者：吃桃子的小松鼠

最近公司重構(gòu)項目，重構(gòu)為最熱的微服務(wù)框架 spring boot，重構(gòu)的時候遇到幾個可以統(tǒng)一處理的問題。這篇文章主要介紹了spring boot+自定義 AOP 實現(xiàn)全局校驗 ,需要的朋友可以參考下

最近公司重構(gòu)項目，重構(gòu)為最熱的微服務(wù)框架 spring boot，重構(gòu)的時候遇到幾個可以統(tǒng)一處理的問題，也是項目中經(jīng)常遇到，列如：統(tǒng)一校驗參數(shù)，統(tǒng)一捕獲異常。。。

僅憑代碼去控制參數(shù)的校驗，有時候是冗余的，但通過框架支持的去控制參數(shù)的校驗，是對于開發(fā)者很友好，先看下面的例子

 @NotEmpty(message="手機號不能為空")
   @Size(min=11,max=11,message="手機號碼長度不正確")
   @Pattern(regexp=StringUtils.REGEXP_MOBILE,message="手機號格式不正確")
  private String mobile;

這是spring boot支持的校驗注解，然后我們在 contoller層加上@Valid 注解就可以達到校驗的目的。這是一種框架自帶的

本章就展示一種自定義的 AOP 校驗，首先寫一個注解，注解里面可以寫上我們需要校驗的規(guī)則，比如長度，正則。。。

@Documented
@Target({ElementType.FIELD,ElementType.METHOD})
@Retention(RetentionPolicy.RUNTIME)
public @interface ValidateParam {
  int min() default 0;
  int max() default Integer.MAX_VALUE;
  String message() default "params is not null";
  String regexp();
  Class<?>[] groups() default { };
   Class<? extends Payload>[] payload() default { };
   boolean isNotNull() default true;
}

然后定義一個AOP類

package com.onecard.primecard.common.aop;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.regex.Pattern;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.context.support.ClassPathXmlApplicationContext;
import org.springframework.stereotype.Component;
import com.jfcf.core.dto.ResultData;
import com.onecard.core.support.util.StringUtils;
import com.onecard.primecard.common.annotation.ValidateParam;
import com.onecard.primecard.common.utils.ResultDataUtil;
/**
 * 全局 切面類(校驗參數(shù))
 * 
 * @author Administrator
 *
 */
@Aspect
@Component
public class GobalHandlerAspect {
  private static Logger logger = LoggerFactory.getLogger(GobalHandlerAspect.class);
  @Pointcut("execution(* 包名.controller..*.*(..)) && execution(* 包名.controller..*.*(..))")
  public void checkAspect(){};
  @Before("checkAspect()")
  public void befor(JoinPoint joinPoint) throws Exception{
    //前置統(tǒng)一輸出參數(shù)
    Object[] args = joinPoint.getArgs();
    if(args != null && args.length>0){
      Object obj = args[0];
      ParameterizedType pt = (ParameterizedType)obj.getClass().getGenericSuperclass();
      Class<?> classzz = (Class<?>) pt.getActualTypeArguments()[0];
      logger.info("【小X卡】-【請求實體入?yún)ⅰ浚?+classzz.newInstance().toString());
    }
  }
  @Around("checkAspect()")
  public Object around(ProceedingJoinPoint joinPoint) throws Throwable{
    //校驗參數(shù)
    Object[] args = joinPoint.getArgs();
    Object obj = null;
    if(args != null && args.length > 0){
      obj = args[0];
      Class classzz = obj.getClass();
      //沒有順序和秩序的數(shù)組
      Field[] fieldArray = classzz.getDeclaredFields();
      ArrayList<Field> fieldList = new ArrayList<Field>(Arrays.asList(fieldArray));
      String res = checkParam(fieldList,obj);
      if(StringUtils.isNotNull(res)){
        return ResultDataUtil.result(ResultData.STATUS_PARAM_ERROR, res);
      }
    }
    return joinPoint.proceed();
  }
  private String checkParam(ArrayList<Field> fieldList, Object obj) throws Exception {
    for(Field field : fieldList){
      ValidateParam validateParam = field.getAnnotation(ValidateParam.class);
      logger.info("【小X卡】獲取注解值："+validateParam.isNotNull()+"min="+validateParam.min()+"max="+validateParam.max());
      Method method = obj.getClass().getMethod("get"+getMethodName(field.getName()));
      logger.info("【小X卡】入?yún)嶓w方法名稱："+method.getName());
      if(method != null){
        Object val = method.invoke(obj);
        logger.info("【小x卡】回調(diào)方法："+val);
        if(validateParam != null && validateParam.isNotNull() == true){
          if(null == val || "".equals(val) ){
            return field.getName()+"必填參數(shù)為空";
          }
        }
        if(validateParam.min()==11 && validateParam.max() == 11){
          if(val.toString().length() != 11){
            return field.getName()+"請輸入?yún)?shù)正確的長度";
          }
        }
        if(validateParam.regexp().equals(StringUtils.REGEXP_MOBILE)){
          if(!Pattern.matches(StringUtils.REGEXP_MOBILE, val.toString())){
            return field.getName()+"參數(shù)格式錯誤";
          }
        }
      }
    }
    return null;
  }
   /**
   * 方法首字母大寫
   * @param fieldName
   * @return
   */
  private String getMethodName(String fieldName) {
    StringBuffer buffer = new StringBuffer();
    String firstLetter = fieldName.substring(0, 1).toUpperCase();
    return buffer.append(firstLetter).append(fieldName.substring(1, fieldName.length())).toString();
  }    
 }

定義一個切點 @Pointcut，用execution 表達式，去獲取要校驗的某個類和某個方法，也就是連接點，然后用定義一個通知，上面代碼中有2個通知，一個前置通知@Before，一個環(huán)繞通知@Around，我們使用功能最強大的環(huán)繞通知。

通過上面的代碼可以看出首先獲取參數(shù)，然后通過反射機制獲取入?yún)ο笾械娜孔侄危?再去獲取我們在字段中加我們自定義注解的字段，通過反射方法的回調(diào)，獲取字段值，對值做判斷，返回校驗結(jié)果。

總結(jié)

以上所述是小編給大家介紹的spring boot+自定義 AOP 實現(xiàn)全局校驗的實例代碼，希望對大家有所幫助，如果大家有任何疑問歡迎給我留言，小編會及時回復(fù)大家的！

您可能感興趣的文章: