文章目录

一、前言二、自定义校验2.1 定义 GenderArrayValuable 接口2.2 定义性别 GenderEnum 枚举类2.3 自定义 @GenderCheck 自定义约束注解2.4 自定义约束的校验器 GenderValidator2.5 定义 UserUpdateGenderDTO2.6 定义一个对外访问接口2.7 请求接口 进行验证 三、总结

一、前言

在上一篇文章 Springboot实现优雅的参数校验（Spring Validation）和 if else说再见，我们介绍了 Spring Validation 的初级用法，在实际开发中，无论是 Bean Validation 定义的约束，还是 Hibernate Validator 附加的约束，都是无法满足我们复杂的业务场景。所以，我们需要自定义约束。开发自定义约束一共只要两步：

编写自定义约束的注解；编写自定义的校验器 ConstraintValidator 。

下面，就让我们一起来实现一个自定义约束，用于校验参数必须在枚举值的范围内吧。

二、自定义校验

比如一个很常见的用户注册的场景，一般都要限定用户性别，这里我们不考虑 同性 未知的情况哈，只考虑 男和女 这种情况。0：代表女性，1：代表男性，

2.1 定义 GenderArrayValuable 接口

为了方便我们后面获取枚举类型值的，我先定义一个接口 GenderArrayValuable,这里的定义了一个 返回一个int[] 数据的方法。后面我们会在 GenderEnum 中实现这个接口。

package com.ratel.validation.core.validator;/** * 可生成 Int 数组的接口 */public interface GenderArrayValuable {    /**     * @return int 数组     */    int[] array();}

2.2 定义性别 GenderEnum 枚举类

GenderEnum 枚举类实现了 GenderArrayValuable 的 IntArrayValuable 接口，返回值数组 ARRAYS。

import com.ratel.validation.core.validator.GenderArrayValuable;import java.util.Arrays;public enum GenderEnum implements GenderArrayValuable {    MALE(1, "男"),    FEMALE(0, "女");    /**     * 值数组     */    public static final int[] ARRAYS = Arrays.stream(values()).mapToInt(GenderEnum::getValue).toArray();    /**     * 性别值     */    private final Integer value;    /**     * 性别名     */    private final String name;    GenderEnum(Integer value, String name) {        this.value = value;        this.name = name;    }    public Integer getValue() {        return value;    }    public String getName() {        return name;    }    @Override    public int[] array() {        return ARRAYS;    }}

2.3 自定义 @GenderCheck 自定义约束注解

package com.ratel.validation.core.validator;import javax.validation.Constraint;import javax.validation.Payload;import java.lang.annotation.Documented;import java.lang.annotation.Retention;import java.lang.annotation.RetentionPolicy;import java.lang.annotation.Target;import static java.lang.annotation.ElementType.*;@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})@Retention(RetentionPolicy.RUNTIME)@Documented@Constraint(validatedBy = GenderValidator.class)public @interface GenderCheck {    /**     * @return 实现 IntArrayValuable 接口的     */    Class<? extends GenderArrayValuable> value();    /**     * @return 提示内容     */    String message() default "必须在指定范围 {value}";    /**     * @return 分组     */    Class<?>[] groups() default {};    /**     * @return Payload 数组     */    Class<? extends Payload>[] payload() default {};    /**     *  Defines several {@code @GenderCheck} constraints on the same element.     */    @Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})    @Retention(RetentionPolicy.RUNTIME)    @Documented    @interface List {       GenderCheck[] value();    }}

2.4 自定义约束的校验器 GenderValidator

package com.ratel.validation.core.validator;import javax.validation.ConstraintValidator;import javax.validation.ConstraintValidatorContext;import java.util.Arrays;import java.util.Collections;import java.util.Set;import java.util.stream.Collectors;public class GenderValidator implements ConstraintValidator<GenderCheck, Integer> {    /**     * 值数组     */    private Set<Integer> values;    @Override    public void initialize(GenderCheck annotation) {        GenderArrayValuable[] values = annotation.value().getEnumConstants();        if (values.length == 0) {            this.values = Collections.emptySet();        } else {            this.values = Arrays.stream(values[0].array()).boxed().collect(Collectors.toSet());        }    }    @Override    public boolean isValid(Integer value, ConstraintValidatorContext context) {        // 1 校验通过        if (values.contains(value)) {            return true;        }        // 2 校验不通过，自定义提示语句（因为，注解上的 value 是枚举类，无法获得枚举类的实际值）        context.disableDefaultConstraintViolation(); // 3 禁用默认的 message 的值        context.buildConstraintViolationWithTemplate(context.getDefaultConstraintMessageTemplate()                .replaceAll("\\{value}", values.toString())).addConstraintViolation(); // 4. 重新添加错误提示语句        return false; // 5     }}

实现 ConstraintValidator 接口。

第一个泛型为 A extends Annotation ，设置对应的自定义约束的注解。例如说，这里我们设置了 @GenderEnum 注解。第二个泛型为 T ，设置对应的参数值的类型。例如说，这里我们设置了 Integer 类型。

实现 initialize(A constraintAnnotation) 方法，获得 @GenderEnum 注解的 values() 属性，获得值数组，设置到 values 属性种。

实现 boolean isValid(T var1, ConstraintValidatorContext var2); 方法，实现校验参数值，是否在 values 范围内。
在注释 1 处，校验参数值在范围内，直接返回 true ，校验通过。
在注释 2 处，校验不通过，自定义提示语句。
在注释 5 处，校验不通过，所以返回 false 。
至此，我们已经完成了自定义约束的实现。

2.5 定义 UserUpdateGenderDTO

定一个 UserUpdateGenderDTO 实体类，在 gender 字段上添加自定义的 @GenderCheck(value = GenderEnum.class, message = "性别必须是 {value}") 注解，限制传入的参数值，必须在 GenderEnum 枚举范围内。

package com.ratel.validation.entity;import com.ratel.validation.core.validator.GenderCheck;import com.ratel.validation.enums.GenderEnum;import javax.validation.constraints.NotNull;/** * 用户更新性别 DTO */public class UserUpdateGenderDTO {    /**     * 用户编号     */    @NotNull(message = "用户编号不能为空")    private Integer id;    /**     * 性别     */    @NotNull(message = "性别不能为空")    @GenderCheck(value = GenderEnum.class, message = "性别必须是 {value}")    private Integer gender;    public Integer getId() {        return id;    }    public UserUpdateGenderDTO setId(Integer id) {        this.id = id;        return this;    }    public Integer getGender() {        return gender;    }    public UserUpdateGenderDTO setGender(Integer gender) {        this.gender = gender;        return this;    }    @Override    public String toString() {        return "UserUpdateGenderDTO{" +                "id=" + id +                ", gender=" + gender +                '}';    }}

2.6 定义一个对外访问接口

import io.swagger.annotations.Api;import io.swagger.annotations.ApiOperation;import io.swagger.annotations.Tag;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import org.springframework.beans.factory.annotation.Autowired;import org.springframework.validation.annotation.Validated;import org.springframework.web.bind.annotation.*;import javax.validation.Valid;import javax.validation.constraints.Min;@RestController@RequestMapping("/users")@Validated@Api(tags = "用户")public class UserController {    private Logger logger = LoggerFactory.getLogger(getClass());        @PostMapping("/update_gender")    public String updateGender(@Valid UserUpdateGenderDTO updateGenderDTO) {        logger.info("[updateGender][updateGenderDTO: {}]", updateGenderDTO);        return "性别更新成功";    }} 

2.7 请求接口 进行验证

我们这里使用 knife4j 接口文档，当我们 把 gender的值传成 非 【0 ，1 】 以外的值，就会直接返回错误信息，"请求参数不合法:性别必须是 [0, 1]",

三、总结

希望阅读完本文，能够让各位 c友 更加舒适且优雅的完成各种需要参数校验的地方。 不说了，赶紧给自己的系统去把参数校验给补全，嘿嘿。

当然，有一点要注意，Bean Validation 更多做的是，无状态的参数校验。怎么理解呢？

例如说，参数的大小长度,判断参数是否为空，是否是身份证号，是否是邮箱，是否是手机号 这些不依赖 外部数据源的等等，是适合通过 Spring Validation 中完成。
例如说，校验用户名，邮箱，手机号 唯一等等，依赖 外部数据源 的，是不适合通过 Spring Validation中完成。