枚举校验器

/**
 * <p>枚举校验器</p>
 */
public class EnumObjectValidator implements ConstraintValidator<EnumObjectValid, Integer> {

    private Class<? extends Enum> enumClass;

    private boolean required;


    @Override
    public void initialize(EnumObjectValid constraintAnnotation) {
        enumClass = constraintAnnotation.enumClass();
        required = constraintAnnotation.required();
    }

    /**
     * <p>校验code是否属于枚举值</p>
     *
     * @param code
     * @param constraintValidatorContext
     * @return
     * @throws
     * @author hongtao.su
     * @date 2023/7/18 15:45
     */
    @Override
    public boolean isValid(Integer code, ConstraintValidatorContext constraintValidatorContext) {
        if (!required && code == null) {
            return true;
        }
        if (code == null) {
            return false;
        }
        EnumConstraint[] enums = (EnumConstraint[]) enumClass.getEnumConstants();
        if (enums == null || enums.length == 0) {
            return false;
        }
        return enums[0].exist(code);
    }
}

/**
 * <p>枚举校验</p>
 */
@Target({ElementType.FIELD, ElementType.METHOD, ElementType.ANNOTATION_TYPE, ElementType.PARAMETER})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = {EnumObjectValidator.class})
@Documented
public @interface EnumObjectValid {

    boolean required() default true;

    String message() default "enum.is.invalid";

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

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

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

    Class<? extends Enum<?>> enumClass();
}

提供一个枚举校验类以及校验注解,用户校验接口入参枚举值的合法性
eg:

@Data
public class AuditVehicleConfDTO {
    @NotNull(message = "状态不可以为空")
    @EnumObjectValid(message = "状态不合法", enumClass = StatusEnum.class)
    private Integer auditStatus;
}
  • 6
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值