update 优化 pr代码

master
疯狂的狮子Li 1 year ago
parent b1badca062
commit 552e543471

@ -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 "输入值不在枚举范围内";

@ -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 @@ public class EnumPatternValidator implements ConstraintValidator<EnumPattern, St
@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;
}
}
}

Loading…
Cancel
Save