Java如何實(shí)現(xiàn)自定義枚舉值校驗(yàn)器

本篇內(nèi)容主要講解“Java如何實(shí)現(xiàn)自定義枚舉值校驗(yàn)器”，感興趣的朋友不妨來看看。本文介紹的方法操作簡(jiǎn)單快捷，實(shí)用性強(qiáng)。下面就讓小編來帶大家學(xué)習(xí)“Java如何實(shí)現(xiàn)自定義枚舉值校驗(yàn)器”吧!

前言

定義一個(gè)枚舉類，現(xiàn)需要對(duì)傳入的枚舉值進(jìn)行范圍校驗(yàn)。

@Getter
public enum UserTypeEnum {
    STUDENT(“1”, "學(xué)生"),
    TEACHER(“2”, "老師");
 
    private Integer code;
    private String name;
 
    UserTypeEnum(Integer code, String name) {
        this.code = code;
        this.name = name;
    }
}

當(dāng)超出范圍時(shí)，報(bào)錯(cuò)。

注解校驗(yàn)

@Documented
@Retention(RetentionPolicy.RUNTIME)
@Target({ElementType.FIELD, ElementType.METHOD})
@Constraint(validatedBy = EnumValidatorHandler.class)
public @interface EnumValidator {
    Class<?> value();
 
    String message() default "入?yún)⒅挡辉谡_枚舉中";
    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("處理枚舉校驗(yàn)異常:{}", 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);
    }
}

使用

在實(shí)體類的屬性上加入注解

@Entity
public class User {
    @EnumValidator(value = UserTypeEnum.class)
    private String userType;
}

在c層入?yún)⒌膮?shù)上加入@Valid注解

public User add(@Valid User user) {
}

如果是其他層的入?yún)⑿ｒ?yàn)，需要所在類上加入@Validated注解。

他拋出的異常為ConstraintViolationException，

最后，加入一個(gè)全局異常處理器處理異常。

除了自定義的注解，還有常規(guī)的注解校驗(yàn)器

到此，相信大家對(duì)“Java如何實(shí)現(xiàn)自定義枚舉值校驗(yàn)器”有了更深的了解，不妨來實(shí)際操作一番吧！這里是億速云網(wǎng)站，更多相關(guān)內(nèi)容可以進(jìn)入相關(guān)頻道進(jìn)行查詢，關(guān)注我們，繼續(xù)學(xué)習(xí)！