SpringBoot 使用 Hibernate-Validato

作者: 小道萧兮 | 来源:发表于2020-09-02 20:22 被阅读0次

在接口服务中，会有各种各样的入参，我们不可能完全不做任何校验就直接进入到业务处理的环节，通常会有一个基础的数据验证的机制，待这些验证过程完毕，结果无误后，参数才会进入到正式的业务处理中。

而数据验证又分为两种，一种是无业务关联的规则性验证，一种是根据现有数据进行的联动性数据验证。而 Hibernate-Validator 则适合做无业务关联的规则性验证，而这类验证的代码大多是可复用的。

一、基本使用

1、首先需要导入 maven 依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
    <groupId>org.hibernate.validator</groupId>
    <artifactId>hibernate-validator</artifactId>
    <version>6.1.5.Final</version>
</dependency>

2、添加一个普通的 Get 接口，参数 name 的长度需要在 1~10 之间，参数 age 的最小值为 0。

import org.springframework.validation.annotation.Validated;
import javax.validation.constraints.Min;
import javax.validation.constraints.Size;

@RestController
@Validated
public class UserController {

    @GetMapping("/validate1")
    public String validate1(@Size(min = 1, max = 10, message = "姓名长度必须为1到10") @RequestParam("name") String name,
                            @Min(value = 0, message = "年龄须大于0") @RequestParam("age") Integer age) {
        return "validate1";
    }
}

注意，Get 请求时，类名需要加注解 @Validated 才能对参数进行校验。当校验失败时，会抛出 ConstraintViolationException 异常。

3、添加 Post 接口。首先给 Model 类增加校验注解。

@Data
public class LoginVo {

    @Length(min = 5, max = 17, message = "userName长度须在[5,17]") // 可以为 null，若不为 null，则长度为 [5, 17]
    private String userName;

    // @Size不能验证Integer，适用于String
    @Size(min = 1, max = 3, message = "password长度须在[1,3]")
    private String password;

    @Size(min = 1,max = 5,message = "list的Size在[1,5]") // list 可以为 null，若不为 null，则长度为 [1, 5]
    private List<String> list;

    @NotNull
    @Valid // 级联校验，该注解将会校验自定义类
    private OrderItem detailInfo;
}
@Data
class DetailInfo {

    @NotEmpty(message = "name不能为空")
    private String name;

    @Pattern(regexp="^\\d+$", message = "itemCode需要数字") // 匹配任何数字
    private String itemCode;

    @Range(min = 10, max = 100, message = "range在[10,100]之间")
    private int age;
}

在控制器中增加一个 POST 请求接口。

@RestController
@RequestMapping("/user")
@Validated
public class UserController {

    @GetMapping("/validate1")
    public String validate1(@Size(min = 1, max = 10, message = "姓名长度须为1到10") @RequestParam("name") String name) {
        return "validate1";
    }

    @PostMapping("/validate2")
    public String validate2(@Valid @RequestBody LoginVo loginVo) {
        System.out.println(loginVo);
        return "validate2";
    }
}

发起一个 POST 请求：

POST http://localhost:8080/user/validate2
Content-Type: application/json

{
  "userName": "Tom",
  "password": "1234",
  "list": [1,2,3],
  "detailInfo": {
    "name": "Tom",
    "itemCode": "510000",
    "age": 120
  }
}

服务器返回 400 的状态码。password的尺寸需在[1,3]之间 userName长度需在[5,17]之间 range在[10,100]之间。

二、校验模式

上面例子中一次性返回了所有验证不通过的集合，通常按顺序验证到第一个字段不符合验证要求时，就可以直接拒绝请求了。Hibernate Validator有以下两种验证模式：

普通模式（默认是这个模式）
普通模式时，会校验完所有的属性，然后返回所有的验证失败信息。

快速失败返回模式
快速失败时，只要有一个验证失败，则返回该失败信息。

import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

import javax.validation.Validation;
import javax.validation.Validator;
import javax.validation.ValidatorFactory;

@Configuration
public class ValidatorConfiguration {
    @Bean
    public Validator validator() {
        ValidatorFactory validatorFactory = Validation.byProvider(HibernateValidator.class).configure().failFast(true).buildValidatorFactory();
        return validatorFactory.getValidator();
    }
}

其中 failFast：true 快速失败返回模式；false 普通模式。

三、统一处理 Controller 层异常

当校验失败时，会抛出 MethodArgumentNotValidException 异常，但是我们不应该在 Controller 层处理异常，Controller 层不应该有过多的逻辑处理，异常也不用每个 Controller 都来处理一遍，因此需要全局捕获异常，统一处理。

import org.springframework.http.HttpStatus;
import org.springframework.util.CollectionUtils;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.ResponseStatus;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
/**
 * 统一异常处理类
 */
@ControllerAdvice
public class GlobalExceptionHandler {

    /**
     * 统一捕获处理ConstraintViolationException异常
     */
    @ExceptionHandler(ConstraintViolationException.class)
    @ResponseBody
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    public String resolveConstraintViolationException(ConstraintViolationException ex) {
        Set<ConstraintViolation<?>> constraintViolations = ex.getConstraintViolations();
        StringBuilder msgBuilder = new StringBuilder();
        if (!CollectionUtils.isEmpty(constraintViolations)) {
            for (ConstraintViolation constraintViolation : constraintViolations) {
                String msg = constraintViolation.getMessage();
                msgBuilder.append(msg).append(",");
            }
        }
        return msgBuilder.toString();
    }

    /**
     * 统一捕获处理MethodArgumentNotValidException异常
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    @ResponseBody
    @ResponseStatus(HttpStatus.OK)
    public String resolveMethodArgumentNotValidException(MethodArgumentNotValidException ex) {
        List<ObjectError> objectErrors = ex.getBindingResult().getAllErrors();
        StringBuilder msgBuilder = new StringBuilder();
        if (!CollectionUtils.isEmpty(objectErrors)) {
            for (ObjectError objectError : objectErrors) {
                String msg = objectError.getDefaultMessage();
                msgBuilder.append(msg).append(" ");
            }
        }
        return msgBuilder.toString();
    }
}

在这里，统一处理了 ConstraintViolationException 和 MethodArgumentNotValidException 这两个异常。

在处理异常时，返回状态码通过 @ResponseStatus 定义。例如抛出 ConstraintViolationException 异常时，前端得到的状态码为 400，而抛出 MethodArgumentNotValidException 异常时，前端的状态码为 200。这都需要根据实际业务情况决定。

此外，方法的返回正也需要根据业务逻辑返回对应的值。

四、自定义校验器参数校验

除了默认提供的校验注解外，我们可以定义自己的校验注解，默认的注解总有不满足要求的时候。

例如需要一个校验性别的注解，要求传递一个只能时 man 或者 female 的字符串。

首先创建约束注解类。

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Target;

@Target({ElementType.FIELD})
@Documented
@Constraint(validatedBy = {GenderValidator.class})
public @interface Gender {
    // 验证失败时显示的信息
    String message() default "gender error";
    // 分组的内容
    Class<?>[] groups() default { };
    // 错误验证的级别
    Class<? extends Payload>[] payload() default {};
}

使用 @interface 定义注解时，意味着它实现了 java.lang.annotation.Annotation 接口，即该注解就是一个Annotation。定义 Annotation 时，@interface 是必须的。

使用 @Documented 修饰该 Annotation，则表示它可以出现在 javadoc 中。定义 Annotation 时，@Documented 可有可无；若没有定义，则 Annotation 不会出现在 javadoc 中。

@Constraint(validatedBy = { GenderValidator.class})用来指定处理这个注解逻辑的类。

@Target 的作用，就是来指定 Annotation 的类型属性。@Target(ElementType.FIELD) 的意思就是指定该 Annotation 的类型是用来修饰“字段（包括枚举常量）”的注解。定义 Annotation 时，@Target 可有可无。若有 @Target，则该 Annotation 只能用于它所指定的地方；若没有 @Target，则该 Annotation 可以用于任何地方。其中，ElementType 可选的类型如下：

TYPE 意味着，它能标注"类、接口（包括注释类型）或枚举声明"。
FIELD 意味着，它能标注"字段声明"。
METHOD 意味着，它能标注"方法"。
PARAMETER 意味着，它能标注"参数"。
CONSTRUCTOR 意味着，它能标注"构造方法"。
LOCAL_VARIABLE 意味着，它能标注"局部变量"。

属性 message 用于显示错误信息，这个字段是必须的，groups 和 payload 也是必须的。

该注解的实现类如下。

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;

public class GenderValidator implements ConstraintValidator<Gender, String> {

    private Gender gender;

    @Override
    public void initialize(Gender constraintAnnotation) {
        this.gender = constraintAnnotation;
    }

    @Override
    public boolean isValid(String value, ConstraintValidatorContext context) {
        return value.equals("man") || value.equals("female");
    }
}

注意需要实现 ConstraintValidator 接口，泛型的第一个参数是注解类（Gender），第二个参数是注解修饰属性的类型，这里需要修饰的是 String gender;，所以是 String。

使用如下：

@Data
public class LoginVo {

    @Length(min = 5, max = 17, message = "userName长度需在[5,17]之间")
    private String userName;

    @Size(min = 1, max = 3, message = "password的尺寸需在[1,3]之间")
    private String password;

    @Gender
    private String gender;
}

发一个 POST 请求：

POST http://localhost:8090/user/validate2
Content-Type: application/json

{
  "userName": "Tomas",
  "password": "1234",
  "gender": "unknow"
}

由于 gender 的内容不是我们定义的 man 或者 female，不符合校验规则，所以返回了错误信息 gender error 。

SpringBoot 使用 Hibernate-Validato

一、基本使用

二、校验模式

三、统一处理 Controller 层异常

四、自定义校验器参数校验

相关文章

网友评论

延伸阅读

深度阅读

栏目导航

热点阅读

设计方案