疯狂的狮子Li
2024-12-10 b1badca062e9f3d22398299247cca8ff191414da
ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java
¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,45 @@
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;
/**
 * è‡ªå®šä¹‰æžšä¸¾æ ¡éªŒæ³¨è§£å®žçް
 *
 * @author ç§‹è¾žæœªå¯’
 * @date 2024-12-09
 */
public class EnumPatternValidator implements ConstraintValidator<EnumPattern, String> {
    private EnumPattern annotation;;
    @Override
    public void initialize(EnumPattern annotation) {
        ConstraintValidator.super.initialize(annotation);
        this.annotation = annotation;
    }
    @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;
                    }
                }
            }
            return false;
        } catch (Exception e) {
            throw new ValidationException(e);
        }
    }
}