Java Validation 只能输入固定值(自定义校验注解、@Pattern)

方法一:

自定义校验注解

注解接口:

import javax.validation.Constraint;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import javax.validation.Payload;
import java.lang.annotation.Documented;
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
import java.util.HashSet;
import java.util.Set;

/**
 * 只能提交指定的值
 * 自定义校验
 */
@Documented
@Constraint(validatedBy = { ListValue.ListValueConstraintValidator.class, ListValue.ListStringValueConstraintValidator.class }) // 指定校验器,这里不指定时,就需要在初始化时指定
@Target({ElementType.METHOD, ElementType.FIELD, ElementType.ANNOTATION_TYPE, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
@Retention(RetentionPolicy.RUNTIME)
public @interface ListValue {

    // 默认的提示内容
    String message() default "必须提交指定的值哦";

    Class[] groups() default { };

    Class[] payload() default { };

    int[] values() default { }; //数值数组,提交的值只能是数组里面

    String[] strValues() default { }; //字符串数组,提交的值只能是数组里面

    class ListValueConstraintValidator implements ConstraintValidator {

        private Set set = new HashSet();

        /**
         * 初始化方法
         */
        @Override
        public void initialize(ListValue constraintAnnotation) {
            int[] values = constraintAnnotation.values();
            for (int val : values) {
                set.add(val);
            }
        }

        /**
         * 判断是否校验成功
         */
        @Override
        public boolean isValid(Integer value, ConstraintValidatorContext context) {
            return set.contains(value);
        }
    }

    class ListStringValueConstraintValidator implements ConstraintValidator {

        private Set set = new HashSet();

        /**
         * 初始化方法
         */
        @Override
        public void initialize(ListValue constraintAnnotation) {
            String[] values = constraintAnnotation.strValues();
            for (String val : values) {
                set.add(val);
            }
        }

        /**
         * 判断是否校验成功
         */
        @Override
        public boolean isValid(String value, ConstraintValidatorContext context) {
            return set.contains(value);
        }
    }
}

DTO使用:

    /**
     * 回避类型:1单位,2专家
     */
    @NotNull(message = "回避类型不为空")
    @ApiModelProperty(value = "回避类型:1单位,2专家", required = true)
    @ListValue(values = {1, 2}, message = "回避类型:1单位,2专家")
    private Integer type;

    /**
     * 回避类型
     */
    @NotBlank(message = "回避类型不为空")
    @ApiModelProperty(value = "回避类型", required = true)
    @ListValue(strValues = {"单位", "专家"}, message = "回避类型:单位,专家")
    private String typeMean;

type只能输入数字1、2

typeMean只能输入字符串“单位”、“专家”

方法二:

@Pattern注解

只能校验字符串,正则表达式为:“单位|专家”

    /**
     * 回避类型
     */
    @NotBlank(message = "回避类型不为空")
    @ApiModelProperty(value = "回避类型", required = true)
    @Pattern(regexp = "单位|专家", message = "回避类型:单位,专家")
    private String typeMean;

typeMean只能输入字符串“单位”、“专家”

本文来自网络,不代表协通编程立场,如若转载,请注明出处:https://net2asp.com/8850e78557.html