javaagent字节码增强浅尝

概述

• javaagent 技术广泛应用于对代码的增强，比如统计方法执行时间、GC 信息打印、分布式链路跟踪等；
• 实现方式包括 javassist 和 bytebuddy，bytebuddy 是对 javassist 的改进；
• 类似于 spring 中的 AOP；

Instrumentation 实现对 main 方法增强

说明：

• 新建两个项目，一个打包成增强 jar 包，一个写待增强业务代码；
• 可通过定义注解进行优化；
• 注意 maven 打包插件；

1）新建一个 maven 项目，引入 pom 依赖，并指定增强类：

<dependencies><!-- javaagent --><dependency><groupId>javassist</groupId><artifactId>javassist</artifactId><version>3.12.1.GA</version><type>jar</type></dependency>
</dependencies><build><plugins><!-- 测试javaagent相关 --><plugin><!-- 打普通的jar包 --><groupId>org.apache.maven.plugins</groupId><artifactId>maven-jar-plugin</artifactId><version>2.2</version><configuration><!-- 打包配置 --><archive><manifestEntries><Project-name>${project.name}</Project-name><Project-version>${project.version}</Project-version><!-- main方法之前需要处理的类 --><Premain-Class>com.firechou.test.javaagent.FireAgent</Premain-Class></manifestEntries></archive><skip>true</skip></configuration></plugin></plugins>
</build>

注意插件中 Premain-Class 指定的是增强类。

2）新建代理类，编写增强逻辑

public class FireAgent {/*** 在Instrumentation中有对应的说明：当jvm以代理类的方式启动时，Instrumentation的实例将传递给premain方法* @param args* @param instrumentation*/public static void premain(String args, Instrumentation instrumentation){System.out.println("agent premain(), args: " + args);}}

3） 打包成 jar
执行打包命令：

注意：不要执行 Plugins》jar》jar:jar 命令，会提示找不到目标类。
生成增强 jar 包如下：

4）在另一个项目中，新建业务代码类，即被增强的逻辑

public class FireAgentMainTest {public static void main(String[] args) {System.out.println("hello world");}
}

添加 JVM 运行参数：

-javaagent:test-javaagent/target/test-javaagent-1.0-SNAPSHOT.jar=name=jar&age=33

运行结果：

agent premain(), args: name=jar&age=33
hello world