springboot 自定义效验规则

自定义校验

1. 创建校验注解（FlagValue）

@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(
        // 添加效验器
        validatedBy = { FlagValueConstraintValidator.class }
)
public @interface FlagValue {
    // 错误信息来源
    String message() default "{com.rock.common.valid.FlagValue.message}";

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

    Class<? extends Payload>[] payload() default {};

    // 定义一个int数组接收效验数据
    int[] values() default {};
}

3. 使用自定义效验注解（BrandEntity）

	/**
	 * 显示状态[0-不显示；1-显示]
	 */
	@FlagValue(values = {1,0}, groups = {AddGroup.class})
	private Integer showStatus;

2. 创建校验器（FlagValueConstraintValidator）

public class FlagValueConstraintValidator implements ConstraintValidator<FlagValue, Integer> {

    private Set<Integer> set = new HashSet<>();

    /**
     * 获取注解中的效验数据
     * @param constraintAnnotation
     */
    @Override
    public void initialize(FlagValue constraintAnnotation) {
        int[] values = constraintAnnotation.values();
        if(values.length != 0){
            for (int value : values) {
                set.add(value);
            }
        }
    }

    /**
     * 制定效验规则，判断是否效验成功
     * @param value 需要校验的值
     * @param constraintValidatorContext
     * @return
     */
    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext constraintValidatorContext) {
        return set.contains(value);
    }
}