本篇内容主要讲解“Java如何实现自定义枚举值校验器”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“Java如何实现自定义枚举值校验器”吧!
前言
定义一个枚举类,现需要对传入的枚举值进行范围校验。
@Getterpublic 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 {};}
@Slf4jclass 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); }}
使用
在实体类的属性上加入注解
@Entitypublic class User { @EnumValidator(value = UserTypeEnum.class) private String userType;}
在c层入参的参数上加入@Valid注解
public User add(@Valid User user) {}
如果是其他层的入参校验,需要所在类上加入@Validated
注解。
他抛出的异常为ConstraintViolationException
,
最后,加入一个全局异常处理器处理异常。
除了自定义的注解,还有常规的注解校验器
到此,相信大家对“Java如何实现自定义枚举值校验器”有了更深的了解,不妨来实际操作一番吧!这里是编程网网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!