add 新增 validation支持枚举校验

秋辞未寒

2024-12-09 9e613488f1a01563913b4f6aade062980a315fea

add 新增 validation支持枚举校验

Signed-off-by: 秋辞未寒 <545073804@qq.com>

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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,39 @@
package org.dromara.common.core.validate.enumd;



import jakarta.validation.Constraint;

import jakarta.validation.Payload;



import java.lang.annotation.*;



import static java.lang.annotation.ElementType.*;

import static java.lang.annotation.RetentionPolicy.RUNTIME;



/**

 * è‡ªå®šä¹‰æžšä¸¾æ ¡éªŒæ³¨è§£

 *

 * @author ç§‹è¾žæœªå¯’

 * @date 2024-12-09

 */

@Documented

@Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})

@Retention(RUNTIME)

@Repeatable(EnumPattern.List.class) // å…è®¸åœ¨åŒä¸€å…ƒç´ ä¸Šå¤šæ¬¡ä½¿ç”¨è¯¥æ³¨è§£

@Constraint(validatedBy = {EnumPatternValidator.class})

public @interface EnumPattern {



    Class<? extends ValidateEnum> type() default ValidateEnum.class;



    String message() default "输入值不在枚举范围内";



    Class<?>[] groups() default {};



    Class<? extends Payload>[] payload() default {};



    @Documented

    @Target({METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE})

    @Retention(RUNTIME)

    @interface List {

        EnumPattern[] value();

    }



}


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

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,43 @@
package org.dromara.common.core.validate.enumd;



import org.dromara.common.core.utils.StringUtils;

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<? extends ValidateEnum> type = annotation.type();

                ValidateEnum[] constants = type.getEnumConstants();

                for (ValidateEnum e : constants) {

                    if (e.validate(value)) {

                        return true;

                    }

                }

            }

            return false;

        } catch (Exception e) {

            throw new ValidationException(e);

        }

    }



}

 ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/ValidateEnum.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,29 @@
package org.dromara.common.core.validate.enumd;



import org.dromara.common.core.utils.StringUtils;



/**

 * æžšä¸¾ç±»åž‹æ ¡éªŒæŽ¥å£

 *

 * @author ç§‹è¾žæœªå¯’

 * @date 2024-12-09

 */

public interface ValidateEnum {



    /**

     * èŽ·å–枚举code

     * <pre>该code用于匹配</pre>

     * @return æžšä¸¾code

     */

    String getCode();



    /**

     * æ ¡éªŒæžšä¸¾code

     * @param code æžšä¸¾code

     * @return æ ¡éªŒç»“æžœ

     */

    default boolean validate(String code) {

        return StringUtils.equals(code, getCode());

    }



}