手把手教你实现一个方法耗时统计的 java agent

前面两个方法比较重要，addTransformer 方法配置之后，后续的类加载都会被 Transformer 拦截。对于已经加载过的类，可以执行 retransformClasses 来重新触发这个 Transformer 的拦截。类加载的字节码被修改后，除非再次被 retransform，否则不会恢复。

通过上面的描述，可知

• 可以通过Transformer修改类
• 类加载时，会被触发 Transformer 拦截

我们需要统计方法耗时，所以想到的就是在方法的执行前，记录一个时间，执行完之后统计一下时间差，即为耗时

直接修改字节码有点麻烦，因此我们借助神器javaassist来修改字节码

实现自定义的ClassFileTransformer，代码如下

public class CostTransformer implements ClassFileTransformer {
    @Override
    public byte[] transform(ClassLoader loader, String className, Class classBeingRedefined,
     ProtectionDomain protectionDomain, byte[] classfileBuffer) {
 // 这里我们限制下，只针对目标包下进行耗时统计
 if (!className.startsWith("com/git/hui/java/")) {
     return classfileBuffer;
 }

 CtClass cl = null;
 try {
     ClassPool classPool = ClassPool.getDefault();
     cl = classPool.makeClass(new ByteArrayInputStream(classfileBuffer));

     for (CtMethod method : cl.getDeclaredMethods()) {
  // 所有方法，统计耗时；请注意，需要通过`addLocalVariable`来声明局部变量
  method.addLocalVariable("start", CtClass.longType);
  method.insertBefore("start = System.currentTimeMillis();");
  String methodName = method.getLongName();
  method.insertAfter("System.out.println("" + methodName + " cost: " + (System" +
   ".currentTimeMillis() - start));");
     }

     byte[] transformed = cl.toBytecode();
     return transformed;
 } catch (Exception e) {
     e.printStackTrace();
 }
 return classfileBuffer;
    }
}

public class SimpleAgent {

    
    public static void premain(String agentArgs, Instrumentation inst) {
 System.out.println("premain");
 customLogic(inst);
    }

    
    public static void agentmain(String agentArgs, Instrumentation inst) {
 System.out.println("agentmain");
 customLogic(inst);
    }

    
    private static void customLogic(Instrumentation inst) {
 inst.addTransformer(new CostTransformer(), true);
    }
}

public class DemoClz {

    public int print(int i) {
 System.out.println("i: " + i);
 try {
     Thread.sleep(1000);
 } catch (InterruptedException e) {
     e.printStackTrace();
 }
 return i + 2;
    }

    public int count(int i) {
 System.out.println("cnt: " + i);
 try {
     Thread.sleep(50);
 } catch (InterruptedException e) {
     e.printStackTrace();
 }
 return i + 1;
    }
}

public class baseMain {
    public static void main(String[] args) throws InterruptedException {
 DemoClz demoClz = new DemoClz();
 int cnt = 0;
 for (int i = 0; i < 20; i++) {
     if (++cnt % 2 == 0) {
  i = demoClz.print(i);
     } else {
  i = demoClz.count(i);
     }
 }
    }
}

public class baseMain {
    public int print(int i) {
 System.out.println("i: " + i);
 try {
     Thread.sleep(1000);
 } catch (InterruptedException e) {
     e.printStackTrace();
 }
 return i + 2;
    }

    public void run() {
 int i = 1;
 while (true) {
     i = print(i);
 }
    }

    public static void main(String[] args) {
 baseMain main = new baseMain();
 main.run();
}