CodeWalk

Java Instrumentation实现热替换与动态追踪

作者：专业代码师 · 2026-05-30 12:55

如何利用Java Instrumentation API实现运行时热替换类和动态方法追踪？包括retransformClasses的限制、redefineClasses与retransform的区别、以及动态代理在无侵入监控中的应用。

回答

专业代码师

1. redefineClasses vs retransformClasses

redefineClasses：

inst.redefineClasses(new ClassDefinition(targetClass, newClassBytes));
  • 直接替换类的字节码
  • 限制：
    • 不能添加/删除字段和方法
    • 不能修改父类或实现的接口
    • 只能修改方法体
  • 支持添加方法（JDK 9+ -XX:+AllowRedefinitionToAddDeleteMethods）

retransformClasses：

inst.addTransformer(transformer, true);  // canRetransform=true
inst.retransformClasses(targetClass);
  • 重新执行已注册的ClassFileTransformer
  • 更灵活，可叠加多个Transformer
  • 同样受redefine限制约束

2. 动态方法追踪实现

public class TraceTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(ClassLoader loader, String className,
                           Class<?> classBeingRedefined,
                           ProtectionDomain protectionDomain,
                           byte[] classfileBuffer) {
        if (!className.equals("com/example/UserService")) return null;
        
        ClassReader cr = new ClassReader(classfileBuffer);
        ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES);
        cr.accept(new ClassVisitor(Opcodes.ASM9, cw) {
            @Override
            public MethodVisitor visitMethod(int access, String name, String desc,
                                           String signature, String[] exceptions) {
                MethodVisitor mv = super.visitMethod(access, name, desc, signature, exceptions);
                return new MethodVisitor(Opcodes.ASM9, mv) {
                    @Override
                    public void visitCode() {
                        // 在方法开头插入System.out.println
                        mv.visitFieldInsn(GETSTATIC, "java/lang/System", "out", "Ljava/io/PrintStream;");
                        mv.visitLdcInsn("Entering: " + name);
                        mv.visitMethodInsn(INVOKEVIRTUAL, "java/io/PrintStream", "println", "(Ljava/lang/String;)V");
                        super.visitCode();
                    }
                };
            }
        }, 0);
        return cw.toByteArray();
    }
}

3. 动态Attach到运行中JVM

// 获取目标JVM的PID
VirtualMachine vm = VirtualMachine.attach(pid);
// 加载Agent jar
vm.loadAgent("/path/to/agent.jar", args);
vm.detach();

4. 常见应用工具

  • Arthas：watch/trace/tunnel/jad 命令
  • Btrace：通过脚本注入追踪代码
  • Alibaba Cloud Toolkit：远程诊断

5. 注意事项

  • 热替换触发Full GC（JDK 8的Bug，JDK 11+修复）
  • 被替换类的当前栈帧仍使用旧代码
  • 不能修改Lambda表达式目标方法（invokedynamic）