From 9e613488f1a01563913b4f6aade062980a315fea Mon Sep 17 00:00:00 2001
From: 秋辞未寒 <545073804@qq.com>
Date: 星期一, 09 十二月 2024 17:03:31 +0800
Subject: [PATCH] add 新增 validation支持枚举校验

---
 ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java |   43 +++++++++++++++++++++
 ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/ValidateEnum.java         |   29 ++++++++++++++
 ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPattern.java          |   39 +++++++++++++++++++
 3 files changed, 111 insertions(+), 0 deletions(-)

diff --git a/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPattern.java b/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPattern.java
new file mode 100644
index 0000000..2f7de75
--- /dev/null
+++ b/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();
+    }
+
+}
diff --git a/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java b/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/EnumPatternValidator.java
new file mode 100644
index 0000000..451dc71
--- /dev/null
+++ b/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);
+        }
+    }
+
+}
\ No newline at end of file
diff --git a/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/ValidateEnum.java b/ruoyi-common/ruoyi-common-core/src/main/java/org/dromara/common/core/validate/enumd/ValidateEnum.java
new file mode 100644
index 0000000..f3a367c
--- /dev/null
+++ b/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>璇ode鐢ㄤ簬鍖归厤</pre>
+     * @return 鏋氫妇code
+     */
+    String getCode();
+
+    /**
+     * 鏍￠獙鏋氫妇code
+     * @param code 鏋氫妇code
+     * @return 鏍￠獙缁撴灉
+     */
+    default boolean validate(String code) {
+        return StringUtils.equals(code, getCode());
+    }
+
+}

--
Gitblit v1.9.3