自己实现一个枚举validation校验器

背景

平时我们在进行接口校验的时候，我们会使用@Validation相关注解来帮助我们简化代码。其常用的注解其实已经能满足很大一部分需求，但是在进行枚举字段的接口校验的时候没有相关字段，尤其在我们不是使用枚举的name()方法来传递的时候，枚举的校验就需要很多类似的大段的方法来进行校验。而validation本身是提供了扩展的方法的，因此在次基础上实现了一个对枚举进行校验的注解。

基础类

1.EnumValidation注解

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;

import javax.validation.Constraint;
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import javax.validation.Payload;
import java.lang.annotation.*;

/** * 自定义枚举验证注解 */
@Documented
@Target({
    ElementType.ANNOTATION_TYPE, ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = {
    EnumValidation.Validator.class})
public @interface EnumValidation {
    
    /** * 错误提示信息 * * @return */
    String message() default "Enumeration validation failed";

    /** * 校验分组 * * @return */
    Class<?>[] groups() default {
    };

    /** * 要检验参数所属的枚举数组 * * @return */
    Class<? extends EnumType>[] enums();

    /** * 要校验的参数是否可以为空 * * @return */
    boolean canBeBlank() default true;

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

    // 验证器内部类
    @Slf4j
    class Validator implements ConstraintValidator<EnumValidation, String> {
    

        Class<? extends EnumType>[] enums;
        boolean canBeBlank;

        @Override
        public void initialize(EnumValidation constraintAnnotation) {
    
            this.enums = constraintAnnotation.enums();
            this.canBeBlank = constraintAnnotation.canBeBlank();
        }

        @Override
        public boolean isValid(String value, ConstraintValidatorContext context) {
    
            //如果校验的字段为空，判断是否通过
            if (StringUtils.isBlank(value) || enums.length < 1) {
    
                return canBeBlank;
            }

            //校验枚举字段匹配
            for (Class<? extends EnumType> clazz : enums) {
    
                if (clazz.isEnum()) {
    
                    EnumType[] enumConstants = clazz.getEnumConstants();
                    for (EnumType enumType : enumConstants) {
    
                        if (value.equals(enumType.getValidKey())) {
    
                            return true;
                        }
                    }
                }
            }
            return false;
        }
    }
}

2.EnumType 枚举公共父类

/** * 枚举继承公共类 */
public interface EnumType {
    

    /** * 获取key * * @return 获取key */
    String getValidKey();
}

使用方法

定义这两个类之后，我们只需要在我们定义的注解上继承EnumType，然后实现其getValidKey方法即可

如下：

/** * 是/否 枚举 */
public enum YesOrNoEnum implements EnumType {
    
    NO(0, "否"),
    YES(1, "是"),
    ;

    YesOrNoEnum(int code, String desc) {
    
        this.code = code;
        this.desc = desc;
    }

    public int code;
    public String desc;

    @Override
    public String getValidKey() {
    
        return code + "";
    }
}

然后在需要校验的字段上加上这个注解即可

import lombok.Data;
import org.hibernate.validator.constraints.Range;

import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
import javax.validation.constraints.Size;

/** * 人类 * */
@Data
public class PersonDTO {
    
    @NotBlank(message = "姓名不能为空")
    @Size(max = 32, message = "姓名长度不能超过320")
    private String name;

    @NotNull(message = "年齡不能为空")
    @Range(min = 0, max = 150, message = "年龄必须在0-150之间")
    private Integer age;

    @EnumValidation(enums = YesOrNoEnum.class, canBeBlank = false, message = "错误的会员标志")
    private Integer memberFlag;
}