admin/lx-admin-backend
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update 优化 pr代码

Browse Source
This commit is contained in:
疯狂的狮子Li 2024-12-10 17:46:45 +08:00
parent b1badca062
commit 552e543471
2 changed files with 12 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPattern.java
View File

@ -24,13 +24,13 @@ public @interface EnumPattern {
/** /**
* 需要校验的枚举类型 * 需要校验的枚举类型
*/ */
Class<? extends Enum> type() default Enum.class; Class<? extends Enum<?>> type();
/** /**
* 枚举类型校验值字段名称 * 枚举类型校验值字段名称
* TODO 需确保该字段实现了 getter 方法 * 需确保该字段实现了 getter 方法
*/ */
String fieldName() default "code"; String fieldName();
String message() default "输入值不在枚举范围内"; String message() default "输入值不在枚举范围内";

24
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java
View File

@ -1,11 +1,9 @@
package org.dromara.common.core.validate.enumd; 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.ConstraintValidator;
import jakarta.validation.ConstraintValidatorContext; import jakarta.validation.ConstraintValidatorContext;
import jakarta.validation.ValidationException; import org.dromara.common.core.utils.StringUtils;
import lombok.extern.slf4j.Slf4j; import org.dromara.common.core.utils.reflect.ReflectUtils;
/** /**
* 自定义枚举校验注解实现 * 自定义枚举校验注解实现
@ -25,21 +23,15 @@ public class EnumPatternValidator implements ConstraintValidator<EnumPattern, St
@Override @Override
public boolean isValid(String value, ConstraintValidatorContext constraintValidatorContext) { public boolean isValid(String value, ConstraintValidatorContext constraintValidatorContext) {
try { if (StringUtils.isNotBlank(value)) {
if (StringUtils.isNotBlank(value)) { String fieldName = annotation.fieldName();
Class<?> type = annotation.type(); for (Object e : annotation.type().getEnumConstants()) {
String fieldName = annotation.fieldName(); if (value.equals(ReflectUtils.invokeGetter(e, fieldName))) {
Object[] enumConstants = type.getEnumConstants(); return true;
for (Object e : enumConstants) {
if (value.equals(ReflectUtils.invokeGetter(e, fieldName))) {
return true;
}
} }
} }
return false;
} catch (Exception e) {
throw new ValidationException(e);
} }
return false;
} }
} }