自己动手实现Agent统计API接口调用耗时

public interface Instrumentation {
  /**
   * 注册提供的转换器。以后的所有类定义都可以通过转换器看到，但所有已注册的转换器所依赖的类定义除外。
   * 如果注册了多个转换器，那么会按添加的顺序调用它们。
   * 如果转换器在执行期间抛出异常，则 JVM 仍将按顺序调用其他已注册的转换器。
   * 可以多次添加同一转换器。在任何外部 JVMTI ClassFileLoadHookAll 事件监听器看到类文件之前，用 addTransformer 注册的所有转换器始终可以看到类文件。
   */
  void addTransformer(ClassFileTransformer transformer);
  /**
   * 注销提供的转换器。以后的类定义将不显示给该转换器。
   * 移除最近添加的转换器的匹配实例。由于类加载的多线程特性，在调用被移除后，转换器还可能接收调用。
   * 所以编写的转换器应防止出现这种情况。
   */
  boolean removeTransformer(ClassFileTransformer transformer);
  /**
   * 返回当前 JVM 配置是否支持类的重定义。
   * 重定义已加载类的能力是 JVM 的一个可选功能。
   * 在执行单个 JVM 的单实例化过程中，对此方法的多个调用将始终返回同一应答。
   */
  boolean isRedefineClassesSupported();
  /**
   * 使用提供的类文件重新定义提供的类集。
   * 此方法用于在不引用现有类文件字节的情况下替换类的定义，就像从源代码重新编译以修复并继续调试时可能会做的那样。
   * 如果要转换现有的类文件字节（例如字节码插入），则应使用retransformClassess。
   * 此方法对一个集合进行操作，以便允许同时对多个类进行相互依存的更改（对类a的重新定义可能需要对类B的重新定义）。
   */
  void redefineClasses(ClassDefinition... definitions) throws  ClassNotFoundException, UnmodifiableClassException;
  /**
   * 返回JVM当前加载的所有类的数组。
   * 返回的数组包括所有类和接口，包括隐藏类或接口，以及所有类型的数组类。
   */
  Class[] getAllLoadedClasses();
  /**
   * 返回指定对象所消耗的存储量的特定于实现的近似值。
   * 该结果可以包括对象的一些或全部开销，因此对于实现内部的比较是有用的，但对于实现之间的比较则不有用。
   * 在JVM的一次调用过程中，估计值可能会发生变化。 
   */
  long getObjectSize(Object objectToSize);
}

* 类文件的转换器。代理使用addTransformer方法注册该接口的实现，以便在加载、重新定义或重新转换类时调用转换器的转换方法。
 */
public interface ClassFileTransformer {

  /**
   * 转换给定的类文件并返回新的替换类文件
   * loader: 要转换的类的定义加载程序，如果引导加载程序，则可以为null。
   * className: 类的名称，内部形式为完全限定的类和接口名称，如Java虚拟机规范中定义的那样。例如，“java/util/List”。
   * classBeingRedefined: 如果这是由重新定义或重传触发的，则类被重新定义或重新转换；如果这是类装入，则为null。
   * protectionDomain: 正在定义或重新定义的类的保护域。
   * classfileBuffer: 类文件格式的输入字节缓冲区-不能修改
   */
  default byte[] transform(ClassLoader loader, 
    String className, 
    Class<?> classBeingRedefined, 
    ProtectionDomain protectionDomain, 
    byte[] classfileBuffer) throws IllegalClassFormatException {
    return null;
  }
}

Manifest-Version: 1.0
Premain-Class: com.pack.agent.MonitorAgent
Can-Redefine-Classes: true

<!--将通过javassit来修改类信息-->
<dependency>
  <groupId>org.javassist</groupId>
  <artifactId>javassist</artifactId>
  <version>3.29.2-GA</version>
</dependency>

public class MonitorTransformer implements ClassFileTransformer {


  @Override
  public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined,
      ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
    // 在上面的API说明中已经说了，这里的className不是. 而是'/'
    className = className.replace("/", ".");
    // 这里只拦截com.pack及子包下的类
    if (className.startsWith("com.pack")) {
      try {
        // 以下的相关API就是javassit；可自行查看javassit相关的文章
        CtClass ctClass = ClassPool.getDefault().get(className) ;
        CtMethod[] ctMethods = ctClass.getDeclaredMethods() ;
        for (CtMethod ctMethod : ctMethods) {
          // 获取执行的方法名称
          String methodName = ctMethod.getName() ;
          // 打印方法执行耗时时间
          String executeTime = "\nSystem.out.println(\"" + methodName + " 耗时:\" + (end - start) + " + "\" ms\");\n" ;
          // 添加2个局部变量
          ctMethod.addLocalVariable("start", CtClass.longType) ;
          ctMethod.addLocalVariable("end", CtClass.longType) ;
          // 为上面2个局部变量赋值
          ctMethod.insertBefore("start = System.currentTimeMillis() ;\n") ;
          ctMethod.insertAfter("end = System.currentTimeMillis();\n") ;
          // 将打印时间的语句插入到方法体的最后一行
          ctMethod.insertAfter(executeTime) ;
        }
        // 返回修改后的字节码（这里就是重写字节码文件）
        return ctClass.toBytecode();
      } catch (Exception e) {
        e.printStackTrace() ;
      }
    }
    return null;
  }
}

public class MonitorAgent {


  // 这里的premain是我们Agent的入口，首先执行的就是该premain，然后才是main
  // agentArgs是agent运行时添加的参数，我们可以在下面看到如何定义参数
  public static void premain(String agentArgs, Instrumentation instrumentation) {
    // 添加转换器
    instrumentation.addTransformer(new MonitorTransformer());
  }
  // 这里完全没必要main，只是为了在eclipse中生成jar包方便
  public static void main(String[] args) {
  }
}

Manifest-Version: 1.0
Premain-Class: com.pack.agent.MonitorAgent
Can-Redefine-Classes: true

@RestController
@RequestMapping("/demos")
public class DemoController {


  @GetMapping("/index")
  public Object index() throws Exception {
    TimeUnit.SECONDS.sleep(new Random().nextInt(5)) ;
    return "success" ;
  }
}

java -javaagent:CostAgent.jar -jar test.jar

index 耗时:0 ms
index 耗时:0 ms
index 耗时:0 ms
index 耗时:1008 ms
index 耗时:2012 ms

jad com.pack.DemoController

@GetMapping(value={"/index"})
publicObject index() throws Exception {
   long start = System.currentTimeMillis();
   TimeUnit.SECONDS.sleep(new Random().nextInt(5));
   String string= "success";
   long l = System.currentTimeMillis();
   String string2 = string;
   System.out.println(new StringBuffer().append("index 耗时:").append(l - var1_1).append(" ms").toString());
   return string2;
}