手寫一個@Valid字段校驗器的示例代碼

上次給大傢講述瞭 Springboot 中的 @Valid 註解 和 @Validated 註解的詳細用法:

詳解Spring中@Valid和@Validated註解用法

當我們用上面這兩個註解的時候,需要首先在對應的字段上打上規則註解,類似如下。

@Data
public class Employee {
 
    /** 姓名 */
    @NotBlank(message = "請輸入名稱")
    @Length(message = "名稱不能超過個 {max} 字符", max = 10)
    public String name;
 
    /** 年齡 */
    @NotNull(message = "請輸入年齡")
    @Range(message = "年齡范圍為 {min} 到 {max} 之間", min = 1, max = 100)
    public Integer age;
 
}

其實,在使用這些規則註解時,我覺得不夠好用,比如我列舉幾個點:

(1)針對每個字段時,如果有多個校驗規則,需要打多個對應的規則註解,這時看上去,就會顯得較為臃腫。

(2)某些字段的類型根本不能校驗,比如在校驗 Double 類型的字段規則時,打上任何校驗註解,都會提示報錯,說不支持 Double 類型的數據;

(3)每打一個規則註解時,都需要寫上對應的 message 提示信息,這不但使得寫起來麻煩,而且代碼看起來又不雅觀,按理說,我們的一類規則提示應該都是相同的,比如 "xxx不能為空",所以,按理來說,我隻要配置一次提示格式,就可以不用再寫瞭,隻需要配置每個字段的名稱xxx即可。

(4)一般來說,我們通常進行字段校驗時,可能還需要一些額外的數據處理,比如去掉字符串前後的空格,某些數據可以為空的時候,我們還可以設置默認值這些等。

(5)不能進行擴展,如果時自己寫的校驗器,還可以進行需求擴展。

(6)他們再進行校驗的時候,都需要再方法參數上打上一個 @Valid 註解或者 @Validate 註解,如果我們采用 AOP 去切所有 controller 中的方法的話,那麼我們寫的自定義規則校驗器,甚至連方法參數註解都可以不用打,是不是又更加簡潔瞭呢。

於是,介於上述點,寫瞭一個自定義註解校驗器,包括下面幾個文件:

Valid

這個註解作用於字段上,用於規則校驗。

package com.zyq.utils.valid;
 
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
 
/**
 * 字段校驗註解
 *
 * @author zyqok
 * @since 2022/05/06
 */
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface Valid {
 
    /**
     * 屬性名稱
     */
    String name() default "";
 
    /**
     * 是否可為空
     */
    boolean required() default true;
 
    /**
     * 默認值(如果默認值寫 null 時,則對所有數據類型有效,不會設置默認值)
     */
    String defaultValue() default "";
 
    /**
     * 【String】是否在原來值的基礎上,去掉前後空格
     */
    boolean trim() default true;
 
    /**
     * 【String】最小長度
     */
    int minLength() default 0;
 
    /**
     * 【String】最大長度
     */
    int maxLength() default 255;
 
    /**
     * 【String】自定義正則校驗(該配置為空時則不進行正則校驗)
     */
    String regex() default "";
 
    /**
     * 【Integer】【Long】【Double】范圍校驗最小值(該配置為空時則不進行校驗)
     */
    String min() default "";
 
    /**
     * 【Integer】【Long】【Double】范圍校驗最大值(該配置為空時則不進行校驗)
     */
    String max() default "";
 
}

ValidUtils

自定義規則校驗工具類

package com.zyq.utils.valid;
 
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.regex.Pattern;
 
 
/**
 * 字段校驗註解工具
 *
 * @author zyqok
 * @since 2022/05/05
 */
public class ValidUtils {
 
    /**
     * 校驗對象,獲取校驗結果(單個提示)
     *
     * @param obj 待校驗對象
     * @return null-校驗通過,非null-校驗未通過
     */
    public static <T> String getMsg(T obj) {
        List<String> msgList = getMsgList(obj);
        return msgList.isEmpty() ? null : msgList.get(0);
    }
 
    /**
     * 校驗對象,獲取校驗結果(所有提示)
     *
     * @param obj 待校驗對象
     * @return null-校驗通過,非null-校驗未通過
     */
    public static <T> List<String> getMsgList(T obj) {
        if (Objects.isNull(obj)) {
            return Collections.emptyList();
        }
        Field[] fields = obj.getClass().getDeclaredFields();
        if (fields.length == 0) {
            return Collections.emptyList();
        }
        List<String> msgList = new ArrayList<>();
        for (Field field : fields) {
            // 沒有打校驗註解的字段則不進行校驗
            Valid valid = field.getAnnotation(Valid.class);
            if (Objects.isNull(valid)) {
                continue;
            }
            field.setAccessible(true);
            // String 類型字段校驗
            if (field.getType().isAssignableFrom(String.class)) {
                String msg = validString(obj, field, valid);
                if (Objects.nonNull(msg)) {
                    msgList.add(msg);
                }
                continue;
            }
            // int / Integer 類型字符校驗
            String typeName = field.getType().getTypeName();
            if (field.getType().isAssignableFrom(Integer.class) || "int".equals(typeName)) {
                String msg = validInteger(obj, field, valid);
                if (Objects.nonNull(msg)) {
                    msgList.add(msg);
                }
                continue;
            }
            // double/Double 類型字段校驗
            if (field.getType().isAssignableFrom(Double.class) || "double".equals(typeName)) {
                String msg = validDouble(obj, field, valid);
                if (Objects.nonNull(msg)) {
                    msgList.add(msg);
                }
                continue;
            }
        }
        return msgList;
    }
 
    /**
     * 校驗String類型字段
     */
    private static <T> String validString(T obj, Field field, Valid valid) {
        // 獲取屬性名稱
        String name = getFieldName(field, valid);
        // 獲取原值
        Object v = getValue(obj, field);
        String val = Objects.isNull(v) ? "" : v.toString();
        // 是否需要去掉前後空格
        boolean trim = valid.trim();
        if (trim) {
            val = val.trim();
        }
        // 是否必填
        boolean required = valid.required();
        if (required && val.isEmpty()) {
            return requiredMsg(name);
        }
        // 是否有默認值
        if (val.isEmpty()) {
            val = isDefaultNull(valid) ? null : valid.defaultValue();
        }
        // 最小長度校驗
        int length = 0;
        if (Objects.nonNull(val)) {
            length = val.length();
        }
        if (length < valid.minLength()) {
            return minLengthMsg(name, valid);
        }
        // 最大長度校驗
        if (length > valid.maxLength()) {
            return maxLengthMsg(name, valid);
        }
        // 正則判斷
        if (!valid.regex().isEmpty()) {
            boolean isMatch = Pattern.matches(valid.regex(), val);
            if (!isMatch) {
                return regexMsg(name);
            }
        }
        // 將值重新寫入原字段中
        setValue(obj, field, val);
        // 如果所有校驗通過後,則返回null
        return null;
    }
 
    private static <T> String validInteger(T obj, Field field, Valid valid) {
        // 獲取屬性名稱
        String name = getFieldName(field, valid);
        // 獲取原值
        Object v = getValue(obj, field);
        Integer val = Objects.isNull(v) ? null : (Integer) v;
        // 是否必填
        boolean required = valid.required();
        if (required && Objects.isNull(val)) {
            return requiredMsg(name);
        }
        // 是否有默認值
        if (Objects.isNull(val)) {
            boolean defaultNull = isDefaultNull(valid);
            if (!defaultNull) {
                val = parseInt(valid.defaultValue());
            }
        }
        // 校驗最小值
        if (!valid.min().isEmpty() && Objects.nonNull(val)) {
            int min = parseInt(valid.min());
            if (val < min) {
                return minMsg(name, valid);
            }
        }
        // 校驗最大值
        if (!valid.max().isEmpty() && Objects.nonNull(val)) {
            int max = parseInt(valid.max());
            if (val > max) {
                return maxMsg(name, valid);
            }
        }
        // 將值重新寫入原字段中
        setValue(obj, field, val);
        // 如果所有校驗通過後,則返回null
        return null;
    }
 
    private static <T> String validDouble(T obj, Field field, Valid valid) {
        return null;
    }
 
    /**
     * 獲取對象指定字段的值
     *
     * @param obj   原對象
     * @param field 指定字段
     * @param <T>   泛型
     * @return 該字段的值
     */
    private static <T> Object getValue(T obj, Field field) {
        try {
            return field.get(obj);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
            return null;
        }
    }
 
    /**
     * 給對象指定字段設值,一般校驗後值可能有變化(生成默認值/去掉前後空格等),需要新的值重新設置到對象中
     *
     * @param obj   原對象
     * @param field 指定字段
     * @param val   新值
     * @param <T>   泛型
     */
    private static <T> void setValue(T obj, Field field, Object val) {
        try {
            field.set(obj, val);
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
    }
 
    /**
     * 獲取字段名稱(主要用於錯誤時提示用)
     *
     * @param field 字段對象
     * @param valid 校驗註解
     * @return 字段名稱(如果註解有寫名稱,則取註解名稱;如果沒有註解名稱,則取字段)
     */
    private static String getFieldName(Field field, Valid valid) {
        return valid.name().isEmpty() ? field.getName() : valid.name();
    }
 
    /**
     * 該字段是否默認為 null
     *
     * @param valid 校驗註解
     * @return true - 默認為 null; false - 默認不為 null
     */
    private static boolean isDefaultNull(Valid valid) {
        return "null".equals(valid.defaultValue());
    }
 
    /**
     * 提示信息(該方法用於統一格式化提示信息樣式)
     *
     * @param name 字段名稱
     * @param msg  提示原因
     * @return 提示信息
     */
    private static String msg(String name, String msg) {
        return "【" + name + "】" + msg;
    }
 
    /**
     * 必填字段提示
     *
     * @param name 字段名稱
     * @return 提示信息
     */
    private static String requiredMsg(String name) {
        return msg(name, "不能為空");
    }
 
    /**
     * String 類型字段少於最小長度提示
     *
     * @param name  字段名稱
     * @param valid 校驗註解
     * @return 提示信息
     */
    private static String minLengthMsg(String name, Valid valid) {
        return msg(name, "不能少於" + valid.minLength() + "個字符");
    }
 
    /**
     * String 類型字段超過最大長度提示
     *
     * @param name  字段名稱
     * @param valid 校驗註解
     * @return 提示信息
     */
    private static String maxLengthMsg(String name, Valid valid) {
        return msg(name, "不能超過" + valid.maxLength() + "個字符");
    }
 
    /**
     * String 類型正則校驗提示
     *
     * @param name 字段名稱
     * @return 提示信息
     */
    private static String regexMsg(String name) {
        return msg(name, "填寫格式不正確");
    }
 
    /**
     * 數字類型小於最小值的提示
     *
     * @param name  字段名稱
     * @param valid 校驗註解
     * @return 提示信息
     */
    private static String minMsg(String name, Valid valid) {
        return msg(name, "不能小於" + valid.min());
    }
 
    /**
     * 數字類型大於最大值的提示
     *
     * @param name  字段名稱
     * @param valid 校驗註解
     * @return 提示信息
     */
    private static String maxMsg(String name, Valid valid) {
        return msg(name, "不能大於" + valid.max());
    }
 
    /**
     * 將字符串數字轉化為 int 類型的數字,轉換異常時返回 0
     *
     * @param intStr 字符串數字
     * @return int 類型數字
     */
    private static int parseInt(String intStr) {
        try {
            return Integer.valueOf(intStr);
        } catch (NumberFormatException e) {
            return 0;
        }
    }
}

ValidAop

這是一個 controller 攔截切面,寫瞭這個,就不用再 controller 方法參數上打上類似於原@Valid 和 @Validate 註解,還原的方法參數的原始整潔度。

但需要註意的是:類中 controller 的路徑需要替換為你的包路徑(我這裡 controller 包路徑為com.zyq.controller)。

package com.zyq.aop;
 
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.JSONObject;
import com.unisoc.outsource.config.global.ValidException;
import com.unisoc.outsource.utils.valid.ValidUtils;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
 
import javax.servlet.http.HttpServletRequest;
import java.util.Map;
import java.util.Objects;
 
/**
 * @author zyqok
 * @since 2022/05/05
 */
@Aspect
@Component
public class ValidAop {
 
    private static final String APPLICATION_JSON = "application/json";
 
    // 這裡為你的 controller 包路徑
    @Pointcut("execution(* com.zyqok.controller.*Controller.*(..))")
    public void pointCut() {
    }
 
    @Before("pointCut()")
    public void doBefore(JoinPoint jp) throws ValidException {
        // 獲取所有請求對象
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();
        // 獲取請求類型
        String contentType = request.getHeader("Content-Type");
        String json = null;
        if (contentType != null && contentType.startsWith(APPLICATION_JSON)) {
            // JSON請求體
            json = JSON.toJSONString(jp.getArgs()[0]);
        } else {
            // 鍵值對參數
            json = getParams(request);
        }
        // 獲取請求類對象
        String validClassName = getParamClassName(jp);
        String msg = valid(json, validClassName);
        if (!isEmpty(msg)) {
            throw new ValidException(msg);
        }
    }
 
    /**
     * 獲取方法參數對象名稱
     */
    private String getParamClassName(JoinPoint jp) {
        // 獲取參數對象
        MethodSignature signature = (MethodSignature) jp.getSignature();
        Class<?>[] types = signature.getParameterTypes();
        // 沒有參數則不進行校驗
        if (types == null || types.length == 0) {
            return null;
        }
        // 返回項目中的對象類名
        for (Class<?> clazz : types) {
            if (clazz.getName().startsWith("com.unisoc.outsource")) {
                return clazz.getName();
            }
        }
        return null;
    }
 
    /**
     * 獲取請求對象
     */
    private String getParams(HttpServletRequest request) {
        Map<String, String[]> parameterMap = request.getParameterMap();
        if (Objects.isNull(parameterMap) || parameterMap.isEmpty()) {
            return "{}";
        }
        JSONObject obj = new JSONObject();
        parameterMap.forEach((k, v) -> {
            if (Objects.nonNull(v) && v.length == 1) {
                obj.put(k, v[0]);
            } else {
                obj.put(k, v);
            }
        });
        return obj.toString();
    }
 
    /**
     * 校驗請求值合規性
     */
    private String valid(String json, String className) {
        if (isEmpty(className)) {
            return null;
        }
        System.out.println("json : " + json);
        System.out.println("className : " + className);
        try {
            Class<?> clazz = Class.forName(className);
            Object o = JSON.parseObject(json, clazz);
            return ValidUtils.getMsg(o);
        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }
 
    /**
     * 校驗字符串是否為空
     */
    private boolean isEmpty(String s) {
        return Objects.isNull(s) || s.trim().isEmpty();
    }
}

ValidException

因為 AOP 切面裡,不能在前置切面中直接返回校驗規則的錯誤提示,所以我們可以采用拋異常的方式,最後對異常進行捕捉,再提示給用戶(原 Springboot 的 @Validate 也是采用類似方式進行處理)。

package com.zyq.valid;
 
/**
 * 自定義註解異常
 *
 * @author zyqok
 * @since 2022/05/06
 */
public class ValidException extends RuntimeException {
 
    private String msg;
 
    public String getMsg() {
        return msg;
    }
 
    public void setMsg(String msg) {
        this.msg = msg;
    }
 
    public ValidException(String msg) {
        this.msg = msg;
    }
}

ValidExceptionHandler

這個異常處理器就是用於捕捉上面的異常,最後提示給前端。

@ControllerAdvice
@ResponseBody
public class ValidExceptionHandler {
 
    @ExceptionHandler(ValidException.class)
    public Map<String, String> validExceptionHandler(ValidException ex) {
        Map<String, String> map = new HashMap();
        map.put("code", 1);
        map.put("msg", ex.getMsg());
        return map;
    }
 
}

當把所有文件復制到文件中後,那麼在使用的時候

隻需要將方法中的參數打上我們定義的 @Valid 即可,其餘不用做任何操作就OK

/**
 * @author zyqok
 * @since 2022/05/06
 */
@Data
public class EntryApplyCancelReq {
 
    @Valid
    private Integer id;
 
    @Valid(name = "取消原因", maxLength = 50)
    private String reason;
 
}

到此這篇關於手寫一個@Valid字段校驗器的示例代碼的文章就介紹到這瞭,更多相關@Valid字段校驗器內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

推薦閱讀: