关于java:自定义枚举值校验器

4次阅读

共计 1510 个字符,预计需要花费 4 分钟才能阅读完成。

前言

定义一个枚举类,现须要对传入的枚举值进行范畴校验。

@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;
}

在 c 层入参的参数上退出 @Valid 注解

public User add(@Valid User user) {}

如果是其余层的入参校验,须要所在类上退出 @Validated 注解。

他抛出的异样为 ConstraintViolationException
最初,退出一个全局异样处理器解决异样。

除了自定义的注解,还有惯例的注解校验器:

正文完
 0