From e23e7969e17931069711c4c179881a7a0c9ed299 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=99=BD=E8=8C=B6=E6=B8=85=E6=AC=A2?=
 <go_developer@163.com>
Date: Tue, 3 Jun 2025 17:28:35 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E6=88=90=E5=8F=8D=E5=B0=84=E8=8E=B7?=
 =?UTF-8?q?=E5=8F=96=E5=85=A8=E9=83=A8=E7=9A=84=E6=9E=9A=E4=B8=BE=E5=80=BC?=
 =?UTF-8?q?,=20=E7=BB=86=E8=8A=82=E5=BE=85=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/main/java/cn/zhangdeman/HashMapCache.java | 23 ++++++++++++++++---
 .../java/cn/zhangdeman/IExceptionConfig.java  |  6 +++++
 2 files changed, 26 insertions(+), 3 deletions(-)
 create mode 100644 src/main/java/cn/zhangdeman/IExceptionConfig.java

diff --git a/src/main/java/cn/zhangdeman/HashMapCache.java b/src/main/java/cn/zhangdeman/HashMapCache.java
index a7ba6b9..cc5dd20 100644
--- a/src/main/java/cn/zhangdeman/HashMapCache.java
+++ b/src/main/java/cn/zhangdeman/HashMapCache.java
@@ -9,7 +9,9 @@ import org.springframework.core.annotation.AnnotationUtils;
 import org.springframework.util.ClassUtils;
 
 import java.io.File;
+import java.lang.reflect.Constructor;
 import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
 import java.util.*;
 
 
@@ -43,9 +45,23 @@ public class HashMapCache implements ServletContextListener {
                     continue;
                 }
                 // 实现了接口
+                Map<String, String> codeTable = new HashMap<>();
                 // 反射获取类实例
-                ICustomExceptionAnnotation iCustomExceptionAnnotation = (ICustomExceptionAnnotation) (classInstance.getDeclaredConstructor().newInstance());
-                Map<String, String> codeTable = iCustomExceptionAnnotation.getCodeTable();
+                if (!classInstance.isEnum()) {
+                    // 不是枚举类
+                    Constructor<?> constructor = classInstance.getDeclaredConstructor();
+                    constructor.setAccessible(true);
+                    ICustomExceptionAnnotation iCustomExceptionAnnotation = (ICustomExceptionAnnotation) (constructor.newInstance());
+                    codeTable = iCustomExceptionAnnotation.getCodeTable();
+                } else {
+                    // 是枚举类
+                    Method method = classInstance.getMethod("values");
+                    Object[] objectList = (Object[]) method.invoke(null);
+                    for (Object itemEnumObject: objectList) {
+                        codeTable.put(((IExceptionConfig)itemEnumObject).getCode(), ((IExceptionConfig)itemEnumObject).getMessage());
+                    }
+                }
+
                 // TODO : 获取分类注解
                 for (Map.Entry<String, String> entry: codeTable.entrySet()) {
                     getInstance().put(entry.getKey(), entry.getValue());
@@ -125,7 +141,7 @@ public class HashMapCache implements ServletContextListener {
 
     @Override
     public void contextInitialized(ServletContextEvent sce) {
-        System.out.println("包内自动触发初始化" + new Date());
+        System.out.println("[start]自动执行异常配置注入" + new Date());
         try {
             this.loadCustomException();
         } catch (ClassNotFoundException e) {
@@ -139,6 +155,7 @@ public class HashMapCache implements ServletContextListener {
         } catch (InstantiationException e) {
             throw new RuntimeException(e);
         }
+        System.out.println("[finish]自动执行异常配置注入" + new Date());
     }
 
     @Override
diff --git a/src/main/java/cn/zhangdeman/IExceptionConfig.java b/src/main/java/cn/zhangdeman/IExceptionConfig.java
new file mode 100644
index 0000000..f01e779
--- /dev/null
+++ b/src/main/java/cn/zhangdeman/IExceptionConfig.java
@@ -0,0 +1,6 @@
+package cn.zhangdeman;
+
+public interface IExceptionConfig {
+    String getCode();
+    String getMessage();
+}