一个最简单的javaagent demo实例

港控/mmm° 2021-06-24 16:12 489阅读 0赞

JavaAgent 是JDK 1.5 以后引入的，也可以叫做Java代理。

JavaAgent 是运行在 main方法之前的拦截器，它内定的方法名叫 premain ，也就是说先执行 premain 方法然后再执行 main 方法。

那么如何实现一个 JavaAgent 呢？很简单，只需要增加 premain 方法即可，后续可以在此基础上实现注入拦截，AOP等。

package com.xifj.agent.demo;  
import java.lang.instrument.Instrumentation;  
   
/\*\*  
 \* Created by uc on 2018/4/18.  
 \*/  
public class agentDemo \{  
    /\*\*  
     \* 该方法在main方法之前运行，与main方法运行在同一个JVM中  
     \*  
     \* @param agentArgs  
     \* @param inst  
     \* @author xifeijian  
     \* @create  2018年4月18日  
     \*/  
    public static void premain(String agentArgs, Instrumentation inst) \{  
        System.out.println("=========premain方法执行1========");  
        System.out.println(agentArgs);  
    \}  
   
    /\*\*  
     \* 如果不存在 premain(String agentArgs, Instrumentation inst)  
     \* 则会执行 premain(String agentArgs)  
     \*  
     \* @param agentArgs  
     \* @author xifeijian  
     \* @create  2018年4月18日  
     \*/  
    public static void premain(String agentArgs) \{  
        System.out.println("=========premain方法执行2========");  
        System.out.println(agentArgs);  
    \}  
\}  
在这个 premain 函数中，开发者可以进行对类的各种操作。  
1、agentArgs 是 premain 函数得到的程序参数，随同 “– javaagent”一起传入。与 main 函数不同的是，这个参数是一个字符串而不是一个字符串数组，如果程序参数有多个，程序将自行解析这个字符串。

2、Inst 是一个 java.lang.instrument.Instrumentation 的实例，由 JVM 自动传入。java.lang.instrument.Instrumentation 是 instrument 包中定义的一个接口，也是这个包的核心部分，集中了其中几乎所有的功能方法，例如类定义的转换和操作等等。

写完这个类后，我们还需要做一步配置工作，在 src 目录下生成 META-INF/MANIFEST.MF 文件。

然后编辑META-INF/MANIFEST.MF，内容按如下定义：

要特别注意，最后一行是空行，还有就是Premain-Class冒号后面有个空格，例如：

Manifest-Version: 1.0  
Premain-Class: com.xifj.agent.demo.agentDemo

然后我们打包代码为 javaagent.jar（Build-Artifact）

接着我们在创建一个带有main方法的主程序工程，截图如下：

工程代码已完成，运行前配置VM参数：

\-javaagent:D:\\workspace\\javaagent\\out\\artifacts\\javaagent\_jar\\javaagent.jar=Hello -javaagent:D:\\workspace\\javaagent\\out\\artifacts\\javaagent\_jar\\javaagent.jar=World

运行agentTest程序，得到以下结果：

也可以将agentTest打成jar包通过java命令行执行，我们通过 -javaagent 参数来指定我们的Java代理包，值得一说的是 -javaagent 这个参数的个数是不限的，如果指定了多个，则会按指定的先后执行，执行完各个 agent 后，才会执行主程序的 main 方法。命令如下：

java -javaagent:D:\\workspace\\javaagent\\out\\artifacts\\javaagent\_jar\\javaagent.jar=hello1 -javaagent:D:\\workspace\\javaagent\\out\\artifacts\\javaagent\_jar\\javaagent.jar=hello2 -jar D:\\workspace\\myTest\\out\\artifacts\\myTest\_jar\\myTest.jar  
执行结果和上图是完全一致的。

特别提醒：如果你把 -javaagent 放在 -jar 后面，则不会生效。也就是说，放在主程序后面的 agent 是无效的。