背景描述

javaagent是在JDK5之后提供的新特性，也可以叫java代理。开发者通过这种机制(Instrumentation)可以在加载class文件之前修改方法的字节码(此时字节码尚未加入JVM)，动态更改类方法实现AOP，提供监控服务如；方法调用时长、可用率、内存等。

开发简述

通过实现ClassFileTransformer接口方法，动态更改方法的字节码。在方法前后加上时间戳，最后执行完成输出执行时长。

环境准备

1、IntelliJ IDEA Community Edition 2018.3.1 x64

2、jdk1.8 64位

配置信息（路径相关修改为自己的）

1、java调试时配置

2.1、配置位置：Run/Debug Configurations ->VM options

2.2、配置内容(编译后的jar放到根目录下)：-javaagent:E:\itstack-demo-javaagent-1.0-SNAPSHOT.jar=agentargs

代码示例

pom.xml

 <dependencies>
 <dependency>
 <groupId>org.javassist</groupId>
 <artifactId>javassist</artifactId>
 <version>3.22.0-GA</version>
 </dependency>
 </dependencies>

MyAgent.java

MyTransformer.java

public class MyTransformer implements ClassFileTransformer {
 private final static String prefix = "\nlong startTime = System.currentTimeMillis();\n";
 private final static String postfix = "\nlong endTime = System.currentTimeMillis();\n";
 // 被处理的方法列表
 private final static Map<String, List<String>> methodMap = new HashMap<String, List<String>>();
 public MyTransformer() {
 //对指定方法监控
 add("org.itstack.demo.test.AgentTest.queryUserAge");
 add("org.itstack.demo.test.AgentTest.queryUserName");
 }
 private void add(String methodString) {
 String className = methodString.substring(0, methodString.lastIndexOf("."));
 String methodName = methodString.substring(methodString.lastIndexOf(".") + 1);
 List<String> list = methodMap.get(className);
 if (list == null) {
 list = new ArrayList<String>();
 methodMap.put(className, list);
 }
 list.add(methodName);
 }
 @Override
 public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined,
 ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
 className = className.replace("/", ".");
 // 判断加载的class的包路径是不是需要监控的类
 if (methodMap.containsKey(className)) {
 CtClass ctclass = null;
 try {
 ctclass = ClassPool.getDefault().get(className);// 使用全称,用于取得字节码类<使用javassist>
 for (String methodName : methodMap.get(className)) {
 String outputStr = "\nSystem.out.println(\"监控信息(执行耗时)：" + className + "." + methodName + " => \" +(endTime - startTime) +\"毫秒\");";
 CtMethod ctmethod = ctclass.getDeclaredMethod(methodName);// 得到这方法实例
 String newMethodName = methodName + "$new";// 新定义一个方法叫做比如queryUserAge$new
 ctmethod.setName(newMethodName);// 将原来的方法名字修改
 // 创建新的方法，复制原来的方法，名字为原来的名字
 CtMethod newMethod = CtNewMethod.copy(ctmethod, methodName, ctclass, null);
 // 构建新的方法体
 StringBuilder methodBodyStr = new StringBuilder();
 methodBodyStr.append("{");
 methodBodyStr.append(prefix);
 methodBodyStr.append(newMethodName + "($$);\n");// 调用原有代码，类似于method();($$)表示所有的参数
 methodBodyStr.append(postfix);
 methodBodyStr.append(outputStr);
 methodBodyStr.append("}");
 newMethod.setBody(methodBodyStr.toString());// 替换新方法
 ctclass.addMethod(newMethod); // 增加新方法
 }
 return ctclass.toBytecode();
 } catch (Exception e) {
 e.printStackTrace();
 }
 }
 return null;
 }
}

MANIFEST.MF

Manifest-Version: 1.0
Premain-Class: org.itstack.demo.agent.MyAgent
Can-Redefine-Classes: true

AgentTest.java

package org.itstack.demo.test;
import java.util.logging.Logger;
/**
 * vm options = -javaagent:E:\itstack-demo-javaagent-1.0-SNAPSHOT.jar=agentargs
 */
public class AgentTest {
 private static Logger logger = Logger.getLogger("AgentTest");
 public static void main(String[] args) {
 String userId = "100001";
 queryUserAge(userId);
 queryUserName(userId);
 }
 private static void queryUserAge(String userId) {
 try {
 Thread.sleep(300);
 logger.info("hello userId：" + userId +" age 18");
 } catch (InterruptedException e) {
 e.printStackTrace();
 }
 }
 private static void queryUserName(String userId) {
 try {
 Thread.sleep(100);
 logger.info("hello userId：" + userId +" name agent");
 } catch (InterruptedException e) {
 e.printStackTrace();
 }
 }
}

测试结果

2019-4-17 11:34:33 org.itstack.demo.test.AgentTest queryUserAge$new
信息: hello userId：100001 age 18
监控信息(执行耗时)：org.itstack.demo.test.AgentTest.queryUserAge => 316毫秒
2019-4-17 11:34:33 org.itstack.demo.test.AgentTest queryUserName$new
信息: hello userId：100001 name agent
监控信息(执行耗时)：org.itstack.demo.test.AgentTest.queryUserName => 100毫秒