SpringBoot 使用 JSR303 自定义校验注解

JSR303 是 Java EE 6 中的一项子规范，叫作 Bean Validation，官方参考实现是hibernate Validator，有了它，咱们能够在实体类的字段上标注不一样的注解实现对数据的校验，不用 if-else 判断，简化了咱们的开发，并且可读性也很好。

但有时候它提供的注解并不能知足咱们的要求，好比，咱们要求字段 color 必须是「red，blue，yellow」这三个值之一，这时候，咱们就须要本身写判断的逻辑了，你能够自定义一个方法在其余地方进行判断，但既然用了 JSR303，为了统一代码，能够自定义校验注解。

一、给字段 color 上标注自定义注解 @ListValue，写上这几个值

@ListValue(vals = {"red", "blue", "yellow"})
private String color;

二、建立注解 @ListValue，能够参考官方的注解，好比 @NotNull，咱们只须要修改下面注释的几处便可

//自定义的约束校验器
@Constraint(validatedBy = {ListValueConstraintValidator.class}) 
@Documented
@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})
@Retention(RUNTIME)
public @interface ListValue {

    //配置文件中错误提示信息的名称
    String message() default "{com.sjl.common.valid.ListValue.message}"; 

    Class<?>[] groups() default {};

    Class<? extends Payload>[] payload() default {};
	
    //自定义值的类型
    String[] vals() default {};
}

三、建立自定义约束校验器，继承 ConstraintValidator，第一个泛型是自定义注解、第二个是校验值的类型，也即注解标注的字段的类型

public class ListValueConstraintValidator implements ConstraintValidator<ListValue,String>{
    
    private static Set<String> set = new HashSet<>();

    @Override
    public void initialize(ListValue constraintAnnotation) {
        for (String val : constraintAnnotation.vals()) {
            set.add(val);
        }
    }

    /**
     * 判断是否经过校验
     *
     * @param value   传入的值
     * @param context
     * @return
     */
    @Override
    public boolean isValid(String value, ConstraintValidatorContext context) {
        return set.contains(value);
    }
}

四、在 resources 目录下建立一个 ValidationMessages.properties 配置文件，key 是第二步 message 设置的默认值，value 是自定义错误信息

com.sjl.common.valid.ListValue.message=必须提交指定的值

至此，全部的工做都已完成，自定义注解 @ListValue 就能够工做了，固然这只是很简单的校验，但方法大同小异。