自定义注解实现枚举值验证

开发运维 2023-10-24 向阳逐梦手机阅读

在本文中，我们将介绍如何使用自定义注解来实现枚举值的验证。这个自定义注解称为@EnumValue，它能够确保给定的值在指定的枚举类中存在。

背景

在Java应用程序中，经常需要验证特定字段的值是否在一个预定义的枚举类中。这种验证通常需要编写相同的重复代码。为了简化这个过程，我们可以创建一个自定义注解，以减少重复性代码。

自定义注解@EnumValue

首先，让我们来了解一下自定义注解@EnumValue的代码结构：

.common.annotations;

import lombok.SneakyThrows;
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;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import java.lang.reflect.Method;

import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER})
@Retention(RUNTIME)
@Documented
@Constraint(validatedBy = {EnumValue.EnumValueValidator.class})
public @interface EnumValue {

    //默认错误消息
    String message() default "必须为指定值";

    //支持枚举列表验证
    Class value();

    //分组
    Class[] groups() default {};

    //负载
    Class enumValue;

        /**
         * 初始化方法
         *
         * @param constraintAnnotation
         */
        @Override
        public void initialize(EnumValue constraintAnnotation) {
            enumValue = constraintAnnotation.value();
        }

        /**
         * 校验方法
         *
         * @param value
         * @param context
         * @return
         */
        @SneakyThrows
        @Override
        public boolean isValid(String value, ConstraintValidatorContext context) {
            if (StringUtils.isBlank(value)) {
                return true;
            }
            //针对枚举类型的校验匹配
            if (enumValue != null && enumValue.isEnum()) {
                //枚举类验证
                Object[] objs = enumValue.getEnumConstants();
                //这里需要注意，定义枚举时，枚举值名称统一用value表示
                Method method = enumValue.getMethod("getCode");
                for (Object temp : objs) {
                    Object code = method.invoke(temp, null);
                    if (value.equals(code.toString())) {
                        return true;
                    }
                }
            }
            return false;
        }
    }
}

使用@EnumValue注解

现在，让我们看看如何在你的Java类中使用@EnumValue注解：

public enum MyEnum {
    VALUE1("1"),
    VALUE2("2"),
    VALUE3("3");

    private String code;

    MyEnum(String code) {
        this.code = code;
    }

    public String getCode() {
        return code;
    }
}

public class MyClass {
    @EnumValue(value = MyEnum.class, message = "必须为MyEnum中的值")
    private String myEnumValue;

    // 省略其他属性和方法
}

在上面的示例中，我们定义了一个枚举类MyEnum，然后在另一个类MyClass中使用了@EnumValue注解来验证myEnumValue字段是否在MyEnum枚举中存在。如果字段的值不在枚举中，将会触发错误消息"必须为MyEnum中的值"。