当前位置:主页 > java教程 > Java枚举值校验器

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

发布:2023-03-27 09:10:01 59


本站精选了一篇相关的编程文章,网友简泰鸿根据主题投稿了本篇教程内容,涉及到Java自定义枚举值校验器、Java枚举值校验器、Java枚举值校验、Java枚举值、Java枚举值校验器相关内容,已被451网友关注,涉猎到的知识点内容可以在下方电子书获得。

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

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

public User add(@Valid User user) {
}

如果是其他层的入参校验,需要所在类上加入@Validated注解。

他抛出的异常为ConstraintViolationException

最后,加入一个全局异常处理器处理异常。

除了自定义的注解,还有常规的注解校验器

到此这篇关于Java实现自定义枚举值校验器的示例代码的文章就介绍到这了,更多相关Java枚举值校验器内容请搜索码农之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持码农之家!


参考资料

相关文章

网友讨论