spring-boot-validation自定义约束验证注解@EnumValue

1.@EnumValue注解类

package com.jb.scrm.validator;


import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.*;


/**
 * @author ww
 */
@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(validatedBy = {EnumValueValidator.class})
public @interface  EnumValue {

    //默认错误消息
    String message() default "必须为指定值";

    String[] strValues() default {};

    int[] intValues() default {};

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

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

    //指定多个时使用
    @Target({ElementType.FIELD, ElementType.METHOD, ElementType.PARAMETER, ElementType.ANNOTATION_TYPE})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    @interface List {
        EnumValue[] value();
    }
}

2.EnumValueValidator约束验证器类

package com.jb.scrm.validator;

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Objects;

/**
 * @author ww
 */
public class EnumValueValidator implements ConstraintValidator<EnumValue, Object> {

    private String[] strValues;
    private int[] intValues;

    @Override
    public void initialize(EnumValue constraintAnnotation) {
        strValues = constraintAnnotation.strValues();
        intValues = constraintAnnotation.intValues();
    }

    @Override
    public boolean isValid(Object value, ConstraintValidatorContext context) {
        if(value instanceof String) {
            for (String s:strValues) {
                if(s.equals(value)){
                    return true;
                }else {
                    //如果是多个，隔开也给放行
                    String[] values = String.valueOf(value).split(",");
                    ArrayList<String> list = new ArrayList(Arrays.asList(strValues));
                    boolean  allok = true;
                    for (String s1 : values) {
                        if (!list.contains(s1)){
                             allok = false;
                        }
                    }
                    return allok;
                }
            }
        }else if(value instanceof Integer){
            for (Integer s:intValues) {
                if(Objects.equals(s,value)){
                    return true;
                }
            }
        }
        return false;

    }
}