!607 add 新增 validation支持枚举校验 Merge pull request !607 from 秋辞未寒/dev

疯狂的狮子Li

2024-12-10 b1badca062e9f3d22398299247cca8ff191414da

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
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); 
        } 
    } 
 
}