【进阶篇】二、实现Java Agent的静态加载和动态加载

文章目录

1、Java Agent
2、两种加载模式
- 静态加载模式
- 动态加载模式
3、静态加载模式实现
4、动态加载的实现

1、Java Agent

通过Java Agent，生成一种特殊的jar包（一种工具），业务程序可以主动去调用jar包里的方法。比如下面这个有打印内存情况功能的agent：

在这里插入图片描述

2、两种加载模式

让Java程序执行Java Agent程序中的代码，实现方式：

静态加载模式

在Java程序开始执行前，先执行Agent的代码。在Java Agent的项目中编写一个premain的方法，并打包成jar包：

//premain，即在用户的main方法执行之前
public static void premain(string agentArgs, Instrumentation inst)

如果自己的程序是test.jar，可执行：

java -javaagent:./agent.jar -jar test.jar

如此，JVM将会加载agent中的代码去执行。premain方法会在主线程中执行：

在这里插入图片描述

动态加载模式

连接用户的Java进程，随时让Java Agent的代码执行，适用于Arthas类似的诊断工具。实现动态加载模式，需要在Java Agent项目中编写一个agentmain方法，并打成jar包：

public static void agentmain(string agentArgs, Instrumentation inst)

以下代码可以让Java Agent的代码在指定的Java进程中执行：

//动态连接到24200进程ID的java程序
VirtualMachine vm = VirtualMachine.attach("24200");//加载java agent的jar并执行
vm.loadAgent("itheima-jvm-java-agent-jar-with-dependencies.jar");

如此，Java Agent的jar就会被执行。和静态加载的permain不同，agentmain方法的执行是在一个单独的线程里：

在这里插入图片描述

一点想法：感觉AOP和Java Agent的区别就是：前者是开发者在自己的程序里写的，且增强的时机更灵活。后者则独立于开发者自己的程序，能适用于所有的Java程序。

3、静态加载模式实现

创建Java Agent的maven项目，添加maven-assembly-plugin插件，以便打出java agent的jar包

<build><plugins><plugin><groupId>org.apache.maven.plugins</groupId><artifactId>maven-assembly-plugin</artifactId><configuration><!--将所有的依赖都打入同一个jar包中,配置了这个，以后打的Agent包，既有我写的Agent代码，也有上面引入的依赖的代码--><descriptorRefs><descriptorRef>jar-with-dependencies</descriptorRef></descriptorRefs><!--指定java agent相关的配置文件--><archive><manifestFile>src/main/resources/MANIFEST.MF</manifestFile></archive></configuration></plugin></plugins>
</build>

定义个类，写premain方法，写代理要执行逻辑

public class AgentMain {public static void premain(String agentArgs, Instrumentation inst) {System.out.println("premain方法执行了...");}}

编写MANIFEST.MF文件，写Java Agent的属性

Manifest-Version: 1.0
Premain-Class: com.llg.AgentMain
Can-Redefine-Classes: true
Can-Retransform-Classes: true
Can-Set-Native-Method-Prefix: true

使用maven-assembly-plugin进行打包

在这里插入图片描述

让另一个普通的Java程序，-javaagent启动，静态代理成功：

在这里插入图片描述

4、动态加载的实现

创建Java Agent的maven项目，添加maven-assembly-plugin插件，以便打出java agent的ja（同上）
定义个类，写agentmain方法，写代理要执行逻辑

public class AgentMain {public static void agentmain(String agentArgs,Instrumentation inst){System.out.println("agentmain执行了...");}}

编写MANIFEST.MF文件，写Java Agent的属性

Manifest-Version: 1.0
Agent-Class: com.llg.AgentMain
Can-Redefine-Classes: true
Can-Retransform-Classes: true
Can-Set-Native-Method-Prefix: true

使用maven-assembly-plugin进行打包（同上）
启动普通的Java程序
写个main方法，动态连接到运行中的java程序并加载执行Java Agent的Jar，实现动态代理（以后这个main方法就可以写在工具里，让用户只输入一个自己Java进程的PID，帮用户把传入到这个main，动态代理）

public class Test {public static void main(String[] args) throws  Exception {//传入用户Java进程的PID，实现连接VirtualMachine vm = VirtualMachine.attach("45627");//执行Java Agent的里的agentmain方法vm.loadAgent("D:\\jmh2\\llg-agent\\target\\llg-agent-1.0-SNAPSHOT-jar-with-dependencies.jar");}
}