利用@Valid自定义参数校验

@EnableValue

完成校验参数为指定值的功能，自定义参数校验注解。

@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Constraint(validatedBy = {EnableValueValidator.class})
public @interface EnableValue {
    String message() default "必须为指定值";

    String[] strValues() default {};

    int[] intValues() default {};

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

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

    @Target({ElementType.FIELD})
    @Retention(RetentionPolicy.RUNTIME)
    @Documented
    @interface List {
        EnableValue[] value();
    }
}

EnableValueValidator

实现ConstraintValidator，完成校验逻辑。

public class EnableValueValidator implements ConstraintValidator<EnableValue, Object> {
    private String[] strValues;

    private int[] intValues;

    @Override
    public void initialize(EnableValue enableValue) {
        strValues = enableValue.strValues();
        intValues = enableValue.intValues();
    }

    @Override
    public boolean isValid(Object value, ConstraintValidatorContext context) {
        if (value instanceof String) {
            for (String item : strValues) {
                if (item.equals(value)) {
                    return true;
                }
            }
        }

        if (value instanceof Integer) {
            for (Integer item : intValues) {
                if (item.equals(value)) {
                    return true;
                }
            }
        }
        return false;
    }
}