如何利用 Java Agent 配合 ASM 字节码插桩实现生产环境的方法耗时无侵入审计-java教程-PHP中文网

Java Agent 必须在 premain 中调用 instrumentation.addTransformer 注册 ClassFileTransformer 才能生效，且需在 MANIFEST.MF 中声明 Premain-Class 和 Can-Retransform-Classes；否则 agent 静默失效。

如何利用 java agent 配合 asm 字节码插桩实现生产环境的方法耗时无侵入审计

premain 里必须注册 ClassFileTransformer 才能生效

Java Agent 不是写完 premain 就自动运行的。JVM 只会在找到正确声明的前提下，调用你的 premain 方法，并传入 Instrumentation 实例。漏掉这一步，整个 agent 彻底静默，连日志都不会打。

关键动作只有两行：

instrumentation.addTransformer(new MyTransformer(), true) —— 第二个参数设为 true 是为了后续能对已加载类做 retransformClasses
必须确保 MANIFEST.MF 中有：Premain-Class: com.example.MyAgent 和 Can-Retransform-Classes: true，缺一不可
jar 包要扁平化打包（比如用 maven-shade-plugin），ASM 等依赖没打进 agent jar，transform 方法一执行就抛 NoClassDefFoundError

transform 方法里别写耗时逻辑或日志

这个方法在类加载关键路径上被高频调用，哪怕加一行 System.out.println 或一次简单字符串拼接，都可能让 Spring Boot 启动慢几秒。它不是调试入口，而是字节码流水线的闸门。

Gradle 9.8.0

下载

只做最轻量判断：if (!className.equals("com/example/Service")) return null;
避免反射、正则、IO、锁、复杂对象构造 —— 这些全会卡住类加载器
上报耗时用无锁队列（如 ConcurrentLinkedQueue）+ 单独消费线程，绝不能同步阻塞在 transform 内
不要试图在 transform 里解析注解或读配置文件 —— 类还没初始化，ClassLoader 可能还没准备好

用 ASM 插入计时代码必须匹配 descriptor 且用 nanoTime

方法签名（descriptor）是字节码层面的精确标识，比如 login(Ljava/lang/String;)Z 和 login(Ljava/lang/String;I)Z 是两个完全不同的方法。错一个字符，插桩就失效。

入口插桩：在 visitCode() 后立即压入 System.nanoTime()，存到局部变量槽（用 LocalVariablesSorter 管理）
出口插桩：拦截 IRETURN/LRETURN/ARETURN/RETURN 指令，在其前插入计算差值 + 上报逻辑
绝对不用 System.currentTimeMillis() —— 它受系统时钟调整影响，可能倒退或跳变，nanoTime 才是唯一可靠的单调递增时钟
注意 void 方法只有 RETURN，返回对象的方法要用 ARETURN，不区分会抛 VerifyError

retransformClasses 是处理 Spring Bean 的唯一出路

Spring 的 Controller、Service 大部分在 Agent 初始化前就完成了类加载和实例化。此时 transform 已经错过它们，必须手动触发重转换。

先调用 instrumentation.getAllLoadedClasses() 扫出目标类，再过滤出你关心的（如含 "Controller" 或 "Service" 的类名）
调用 instrumentation.retransformClasses(classes)，前提是 Can-Retransform-Classes: true 且 JVM 支持（isRetransformClassesSupported() 返回 true）
注意：正在执行的方法栈帧不会被中断，新字节码只对下次调用生效；如果方法里有死循环，它会一直跑旧版本直到退出
别在 premain 里直接遍历所有类去 retransform —— 启动期大量类正在加载中，可能抛 UnmodifiableClassException