疯狂的狮子Li
2024-12-10 552e543471984fd37fd5cd70cde0218398258b3e
update 优化 pr代码
已修改2个文件
32 ■■■■■ 文件已修改
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPattern.java 6 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java 26 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPattern.java
@@ -24,13 +24,13 @@
    /**
     * 需要校验的枚举类型
     */
    Class<? extends Enum> type() default Enum.class;
    Class<? extends Enum<?>> type();
    /**
     * 枚举类型校验值字段名称
     * TODO 需确保该字段实现了 getter 方法
     * 需确保该字段实现了 getter 方法
     */
    String fieldName() default "code";
    String fieldName();
    String message() default "输入值不在枚举范围内";
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java
@@ -1,11 +1,9 @@
package org.dromara.common.core.validate.enumd;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.core.utils.reflect.ReflectUtils;
import jakarta.validation.ConstraintValidator;
import jakarta.validation.ConstraintValidatorContext;
import jakarta.validation.ValidationException;
import lombok.extern.slf4j.Slf4j;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.core.utils.reflect.ReflectUtils;
/**
 * 自定义枚举校验注解实现
@@ -25,21 +23,15 @@
    @Override
    public boolean isValid(String value, ConstraintValidatorContext constraintValidatorContext) {
        try {
            if (StringUtils.isNotBlank(value)) {
                Class<?> type = annotation.type();
                String fieldName = annotation.fieldName();
                Object[] enumConstants = type.getEnumConstants();
                for (Object e : enumConstants) {
                    if (value.equals(ReflectUtils.invokeGetter(e, fieldName))) {
                        return true;
                    }
        if (StringUtils.isNotBlank(value)) {
            String fieldName = annotation.fieldName();
            for (Object e : annotation.type().getEnumConstants()) {
                if (value.equals(ReflectUtils.invokeGetter(e, fieldName))) {
                    return true;
                }
            }
            return false;
        } catch (Exception e) {
            throw new ValidationException(e);
        }
        return false;
    }
}
}