字节码增强技术之 Java Agent 入门

前言

分布式链路追踪中为了获取服务之间调用链信息，采集器通常需要在方法的前后做埋点。在 Java 生态中，常见的埋点方式有两种：

1. 依赖 SDK 手动埋点；
2. 利用 Java Agent 技术来做无侵入埋点。

我们所熟知的分布式监控系统，是 Zipkin 开始的，最经典的是搞懂 X-B3 Ttrace 协议，使用 Brave SDK，手动埋点生成 Trace。但是 SDK 埋点的方式，对业务代码存在侵入性，当升级埋点时，必须要做代码的变更。

那么如何和业务逻辑解绑呢？

Java 还提供了另外一种方式：依赖 Java Agent 技术，修改目标方法的字节码，做到无侵入的埋点。这种利用 Java Agent 的方式的采集器，也叫做探针。在应用程序启动时使用 -javaagent 参数 ，或者运行时使用 attach(pid) 方式，就可以将探针包注入目标应用程序，完成埋点的植入。对业务代码无侵入的方式，可以做到无感的热升级。用户不需要理解深层的原理，就可以使用完整的监控服务

Java Agent 简介

Java Agent 是 Java 1.5 版本之后引⼊的特性，其主要作⽤是在 class 被加载之前对其拦截，已插⼊我们的监听字节码。使用 Java 的Instrumentation 接口（java.lang.instrument）来编写 Agent。

基本的思路是在 JVM 启动的时候添加一个代理（Java Agent），每个代理是一个 Jar 包，其 MANIFEST.MF 文件里指定了代理类，这个代理类包含一个 premain 方法。JVM 在类加载时候会先执行代理类的 premain 方法，再执行 Java 程序本身的 main 方法，这就是 premain 名字的来源。在 premain 方法中可以对加载前的 class 文件进行修改。

这种机制可以认为是虚拟机级别的 AOP，无需对原有应用做任何修改，就可以实现类的动态修改和增强。

从 JDK 1.6 开始支持更加强大的动态 Instrument，在JVM 启动后通过 Attach(pid) 远程加载。

注意：

无论是通过 Native 的方式还是通过 Java Instrumentation 接口的方式来编写 Agent，它们的工作都是借助 JVMTI 来进行完成。JVMTI 是一套 Native 接口，在 Java 1.5 之前，要实现一个 Agent 只能通过编写Native 代码来实现。

Java Instrumentation 核心方法

Instrumentation 是 java.lang.instrument 包下的一个接口，这个接口的方法提供了注册类文件转换器、获取所有已加载的类等功能，允许我们在对已加载和未加载的类进行修改，实现 AOP、性能监控等功能。

常用方法：

/**
 * 为 Instrumentation 注册一个类文件转换器，可以修改读取类文件字节码
 */
void addTransformer(ClassFileTransformer transformer, boolean canRetransform);

/**
 * 对JVM已经加载的类重新触发类加载
 */
void retransformClasses(Class<?>... classes) throws UnmodifiableClassException;

/**
 * 获取当前 JVM 加载的所有类对象
 */
Class[] getAllLoadedClasses()

它的 addTransformer 给 Instrumentation 注册一个 transformer，transformer 是 ClassFileTransformer 接口的实例，这个接口就只有一个 transform 方法，调用 addTransformer 设置 transformer 以后，后续 JVM 加载所有类之前都会被这个 tran