Java实现自定义枚举值校验器的示例代码

@Getter

public enum UserTypeEnum {

     STUDENT([ 1 ], "学生" ),

     TEACHER([ 2 ], "老师" );

     private Integer code;

     private String name;

     UserTypeEnum(Integer code, String name) {

         this .code = code;

         this .name = name;

     }

}

@Documented

@Retention (RetentionPolicy.RUNTIME)

@Target ({ElementType.FIELD, ElementType.METHOD})

@Constraint (validatedBy = EnumValidatorHandler. class )

public @interface EnumValidator {

     Class<?> value();

     String message() default "入参值不在正确枚举中" ;

     String method() default "getCode" ;

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

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

}

@Slf4j

class EnumValidatorHandler implements ConstraintValidator<EnumValidator, Object> {

     private List<Object> values = new ArrayList<>();

     @Override

     public void initialize(EnumValidator enumValidator) {

         Class<?> clazz = enumValidator.value();

         Object[] objects = clazz.getEnumConstants();

         try {

             Method method = clazz.getMethod(enumValidator.method());

             Object value;

             for (Object obj : objects) {

                 value = method.invoke(obj);

                 values.add(value);

             }

         } catch (Exception e) {

             log.error( "处理枚举校验异常:{}" , e);

         }

     }

     @Override

     public boolean isValid(Object value, ConstraintValidatorContext constraintValidatorContext) {

         if (value instanceof String) {

             String valueStr = (String)value;

             return StringUtils.isEmpty(valueStr)|| values.contains(value);

         }

         return Objects.isNull(value) || values.contains(value);

     }

}

@Entity

public class User {

     @EnumValidator (value = UserTypeEnum. class )

     private String userType;

}

public User add( @Valid User user) {

}