Java Agent技能实现指南：从基础原理到实战避坑

1次阅读

共计 3165 个字符，预计需要花费 8 分钟才能阅读完成。

Java Agent 技术是 JVM 提供的一套强大机制，它能够在运行时动态修改字节码，为应用添加监控、调试、热修复等能力。在 APM（Application Performance Monitoring）领域，Agent 可以实现无侵入的性能数据采集；在热修复场景中，它能够绕过发布流程直接修复线上问题；对开发者而言，Agent 更是调试复杂系统的利器。

Java Agent 支持两种加载方式，它们各有适用场景：

特性	Premain	Agentmain
加载时机	JVM 启动时	JVM 运行中
入口方法	premain(String args, Instrumentation inst)	agentmain(String args, Instrumentation inst)
典型应用	监控启动过程、初始化埋点	热修复、动态诊断
使用复杂度	简单	需要 Attach API 配合
对应用影响	小	可能引起短暂停顿

在 META-INF/MANIFEST.MF 中必须包含这些配置项：

Premain-Class: com.example.MyAgent
Can-Redefine-Classes: true
Can-Retransform-Classes: true
Boot-Class-Path: lib/asm-9.2.jar

Premain-Class指定入口类
Can-Redefine-Classes允许类重定义
Can-Retransform-Classes允许类重新转换
Boot-Class-Path设置 Agent 依赖库路径

以下是使用 ASM 9.2 实现方法拦截的完整示例：

public class TimingTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(ClassLoader loader, String className, 
            Class<?> classBeingRedefined,
            ProtectionDomain protectionDomain, 
            byte[] classfileBuffer) {if (!className.startsWith("com/myapp")) {return null; // 只处理特定包}

        ClassReader reader = new ClassReader(classfileBuffer);
        ClassWriter writer = new ClassWriter(reader, ClassWriter.COMPUTE_MAXS);

        ClassVisitor visitor = new ClassVisitor(Opcodes.ASM9, writer) {
            @Override
            public MethodVisitor visitMethod(int access, String name, 
                    String descriptor, String signature, 
                    String[] exceptions) {
                MethodVisitor mv = super.visitMethod(access, name, 
                        descriptor, signature, exceptions);
                return new TimingMethodVisitor(mv, name);
            }
        };

        reader.accept(visitor, ClassReader.EXPAND_FRAMES);
        return writer.toByteArray();}
}

class TimingMethodVisitor extends MethodVisitor {
    private final String methodName;

    TimingMethodVisitor(MethodVisitor mv, String methodName) {super(Opcodes.ASM9, mv);
        this.methodName = methodName;
    }

    @Override
    public void visitCode() {
        // 方法开始插入计时
        visitMethodInsn(Opcodes.INVOKESTATIC, "java/lang/System", 
                "currentTimeMillis", "()J", false);
        visitVarInsn(Opcodes.LSTORE, 1);
        super.visitCode();}

    @Override
    public void visitInsn(int opcode) {if (opcode >= Opcodes.IRETURN && opcode <= Opcodes.RETURN) {
            // 方法返回前插入耗时计算
            visitMethodInsn(Opcodes.INVOKESTATIC, "java/lang/System", 
                    "currentTimeMillis", "()J", false);
            visitVarInsn(Opcodes.LLOAD, 1);
            visitInsn(Opcodes.LSUB);
            visitLdcInsn(methodName);
            visitMethodInsn(Opcodes.INVOKESTATIC, "com/myapp/Metrics", 
                    "record", "(JLjava/lang/String;)V", false);
        }
        super.visitInsn(opcode);
    }
}

线程安全：transform 方法可能被多线程调用，避免使用共享可变状态
快速失败：遇到意外输入时应立即返回 null 而非抛出异常
类过滤：明确过滤目标类（如示例中的包名前缀检查）
资源释放：确保 ClassReader/Writer 等资源被正确回收

使用 JMH 测试添加 Agent 前后的性能差异：

@BenchmarkMode(Mode.AverageTime)
@OutputTimeUnit(TimeUnit.MICROSECONDS)
public class AgentOverheadBenchmark {

    @Benchmark
    public void baseline() {// 原始方法调用}

    @Benchmark
    public void withAgent() {// 经过 Agent 增强的方法调用}
}

典型测试结果（仅供参考）：