SpringBoot實現各種引數校驗,寫得太好了,建議收藏!

👉 這是一個或許對你有用的社群
🐱 一對一交流/面試小冊/簡歷最佳化/求職解惑,歡迎加入芋道快速開發平臺知識星球。下面是星球提供的部分資料:
👉這是一個或許對你有用的開源專案
國產 Star 破 10w+ 的開源專案,前端包括管理後臺 + 微信小程式,後端支援單體和微服務架構。
功能涵蓋 RBAC 許可權、SaaS 多租戶、資料許可權、商城、支付、工作流、大屏報表、微信公眾號、CRM 等等功能:
  • Boot 倉庫:https://gitee.com/zhijiantianya/ruoyi-vue-pro
  • Cloud 倉庫:https://gitee.com/zhijiantianya/yudao-cloud
  • 影片教程:https://doc.iocoder.cn
【國內首批】支援 JDK 21 + SpringBoot 3.2.2、JDK 8 + Spring Boot 2.7.18 雙版本 

之前也寫過一篇關於Spring Validation使用的文章,不過自我感覺還是浮於表面,本次打算徹底搞懂Spring Validation。本文會詳細介紹Spring Validation各種場景下的最佳實踐及其實現原理,死磕到底!
專案原始碼:https://github.com/chentianming11/spring-validation

簡單使用

Java API規範(JSR303)定義了Bean校驗的標準validation-api,但沒有提供實現。hibernate validation是對這個規範的實現,並增加了校驗註解如@Email、@Length等。
Spring Validation是對hibernate validation的二次封裝,用於支援spring mvc引數自動校驗。接下來,我們以spring-boot專案為例,介紹Spring Validation的使用。

引入依賴

如果spring-boot版本小於2.3.x,spring-boot-starter-web會自動傳入hibernate-validator依賴。如果spring-boot版本大於2.3.x,則需要手動引入依賴:
<dependency><groupId>org.hibernate</groupId><artifactId>hibernate-validator</artifactId><version>6.0.1.Final</version></dependency>
對於web服務來說,為防止非法引數對業務造成影響,在Controller層一定要做引數校驗的!大部分情況下,請求引數分為如下兩種形式:
  • POST、PUT請求,使用requestBody傳遞引數;
  • GET請求,使用requestParam/PathVariable傳遞引數。
下面我們簡單介紹下requestBody和requestParam/PathVariable的引數校驗實戰!

requestBody引數校驗

POST、PUT請求一般會使用requestBody傳遞引數,這種情況下,後端使用DTO物件進行接收。只要給DTO物件加上@Validated註解就能實現自動引數校驗。比如,有一個儲存User的介面,要求userName長度是2-10,account和password欄位長度是6-20。
如果校驗失敗,會丟擲MethodArgumentNotValidException異常,Spring預設會將其轉為400(Bad Request)請求。
DTO表示資料傳輸物件(Data Transfer Object),用於伺服器和客戶端之間互動傳輸使用的。在spring-web專案中可以表示用於接收請求引數的Bean物件。
在DTO欄位上宣告約束註解
@DatapublicclassUserDTO{private Long userId;@NotNull@Length(min = 2, max = 10)private String userName;@NotNull@Length(min = 6, max = 20)private String account;@NotNull@Length(min = 6, max = 20)private String password;}
在方法引數上宣告校驗註解
@PostMapping("/save")public Result saveUser(@RequestBody @Validated UserDTO userDTO){// 校驗透過,才會執行業務邏輯處理return Result.ok();}
這種情況下,使用@Valid和@Validated都可以。

requestParam/PathVariable引數校驗

GET請求一般會使用requestParam/PathVariable傳參。如果引數比較多(比如超過6個),還是推薦使用DTO物件接收。
否則,推薦將一個個引數平鋪到方法入參中。在這種情況下,必須在Controller類上標註@Validated註解,並在入參上宣告約束註解(如@Min等)。如果校驗失敗,會丟擲ConstraintViolationException異常。
程式碼示例如下:
@RequestMapping("/api/user")@RestController@ValidatedpublicclassUserController{// 路徑變數@GetMapping("{userId}")public Result detail(@PathVariable("userId") @Min(10000000000000000L) Long userId) {// 校驗透過,才會執行業務邏輯處理        UserDTO userDTO = new UserDTO();        userDTO.setUserId(userId);        userDTO.setAccount("11111111111111111");        userDTO.setUserName("xixi");        userDTO.setAccount("11111111111111111");return Result.ok(userDTO);    }// 查詢引數@GetMapping("getByAccount")public Result getByAccount(@Length(min = 6, max = 20) @NotNull String  account) {// 校驗透過,才會執行業務邏輯處理        UserDTO userDTO = new UserDTO();        userDTO.setUserId(10000000000000003L);        userDTO.setAccount(account);        userDTO.setUserName("xixi");        userDTO.setAccount("11111111111111111");return Result.ok(userDTO);    }}

統一異常處理

前面說過,如果校驗失敗,會丟擲MethodArgumentNotValidException或者ConstraintViolationException異常。在實際專案開發中,通常會用統一異常處理來返回一個更友好的提示。
比如我們系統要求無論傳送什麼異常,http的狀態碼必須返回200,由業務碼去區分系統的異常情況。
@RestControllerAdvicepublicclassCommonExceptionHandler{@ExceptionHandler({MethodArgumentNotValidException.class})    @ResponseStatus(HttpStatus.OK)    @ResponseBodypublicResulthandleMethodArgumentNotValidException(MethodArgumentNotValidExceptionex{        BindingResult bindingResult = ex.getBindingResult();        StringBuilder sb = new StringBuilder("校驗失敗:");for (FieldError fieldError : bindingResult.getFieldErrors()) {            sb.append(fieldError.getField()).append(":").append(fieldError.getDefaultMessage()).append(", ");        }        String msg = sb.toString();return Result.fail(BusinessCode.引數校驗失敗, msg);    }@ExceptionHandler({ConstraintViolationException.class})    @ResponseStatus(HttpStatus.OK)    @ResponseBodypublicResulthandleConstraintViolationException(ConstraintViolationExceptionex{return Result.fail(BusinessCode.引數校驗失敗, ex.getMessage());    }}
基於 Spring Boot + MyBatis Plus + Vue & Element 實現的後臺管理系統 + 使用者小程式,支援 RBAC 動態許可權、多租戶、資料許可權、工作流、三方登入、支付、簡訊、商城等功能
  • 專案地址:https://github.com/YunaiV/ruoyi-vue-pro
  • 影片教程:https://doc.iocoder.cn/video/

進階使用

分組校驗

在實際專案中,可能多個方法需要使用同一個DTO類來接收引數,而不同方法的校驗規則很可能是不一樣的。這個時候,簡單地在DTO類的欄位上加約束註解無法解決這個問題。因此,spring-validation支援了分組校驗的功能,專門用來解決這類問題。
還是上面的例子,比如儲存User的時候,UserId是可空的,但是更新User的時候,UserId的值必須>=10000000000000000L;其它欄位的校驗規則在兩種情況下一樣。這個時候使用分組校驗的程式碼示例如下:
約束註解上宣告適用的分組資訊groups
@DatapublicclassUserDTO{@Min(value = 10000000000000000L, groups = Update.class)privateLonguserId;@NotNull(groups = {Save.classUpdate.class})    @Length(min2, max = 10, groups = {Save.classUpdate.class})privateStringuserName;@NotNull(groups = {Save.classUpdate.class})    @Length(min6, max = 20, groups = {Save.classUpdate.class})privateStringaccount;@NotNull(groups = {Save.classUpdate.class})    @Length(min6, max = 20, groups = {Save.classUpdate.class})privateStringpassword;/**     * 儲存的時候校驗分組     */publicinterfaceSave{    }/**     * 更新的時候校驗分組     */publicinterfaceUpdate{    }}
@Validated註解上指定校驗分組
@PostMapping("/save")public Result saveUser(@RequestBody @Validated(UserDTO.Save.class) UserDTO userDTO) {// 校驗透過,才會執行業務邏輯處理return Result.ok();}@PostMapping("/update")public Result updateUser(@RequestBody @Validated(UserDTO.Update.class) UserDTO userDTO) {// 校驗透過,才會執行業務邏輯處理return Result.ok();}

巢狀校驗

前面的示例中,DTO類裡面的欄位都是基本資料型別和String型別。但是實際場景中,有可能某個欄位也是一個物件,這種情況先,可以使用巢狀校驗。
比如,上面儲存User資訊的時候同時還帶有Job資訊。需要注意的是,此時DTO類的對應欄位必須標記@Valid註解。
@DatapublicclassUserDTO{@Min(value = 10000000000000000L, groups = Update.class)privateLonguserId;@NotNull(groups = {Save.classUpdate.class})    @Length(min2, max = 10, groups = {Save.classUpdate.class})privateStringuserName;@NotNull(groups = {Save.classUpdate.class})    @Length(min6, max = 20, groups = {Save.classUpdate.class})privateStringaccount;@NotNull(groups = {Save.classUpdate.class})    @Length(min6, max = 20, groups = {Save.classUpdate.class})privateStringpassword;@NotNull(groups = {Save.classUpdate.class})    @ValidprivateJobjob;@DatapublicstaticclassJob{@Min(value = 1, groups = Update.class)privateLongjobId;@NotNull(groups = {Save.classUpdate.class})        @Length(min2, max = 10, groups = {Save.classUpdate.class})privateStringjobName;@NotNull(groups = {Save.classUpdate.class})        @Length(min2, max = 10, groups = {Save.classUpdate.class})privateStringposition;    }/**     * 儲存的時候校驗分組     */publicinterfaceSave{    }/**     * 更新的時候校驗分組     */publicinterfaceUpdate{    }}
巢狀校驗可以結合分組校驗一起使用。還有就是巢狀集合校驗會對集合裡面的每一項都進行校驗,例如List<Job>欄位會對這個list裡面的每一個Job物件都進行校驗

集合校驗

如果請求體直接傳遞了json陣列給後臺,並希望對陣列中的每一項都進行引數校驗。此時,如果我們直接使用java.util.Collection下的list或者set來接收資料,引數校驗並不會生效!我們可以使用自定義list集合來接收引數:
包裝List型別,並宣告@Valid註解
publicclassValidationList<EimplementsList<E{@Delegate// @Delegate是lombok註解@Valid// 一定要加@Valid註解public List<E> list = new ArrayList<>();// 一定要記得重寫toString方法@Overridepublic String toString(){return list.toString();    }}
@Delegate註解受lombok版本限制,1.18.6以上版本可支援。如果校驗不透過,會丟擲NotReadablePropertyException,同樣可以使用統一異常進行處理。
比如,我們需要一次性儲存多個User物件,Controller層的方法可以這麼寫:
@PostMapping("/saveList")public Result saveList(@RequestBody @Validated(UserDTO.Save.class) ValidationList<UserDTO> userList) {// 校驗透過,才會執行業務邏輯處理return Result.ok();}

自定義校驗

業務需求總是比框架提供的這些簡單校驗要複雜的多,我們可以自定義校驗來滿足我們的需求。
自定義spring validation非常簡單,假設我們自定義加密id(由數字或者a-f的字母組成,32-256長度)校驗,主要分為兩步:
自定義約束註解
@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER})@Retention(RUNTIME)@Documented@Constraint(validatedBy = {EncryptIdValidator.class})public @interfaceEncryptId{// 預設錯誤訊息String message()default "加密id格式錯誤";// 分組    Class<?>[] groups() default {};// 負載    Class<? extends Payload>[] payload() default {};}
實現ConstraintValidator介面編寫約束校驗器
publicclassEncryptIdValidatorimplementsConstraintValidator<EncryptIdString{privatestaticfinal Pattern PATTERN = Pattern.compile("^[a-f\\d]{32,256}$");@OverridepublicbooleanisValid(String value, ConstraintValidatorContext context){// 不為null才進行校驗if (value != null) {            Matcher matcher = PATTERN.matcher(value);return matcher.find();        }returntrue;    }}
這樣我們就可以使用@EncryptId進行引數校驗了!

程式設計式校驗

上面的示例都是基於註解來實現自動校驗的,在某些情況下,我們可能希望以程式設計方式呼叫驗證。這個時候可以注入javax.validation.Validator物件,然後再呼叫其api。
@Autowiredprivate javax.validation.Validator globalValidator;// 程式設計式校驗@PostMapping("/saveWithCodingValidate")public Result saveWithCodingValidate(@RequestBody UserDTO userDTO){    Set<ConstraintViolation<UserDTO>> validate = globalValidator.validate(userDTO, UserDTO.Save.class);// 如果校驗透過,validate為空;否則,validate包含未校驗透過項if (validate.isEmpty()) {// 校驗透過,才會執行業務邏輯處理    } else {for (ConstraintViolation<UserDTO> userDTOConstraintViolation : validate) {// 校驗失敗,做其它邏輯            System.out.println(userDTOConstraintViolation);        }    }return Result.ok();}

快速失敗(Fail Fast)

Spring Validation預設會校驗完所有欄位,然後才丟擲異常。可以透過一些簡單的配置,開啟Fali Fast模式,一旦校驗失敗就立即返回。
@Beanpublic Validator validator(){    ValidatorFactory validatorFactory = Validation.byProvider(HibernateValidator.class)            .configure()            // 快速失敗模式            .failFast(true)            .buildValidatorFactory();return validatorFactory.getValidator();}

@Valid和@Validated區別

基於 Spring Cloud Alibaba + Gateway + Nacos + RocketMQ + Vue & Element 實現的後臺管理系統 + 使用者小程式,支援 RBAC 動態許可權、多租戶、資料許可權、工作流、三方登入、支付、簡訊、商城等功能
  • 專案地址:https://github.com/YunaiV/yudao-cloud
  • 影片教程:https://doc.iocoder.cn/video/

實現原理

requestBody引數校驗實現原理

在spring-mvc中,RequestResponseBodyMethodProcessor是用於解析@RequestBody標註的引數以及處理@ResponseBody標註方法的返回值的。顯然,執行引數校驗的邏輯肯定就在解析引數的方法resolveArgument()中:
publicclassRequestResponseBodyMethodProcessorextendsAbstractMessageConverterMethodProcessor{@Overridepublic Object resolveArgument(MethodParameter parameter, @Nullable ModelAndViewContainer mavContainer,                                  NativeWebRequest webRequest, @Nullable WebDataBinderFactory binderFactory)throws Exception {        parameter = parameter.nestedIfOptional();//將請求資料封裝到DTO物件中        Object arg = readWithMessageConverters(webRequest, parameter, parameter.getNestedGenericParameterType());        String name = Conventions.getVariableNameForParameter(parameter);if (binderFactory != null) {            WebDataBinder binder = binderFactory.createBinder(webRequest, arg, name);if (arg != null) {// 執行資料校驗                validateIfApplicable(binder, parameter);if (binder.getBindingResult().hasErrors() && isBindExceptionRequired(binder, parameter)) {thrownew MethodArgumentNotValidException(parameter, binder.getBindingResult());                }            }if (mavContainer != null) {                mavContainer.addAttribute(BindingResult.MODEL_KEY_PREFIX + name, binder.getBindingResult());            }        }return adaptArgumentIfNecessary(arg, parameter);    }}
可以看到,resolveArgument()呼叫了validateIfApplicable()進行引數校驗。
protectedvoidvalidateIfApplicable(WebDataBinder binder, MethodParameter parameter){// 獲取引數註解,比如@RequestBody、@Valid、@Validated    Annotation[] annotations = parameter.getParameterAnnotations();for (Annotation ann : annotations) {// 先嚐試獲取@Validated註解        Validated validatedAnn = AnnotationUtils.getAnnotation(ann, Validated.class);//如果直接標註了@Validated,那麼直接開啟校驗。//如果沒有,那麼判斷引數前是否有Valid起頭的註解。if (validatedAnn != null || ann.annotationType().getSimpleName().startsWith("Valid")) {            Object hints = (validatedAnn != null ? validatedAnn.value() : AnnotationUtils.getValue(ann));            Object[] validationHints = (hints instanceof Object[] ? (Object[]) hints : new Object[] {hints});//執行校驗            binder.validate(validationHints);break;        }    }}
看到這裡,大家應該能明白為什麼這種場景下@Validated、@Valid兩個註解可以混用。我們接下來繼續看WebDataBinder.validate()實現。
@Overridepublicvoidvalidate(Object target, Errors errors, Object... validationHints){if (this.targetValidator != null) {        processConstraintViolations(//此處呼叫Hibernate Validator執行真正的校驗this.targetValidator.validate(target, asValidationGroups(validationHints)), errors);    }}
最終發現底層最終還是呼叫了Hibernate Validator進行真正的校驗處理。

方法級別的引數校驗實現原理

上面提到的將引數一個個平鋪到方法引數中,然後在每個引數前面宣告約束註解的校驗方式,就是方法級別的引數校驗。
實際上,這種方式可用於任何Spring Bean的方法上,比如Controller/Service等。其底層實現原理就是AOP,具體來說是透過MethodValidationPostProcessor動態註冊AOP切面,然後使用MethodValidationInterceptor對切點方法織入增強。
publicclassMethodValidationPostProcessorextendsAbstractBeanFactoryAwareAdvisingPostProcessorimplementsInitializingBean{@OverridepublicvoidafterPropertiesSet(){//為所有`@Validated`標註的Bean建立切面        Pointcut pointcut = new AnnotationMatchingPointcut(this.validatedAnnotationType, true);//建立Advisor進行增強this.advisor = new DefaultPointcutAdvisor(pointcut, createMethodValidationAdvice(this.validator));    }//建立Advice,本質就是一個方法攔截器protected Advice createMethodValidationAdvice(@Nullable Validator validator){return (validator != null ? new MethodValidationInterceptor(validator) : new MethodValidationInterceptor());    }}
接著看一下MethodValidationInterceptor:
publicclassMethodValidationInterceptorimplementsMethodInterceptor{@Overridepublic Object invoke(MethodInvocation invocation)throws Throwable {//無需增強的方法,直接跳過if (isFactoryBeanMetadataMethod(invocation.getMethod())) {return invocation.proceed();        }//獲取分組資訊        Class<?>[] groups = determineValidationGroups(invocation);        ExecutableValidator execVal = this.validator.forExecutables();        Method methodToValidate = invocation.getMethod();        Set<ConstraintViolation<Object>> result;try {//方法入參校驗,最終還是委託給Hibernate Validator來校驗            result = execVal.validateParameters(                invocation.getThis(), methodToValidate, invocation.getArguments(), groups);        }catch (IllegalArgumentException ex) {            ...        }//有異常直接丟擲if (!result.isEmpty()) {thrownew ConstraintViolationException(result);        }//真正的方法呼叫        Object returnValue = invocation.proceed();//對返回值做校驗,最終還是委託給Hibernate Validator來校驗        result = execVal.validateReturnValue(invocation.getThis(), methodToValidate, returnValue, groups);//有異常直接丟擲if (!result.isEmpty()) {thrownew ConstraintViolationException(result);        }return returnValue;    }}
實際上,不管是requestBody引數校驗還是方法級別的校驗,最終都是呼叫Hibernate Validator執行校驗,Spring Validation只是做了一層封裝。

歡迎加入我的知識星球,全面提升技術能力。
👉 加入方式,長按”或“掃描”下方二維碼噢
星球的內容包括:專案實戰、面試招聘、原始碼解析、學習路線。

文章有幫助的話,在看,轉發吧。
謝謝支援喲 (*^__^*)

相關文章