SpringBoot 使用 Validation 进行参数校验并统一返回校验异常

怼烎@ 2023-09-27 22:32 126阅读 0赞

> 在 SpringBoot项目开发中，有一个观点是不要相信前端传入的参数，因为你不知道用户是怎么操作我们接口的，所以在后端也需要对参数进行校验，这篇文章主要讲讲我们项目中最常使用的验证方案。

## 一、引入相应的依赖 ##

pom复制代码<dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-validation</artifactId>
    </dependency>

spring-boot-starter-validation本质是使用的Hibernate Validator，它并没有自己的实现。

## 二、Validation的基本校验注解 ##

less复制代码空检查
    @Null			验证对象是否为null
    @NotNull		验证对象是否不为null, 无法查检长度为0的字符串
    @NotBlank		检查约束字符串是不是Null还有被Trim的长度是否大于0,只对字符串,且会去掉前后空格.
    @NotEmpty		检查约束元素是否为NULL或者是EMPTY. 
    
    Booelan检查
    @AssertTrue		验证 Boolean 对象是否为 true  
    @AssertFalse	验证 Boolean 对象是否为 false  
    
    长度检查
    @Size(min=, max=)		验证对象（Array,Collection,Map,String）长度是否在给定的范围之内  
    @Length(min=, max=)		验证注解的元素值长度在min和max区间内
    
    日期检查
    @Past		验证 Date 和 Calendar 对象是否在当前时间之前  
    @Future		验证 Date 和 Calendar 对象是否在当前时间之后  
    @Pattern	验证 String 对象是否符合正则表达式的规则
    
    数值检查，建议使用在Stirng,Integer类型，不建议使用在int类型上，因为表单值为“”时无法转换为int，但可以转换为Stirng为"",Integer为null
    @Min			验证 Number 和 String 对象是否大等于指定的值  
    @Max			验证 Number 和 String 对象是否小等于指定的值  
    @DecimalMax		被标注的值必须不大于约束中指定的最大值. 这个约束的参数是一个通过BigDecimal定义的最大值的字符串表示.小数存在精度
    @DecimalMin		被标注的值必须不小于约束中指定的最小值. 这个约束的参数是一个通过BigDecimal定义的最小值的字符串表示.小数存在精度
    @Digits			验证 Number 和 String 的构成是否合法  
    @Digits(integer=,fraction=)		验证字符串是否是符合指定格式的数字，interger指定整数精度，fraction指定小数精度。
    
    @Range(min=, max=)	验证注解的元素值在最小值和最大值之间
    @Range(min=10000,max=50000,message="range.bean.wage")
    
    @Valid 写在方法参数前，递归的对该对象进行校验, 如果关联对象是个集合或者数组,那么对其中的元素进行递归校验,如果是一个map,则对其中的值部分进行校验.(是否进行递归验证)
    @CreditCardNumber信用卡验证
    @Email  验证是否是邮件地址，如果为null,不进行验证，算通过验证。
    @ScriptAssert(lang= ,script=, alias=)
    @URL(protocol=,host=, port=,regexp=, flags=)

## 三、添加参数校验 ##

在我们对应的DTO上并在controller的上添加校验。

1.  在DTO的属性上添加校验

java复制代码import com.fasterxml.jackson.annotation.JsonIgnore;
    import lombok.AllArgsConstructor;
    import lombok.Data;
    import lombok.NoArgsConstructor;
    import org.hibernate.validator.constraints.Length;
    
    import javax.validation.constraints.Email;
    import javax.validation.constraints.Max;
    import javax.validation.constraints.Min;
    import javax.validation.constraints.NotBlank;
    import java.util.UUID;
    
    @Data
    @AllArgsConstructor
    @NoArgsConstructor
    public class registryUserDto {
        @NotBlank(message = "用户名不能为空")
        private String username;
        @NotBlank(message = "密码不能为空")
        @Length(min = 6, max = 20, message = "密码长度在6-20之间")
        private String password;
        @Min(value = 0, message = "年龄最小为0")
        @Max(value = 200, message = "年龄最大为200")
        private Integer age;
        @NotBlank(message = "邮箱不能为空")
        @Email(message = "邮箱格式不正确")
        private String email;
        @JsonIgnore
        private String salt = UUID.randomUUID().toString().replaceAll("-", "");
        private Boolean admin;
    }

通过在参数上添加各种校验注解实现校验

1.  在controller对应的DTO添加@Valid或者@Validated

java复制代码@PostMapping("/registry")
    public ResponseResult registryUser(@RequestBody @Valid registryUserDto registryUserDto) {
        return ResponseResult.okResult(registryUserDto);
    }

这样添加后就可以对其中的参数实现校验了，当校验失败时接口就会返回500异常和相应的异常信息。

![format_png][]

对于复杂String校验我们可以使用正则来校验，如下所示：

less复制代码@Pattern(regexp = "^1(3|4|5|7|8)\d{9}$",message = "手机号码格式错误")
    @NotBlank(message = "手机号码不能为空")
    private String phone;

另外对于单个参数的校验，没有用DTO对象来接收的参数也可以校验，先在controller类上添加@Validated，再在对应的参数前加校验注解，如下所示：

less复制代码@RestController
    @RequestMapping("/user")
    @Validated
    public class UserController {
        @PostMapping("/registry")
        public ResponseResult registryUser(@NotBlank(message = "name不能为空") String name) {
            return ResponseResult.okResult(name);
        }
    }

## 四、自定义校验注解 ##

对于一些常见的或复杂的校验需要我们需要自定义校验注解，实现如下：

1.  新建自定义注解

annotation.validator.Status:

java复制代码import com.jk.validator.StatusValidator;
    
    import javax.validation.Constraint;
    import javax.validation.Payload;
    import java.lang.annotation.*;
    
    @Target({ElementType.FIELD, ElementType.PARAMETER})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    @Constraint(validatedBy = {StatusValidator.class})
    public @interface Status {
    
        String[] statusType() default {};
    
        String message() default "状态传递有误";
    
        Class<?>[] groups() default {};
    
        Class<? extends Payload>[] payload() default {};
    
    }

1.  实现相应的校验

java复制代码import com.jk.annotation.validator.Status;
    
    import javax.validation.ConstraintValidator;
    import javax.validation.ConstraintValidatorContext;
    import java.util.Arrays;
    import java.util.List;
    
    public class StatusValidator implements ConstraintValidator<Status, Integer> {
    
        private List<String> typeStatus ;
    
        @Override
        public void initialize(Status constraintAnnotation) {
            typeStatus = Arrays.asList(constraintAnnotation.statusType());
            ConstraintValidator.super.initialize(constraintAnnotation);
        }
    
        @Override
        public boolean isValid(Integer value, ConstraintValidatorContext constraintValidatorContext) {
            if(value !=null){
                if(!typeStatus.contains(String.valueOf(value))){
                    return false;
                }
            }
            return true;
        }
    }

1.  自定义注解的使用

java复制代码@Status(statusType = {"1", "2"})
    private Integer status;

## 五、校验失败统一异常处理 ##

大家可以看到我们上面校验失败的响应msg非常不友好，有很多前端不需要知道的消息。

我们基于全局统一异常处理来添加校验失败的处理，不太清楚统一异常处理的可以看我另一篇文章SpringBoot 中统一响应格式及统一异常处理，你应该这样做

在统一异常处理中添加BindException的处理

exception.GlobalExceptionHandler:

kotlin复制代码import com.jk.enums.AppHttpCodeEnum;
    import com.jk.exception.SystemException;
    import com.jk.domain.vo.ResponseResult;
    import lombok.extern.slf4j.Slf4j;
    import org.springframework.validation.BindException;
    import org.springframework.web.bind.annotation.ExceptionHandler;
    import org.springframework.web.bind.annotation.RestControllerAdvice;
    
    @RestControllerAdvice
    @Slf4j
    public class GlobalExceptionHandler {
        @ExceptionHandler(SystemException.class)
        public ResponseResult systemExceptionHandler(SystemException e) {
            log.error("出现了异常! {}", e);
            return ResponseResult.errorResult(e.getCode(), e.getMsg());
        }
        @ExceptionHandler(Exception.class)
        public ResponseResult exceptionHandler(Exception e) {
            log.error("出现了异常! {}", e);
            return ResponseResult.errorResult(AppHttpCodeEnum.SYSTEM_ERROR.getCode(), e.getMessage());
        }
        /* 添加校验参数异常处理 */
        @ExceptionHandler(BindException.class)
        public ResponseResult bindExceptionHandler(BindException e) {
            log.error("出现了异常! {}", e);
            return ResponseResult.errorResult(AppHttpCodeEnum.SYSTEM_ERROR, e.getBindingResult().getAllErrors().get(0).getDefaultMessage());
        }
    }

校验失败时前端接收的到的响应如下：

![format_png 1][]

可以看到异常信息非常友好，也非常方便前端弹出消息框提示！

> 这样就在SpringBoot的项目中添加了参数校验及统一异常处理，其实整体非常简单，也希望大家在项目中用起来！

[format_png]: https://img-blog.csdnimg.cn/img_convert/866385fd867c325fa541e5dbc194137e.webp?x-oss-process=image/format,png
[format_png 1]: https://img-blog.csdnimg.cn/img_convert/2971c249e643be6ac449fb300ede2d03.webp?x-oss-process=image/format,png