『互联网架构』插桩处理埋点

上节说了javaagent和javassist，其实javassist也是基于ASM实现的。一般人不懂得JVM指令的话，根本ASM搞不起来，也用到了访问者的设计模式，看起来跟咱们写代码不是一个套路，学习成本比较高，所以有了javassist。

（一）插桩

• 修改上节源码，进行插桩

IdigAgentTest

package com.idig8;

import com.idig8.agent.test.UserServiceImpl;

import java.lang.instrument.Instrumentation;

public class IdigAgentTest {public static void main(String[] args) {System.out.println("hello world idig8!");UserServiceImpl userService = new UserServiceImpl();userService.hello();}}

 > UserServiceImpl 增加hello方法，打印方法里面的内容 ``` java package com.idig8.agent.test; public class UserServiceImpl { public UserServiceImpl(){ System.out.println("hello world！"); } public void hello(){ String p1 ="100"; System.out.print("p1 = "+p1); } }

IdigAgent 增加插桩的方法

package com.idig8.agent.test;

import javassist.*;

import java.io.IOException;import java.lang.instrument.ClassFileTransformer;import java.lang.instrument.IllegalClassFormatException;import java.lang.instrument.Instrumentation;import java.security.ProtectionDomain;

public class IdigAgent {public static void premain(String args, Instrumentation instrumentation) {System.out.println("premain：" + args);

instrumentation.addTransformer(new ClassFileTransformer() { @Override public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException { if (!"com/idig8/agent/test/UserServiceImpl".equals(className)) { return null; } // javassist 工具 改造 try { ClassPool pool = new ClassPool(); pool.insertClassPath(new LoaderClassPath(loader)); CtClass ctclass = pool.get("com.idig8.agent.test.UserServiceImpl"); CtMethod method = ctclass.getDeclaredMethod("hello"); method.insertBefore(" System.out.println(System.currentTimeMillis());");

// method.insertBefore("long begin = System.currentTimeMillis();"// +" System.out.println(begin);");//// method.insertAfter(" long end = System.currentTimeMillis();\n" +// " System.out.println(end - begin);");return ctclass.toBytecode();} catch (NotFoundException e) {e.printStackTrace();} catch (CannotCompileException e) {e.printStackTrace();} catch (IOException e) {e.printStackTrace();}

return null; } }); }

}

 > pom中添加javassist的jar包 ``` java <dependency> <groupId>org.javassist</groupId> <artifactId>javassist</artifactId> <version>3.18.1-GA</version> </dependency>

• 代码的执行流程

  IdigAgentTest执行这个方法IdigAgent ->IdigAgentTest->UserServiceImpl的类->UserServiceImpl被插桩启动开始的hello-> 打印hello方法体

  premain：abc //IdigAgent hello world idig8! //IdigAgentTest hello world！ //UserServiceImpl的类 1562479947074 //UserServiceImpl被插桩启动开始的hello p1 = 100 //打印hello方法体

  

• 注意代码

  加入insertBefore中的bgin 和 insertAfter 的 end 通过end-begin 但是后台报错了

javassist.CannotCompileException: [source error] no such field: begin

为什么呢，因为插桩的时候都是以代码快的形式，局部变量。

在实际开发中不用修改原有的方法，而是会新写一个方法，在新方法进行出来，调用要插桩的方法。新方法的参数和要插桩的保持一致，包括注解，参数。其实有点类似动态代理。

public void hello$agent() { { long begin = System.currentTimeMillis(); try{ hello(); }finally { long end = System.currentTimeMillis(); System.out.println(end - begin); } } }

具体如何实现，后面会说

PS：还需要结合之前文章，了解如何完成插桩和埋点。