Java Agent学习路线：从基础到生产环境的系统化实践指南

1次阅读

没有评论

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

在 Java 开发中，传统的 AOP（面向切面编程）方案虽然可以实现方法拦截和增强，但是存在一些明显的局限性。

侵入性强：需要在代码中显式引入切面定义，对业务代码有侵入性
功能受限：无法在类加载阶段修改字节码，难以实现某些底层功能
性能损耗：运行时动态代理会带来额外的性能开销

相比之下，Java Agent 技术提供了更底层的解决方案：

无侵入性：无需修改源代码，通过字节码增强实现功能
功能强大：可以在类加载阶段修改字节码，实现更复杂的增强逻辑
性能更优：类加载时的修改通常比运行时动态代理效率更高

在字节码操作工具的选择上，ASM 和 Javassist 是两种主流方案：

ASM：
性能更高，直接操作字节码指令
学习曲线陡峭，需要了解 JVM 字节码
适合对性能要求极高的场景
Javassist：
提供更高级的 API，使用 Java 语法操作字节码
开发效率更高，适合快速实现
性能略低于 ASM，但对大多数场景足够

Java 提供了两种 Agent 加载方式：

Instrumentation API：
通过 -javaagent 参数在 JVM 启动时加载
适合需要早期介入的场景
实现 premain 方法作为入口
Attach API：
可以在 JVM 运行时动态加载 Agent
需要目标 JVM 启用 Attach 机制
实现 agentmain 方法作为入口

下面是一个使用 Javassist 实现方法耗时统计 Agent 的完整示例：

创建 Premain 类

public class TimingAgent {public static void premain(String agentArgs, Instrumentation inst) {inst.addTransformer(new ClassFileTransformer() {
            @Override
            public byte[] transform(ClassLoader loader, String className, 
                                  Class<?> classBeingRedefined,
                                  ProtectionDomain protectionDomain,
                                  byte[] classfileBuffer) {
                try {if (!className.startsWith("com/yourpackage/")) {return null; // 只处理特定包下的类}

                    ClassPool pool = ClassPool.getDefault();
                    CtClass ctClass = pool.makeClass(new ByteArrayInputStream(classfileBuffer));

                    for (CtMethod method : ctClass.getDeclaredMethods()) {
                        method.insertBefore("long start = System.currentTimeMillis();");
                        method.insertAfter("System.out.println(\"" + method.getName() + "executed in \" + "+"(System.currentTimeMillis() - start) + \"ms\");");
                    }

                    return ctClass.toBytecode();} catch (Exception e) {e.printStackTrace();
                    return null;
                }
            }
        });
    }
}

MANIFEST.MF 配置

Manifest-Version: 1.0
Premain-Class: com.yourpackage.TimingAgent
Can-Redefine-Classes: true
Can-Retransform-Classes: true

Attach 动态加载示例

public class AgentLoader {public static void loadAgent(String pid, String agentJarPath) {
        VirtualMachine vm = null;
        try {vm = VirtualMachine.attach(pid);
            vm.loadAgent(agentJarPath);
        } catch (Exception e) {throw new RuntimeException("Failed to attach agent", e);
        } finally {if (vm != null) {
                try {vm.detach();
                } catch (IOException e) {// Log warning}
            }
        }
    }
}

为了避免 Agent 与应用程序的类加载冲突，推荐使用独立的 ClassLoader：