Spring MVC中参数校验器的使用和自定义参数校验器

小灰灰 2022-05-16 15:57 237阅读 0赞

# Spring MVC中参数校验器的使用和自定义参数校验器 #

项目案例代码：[https://github.com/IndustriousSnail/learning-notes/tree/master/springboot/SpringMvc自定义参数校验器/validator][https_github.com_IndustriousSnail_learning-notes_tree_master_springboot_SpringMvc_validator]

*  一、参数校验器简介
 *  二、自定义参数校验器案例
    
     *  2.1 场景介绍
     *  2.2 参数校验具体实现
        
         *  2.2.1 为LoginVO增加注解
         *  2.2.2 在需要校验的VO类型增加@Valid注解
         *  2.2.3 自定义校验注解
         *  2.2.4 校验器的具体实现
         *  2.2.5 校验失败统一异常处理
         *  2.2.6 验证结果

## 一、参数校验器简介 ##

在Http请求中会传递一些参数，对于这些参数，后台需要经过一些校验，如果校验不通过，就没有必要进行接下来的处理。  
比如，验证账号密码是否为空等等。

Spring为我们提供了一些默认的参数校验器，比如@NotNull, @Length 等。  
但是这些并不能满足所有需求，所以需要自定义参数校验器。

## 二、自定义参数校验器案例 ##

### 2.1 场景介绍 ###

假设我们现在需要对登录的“用户名”和“密码进行校验”。 我们现在有如下Controller:

@RestController
    public class LoginController {
    
        @GetMapping("/login")
        public Boolean login(LoginVO loginVO){
            //todo 登录处理 ....
            return true;
        }
    }
    
    // public class LoginVO {
    //     private String username;
    //     private String password;
    // }

我们需要对 loginVO 中的 username 和 password 字段进行校验。  
如果在login()方法中进行该逻辑的编写，也可以，但是如果还存在其他的方法（如修改密码）需要对loginVO进行校验，那么就又要再写一遍。  
这样就出现了重复逻辑。所以需要对参数进行统一校验。

该案例，以判断password是否同时包含大小写为例。进行自定义参数校验器的实现。

### 2.2 参数校验具体实现 ###

#### 2.2.1 为LoginVO增加注解 ####

@NotNull  //参数校验，不能为空
    private String username;
    
    @NotNull
    @Length(min = 6)  //长度不小于6
    @IsComplex //增加自定义的验证器
    private String password;

#### 2.2.2 在需要校验的VO类型增加@Valid注解 ####

//增加@Valid注解，校验loginVO对象
    public Boolean login(@Valid LoginVO loginVO){}

> 此时重启程序，发现如果传递的参数不对，会返回400错误。说明参数校验已经生效。

#### 2.2.3 自定义校验注解 ####

@Target({ElementType.FIELD})  //Target注解描述了该注解需要可以修饰哪些内容。 这里指明只能修饰字段。
    @Retention(RetentionPolicy.RUNTIME)  //Retention注解描述了该注解是在什么期间生效的。 这里指明是运行期生效。
    @Documented   //表明这个注解应该被 javadoc工具记录
    @Constraint(validatedBy = {IsComplexValidator.class})  //@Constraint注解就说明这个字段是要被经过验证的。 validateBy参数填入需要经过的验证器类。
    public @interface IsComplex {
    
        String message() default "密码不够复杂"; //当校验不通过时的提示信息。 必须有
    
        Class<?>[] groups() default { };  //必须有
    
        Class<? extends Payload>[] payload() default { }; //必须有
    }

> 定义该校验注解，使用该注解修饰字段。

#### 2.2.4 校验器的具体实现 ####

/**
     * @Description: IsComplex校验器的具体实现。
     * @implSpec ConstraintValidator<A extends Annotation, T>: A：注解， T：A注解修改的类型
     */
    public class IsComplexValidator implements ConstraintValidator<IsComplex, String> {
    
        /**
         * 初始化， 整个生命周期只会执行一次，在第一次验证时执行。
         * @param constraintAnnotation
         */
        @Override
        public void initialize(IsComplex constraintAnnotation) {
            System.out.println("initialize");
        }
    
        /**
         * 通过该方法，对参数进行验证，看是否通过。
         * @param value 修饰字段的值。
         * @param context 上下文
         * @return true：验证通过。 false：验证不通过。
         */
        @Override
        public boolean isValid(String value, ConstraintValidatorContext context) {
            //判断密码是否同时包含大小写
            boolean hasUpper=false,hasLower=false;
            for(byte b: value.getBytes()){
                if('A'<=b&&b<='Z') hasUpper=true;
                if('a'<=b&&b<='z') hasLower=true;
            }
            return hasUpper&&hasLower;  //如果同时包含大小写，则验证通过。
        }
    }

> 1.  实现ConstraintValidator接口，重写其方法。
> 2.  在isValid写具体校验逻辑。通过返回true,不通过返回false

#### 2.2.5 校验失败统一异常处理 ####

当校验失败时，会抛出BindException异常，http会响应400。 如果要做统一异常处理，可以使用ControllerAdvice捕获异常。

@ControllerAdvice
    @ResponseBody
    public class ValidatorExceptionHandler{
    
        /**
         * 使用@ExceptionHandler注解，处理BindException异常。该异常是spring提供的参数校验失败时抛出的异常。
         * @param e 异常对象
         * @return 返回的http内容
         */
        @ExceptionHandler(value = BindException.class)
        public String exceptionHandler(BindException e){
            return e.getAllErrors().get(0).getDefaultMessage();  //参数校验会抛出多个异常，这里取第一个。而它的默认信息就是IsComplex注解中定义的message的默认值。
        }
    }

#### 2.2.6 验证结果 ####

此时在浏览器中输入

http://localhost:8080/login?username=1111&password=AAAAA

可以看到浏览器打印的是“长度需要在6和2147483647之间”。  
这个信息是@Length注解的message信息。

输入:

http://localhost:8080/login?username=1111&password=AAAAAAA

浏览器打印“密码不够复杂”

输入

http://localhost:8080/login?username=1111&password=AAAAAAAaaa

浏览器打印“true”

[https_github.com_IndustriousSnail_learning-notes_tree_master_springboot_SpringMvc_validator]: https://github.com/IndustriousSnail/learning-notes/tree/master/springboot/SpringMvc%E8%87%AA%E5%AE%9A%E4%B9%89%E5%8F%82%E6%95%B0%E6%A0%A1%E9%AA%8C%E5%99%A8/validator