java agent模式_Java Agent的agentmain和premain方式

agentmain方式

首先假如我们有个应用1启动类为task，

package com.st;

import java.lang.instrument.Instrumentation;

import java.util.Scanner;

public class Task {

public static void main (String[] args) {

Scanner scanner = new Scanner(System.in);

scanner.hasNext();

}

}

需要执行的代码

package com.st;

import java.lang.instrument.Instrumentation;

public class AgentMainTest {

public static void agentmain(String agentArgs, Instrumentation inst){

System.out.println("打印全部加载的类：");

Class[] allLoadedClasses = inst.getAllLoadedClasses();

for (Class allLoadedClass : allLoadedClasses) {

System.out.println(allLoadedClass.getName());

}

}

}

maven打包配置：

maven-jar-plugin

3.0.2

true

com.st.Task

com.st.AgentMainTest

打包成ks-demo-1.0-SNAPSHOT.jar，将其运行起来java -jar ks-demo-1.0-SNAPSHOT.jar

创建一个java类做应用2，导入jdk目录下的lib/tools.jar包，运行main方法

package com.st;

import com.sun.tools.attach.AgentInitializationException;

import com.sun.tools.attach.AgentLoadException;

import com.sun.tools.attach.AttachNotSupportedException;

import com.sun.tools.attach.VirtualMachine;

import java.io.IOException;

/**

* Hello world!

**/

public class App

{

public static void main( String[] args )

{

try {

//VirtualMachine 来自tools.jar

// VirtualMachine.attach("9444") 9444为线程PID

VirtualMachine vm = VirtualMachine.attach("9444");

//指定应用路径

vm.loadAgent("D:\\work\\ks-demo\\target\\ks-demo-1.0-SNAPSHOT.jar");

} catch (AttachNotSupportedException e) {

e.printStackTrace();

} catch (IOException e) {

e.printStackTrace();

} catch (AgentLoadException e) {

e.printStackTrace();

} catch (AgentInitializationException e) {

e.printStackTrace();

}

}

}

该功能多用于收集系统信息，以及在系统加载后对class文件重新编辑等功能。

premain方式

package com.st;

public class Task {

public static void main (String[] args) {

System.out.println("mian执行");

}

}

package com.st;

import java.lang.instrument.Instrumentation;

public class AgentMainTest {

public static void premain(String agentArgs, Instrumentation inst){

System.out.println("premain执行：");

}

}

maven-jar-plugin

3.0.2

true

com.st.Task

com.st.AgentMainTest

运行命令：java -javaagent:ks-demo-1.0-SNAPSHOT.jar  -jar ks-demo-1.0-SNAPSHOT.jar

运行结果：

premain在应用加载前执行，通常用于加载重新编辑类