61快乐

2026-09-22 01:30:43 +08:00 · 2024-06-01 12:20:17 +08:00
parent 1e4291e965
commit b6e34b31e2
99 changed files with 11806 additions and 128 deletions
@@ -0,0 +1,48 @@
+package jmg.godzilla.util;
+
+import jmg.core.config.Constants;
+import jmg.godzilla.memshell.GodzillaFilter;
+import jmg.godzilla.memshell.GodzillaInterceptor;
+import jmg.godzilla.memshell.GodzillaListener;
+import jmg.godzilla.memshell.GodzillaWebFluxHandlerMethod;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class ShellUtil {
+
+    private static final Map<String, String> SHELL_CLASSNAME_MAP = new HashMap();
+    private static final Map<String, Map<String, String>> toolMap = new HashMap();
+
+    public ShellUtil() {
+    }
+
+    public static String getShellName(String toolType, String shellType) {
+        Map<String, String> shellMap = toolMap.get(toolType);
+        return shellMap == null ? "" : shellMap.getOrDefault(shellType, "");
+    }
+
+    public static String getShellClassName(String shellName) throws Exception {
+        if (SHELL_CLASSNAME_MAP.get(shellName) == null) {
+            throw new Exception("Invalid shell type '" + shellName + "'");
+        } else {
+            return SHELL_CLASSNAME_MAP.getOrDefault(shellName, "");
+        }
+    }
+
+    static {
+        SHELL_CLASSNAME_MAP.put(GodzillaFilter.class.getSimpleName(), GodzillaFilter.class.getName());
+        SHELL_CLASSNAME_MAP.put(GodzillaListener.class.getSimpleName(), GodzillaListener.class.getName());
+        SHELL_CLASSNAME_MAP.put(GodzillaInterceptor.class.getSimpleName(), GodzillaInterceptor.class.getName());
+        SHELL_CLASSNAME_MAP.put(GodzillaWebFluxHandlerMethod.class.getSimpleName(), GodzillaWebFluxHandlerMethod.class.getName());
+
+        Map<String, String> godzillaMap = new HashMap();
+        godzillaMap.put(Constants.SHELL_FILTER, GodzillaFilter.class.getSimpleName());
+        godzillaMap.put(Constants.SHELL_LISTENER, GodzillaListener.class.getSimpleName());
+        godzillaMap.put(Constants.SHELL_INTERCEPTOR, GodzillaInterceptor.class.getSimpleName());
+        godzillaMap.put(Constants.SHELL_WF_HANDLERMETHOD, GodzillaWebFluxHandlerMethod.class.getSimpleName());
+        toolMap.put("Godzilla", godzillaMap);
+    }
+
+
+}