自己动手实现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.0Premain-Class: com.pack.agent.MonitorAgentCan-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.0Premain-Class: com.pack.agent.MonitorAgentCan-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 msindex 耗时:0 msindex 耗时:0 msindex 耗时:1008 msindex 耗时: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;}