Skip to content

Commit 081a258

Browse files
committed
fix: TongWeb8 Valve not work
1 parent 9494e43 commit 081a258

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

memshell/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -84,12 +84,18 @@ public void inject(Object context, Object valve) throws Exception {
8484
return;
8585
}
8686
Class valveClass = null;
87+
ClassLoader contextClassLoader = context.getClass().getClassLoader();
8788
try {
8889
// tongweb7
89-
valveClass = context.getClass().getClassLoader().loadClass("com.tongweb.catalina.Valve");
90+
valveClass = contextClassLoader.loadClass("com.tongweb.catalina.Valve");
9091
} catch (ClassNotFoundException e) {
91-
// tongweb6
92-
valveClass = context.getClass().getClassLoader().loadClass("com.tongweb.web.thor.Valve");
92+
try {
93+
// tongweb6
94+
valveClass = contextClassLoader.loadClass("com.tongweb.web.thor.Valve");
95+
} catch (ClassNotFoundException e1) {
96+
// tongweb8
97+
valveClass = contextClassLoader.loadClass("com.tongweb.server.Valve");
98+
}
9399
}
94100
invokeMethod(pipeline, "addValve", new Class[]{valveClass}, new Object[]{valve});
95101
System.out.println("valve injected successfully");

0 commit comments

Comments
 (0)