ASM（动态）
前面那篇是静态修改字节码的方式，这种方式只是一种demo，方便初学者进行学习基本的API操作，真正在线上环境有作用的是动态的去修改字节码。
例如一个进程正在运行着，这个时候我们需要在业务无感知的情况下进行打印日志，或者计算接口的操作时间这种操作，这个时候我们就需要动态的去操作JVM中的字节码文件了
动态操作字节码
动态操作字节码的方式，其实还是主要使用agentmain，在javaagent的类attch到jvm上面的时候会调用这个方法执行内部的代码逻辑。
如何使用
首先我们需要有一个agent，这个agent实现了agentmain的方法，并且改写对应类的字节码文件。DynamicPreMainAddTimeStatAgent
public class DynamicPreMainAddTimeStatAgent {
public static void agentmain(String agentArgs, Instrumentation inst) throws Exception {
System.out.println("Agent Main called");
System.out.println("agentArgs:" + agentArgs);
inst.addTransformer(new ClassFileTransformer() {
public byte[] transform(ClassLoader loader, String className, Class<?> classBeingRedefined, ProtectionDomain protectionDomain, byte[] classfileBuffer) throws IllegalClassFormatException {
System.out.println("load transform");
if (className.equals("com/wsqandgy/asm/dynamic/Account")) {
System.out.println("meet com/wsqandgy/asm/dynamic/Account ");
/** classfileBuffer */
ClassReader classReader = new ClassReader(classfileBuffer);
ClassWriter cw = new ClassWriter(ClassWriter.COMPUTE_FRAMES | ClassWriter.COMPUTE_MAXS);
TimeStatClassAdapter timeStatClassAdapter = new TimeStatClassAdapter(cw);
classReader.accept(timeStatClassAdapter, ClassReader.SKIP_DEBUG);
byte[] data = cw.toByteArray();
return data;
} else {
System.out.println("load class:" + className);
System.out.println("new !");
return classfileBuffer;
}
}
},true);
System.out.println(Account.class.getClassLoader());
inst.retransformClasses(Account.class);
}
}
在上述操作中使用了TimeStatClassAdapter这个Class修改适配器和TimeStatMethodAdapter修改适配器。
增加TimeStat的方法，提供打印运行时间的方法。
public class TimeStat {
static ThreadLocal<Long> t = new ThreadLocal<Long>();
public static void start() {
t.set(System.currentTimeMillis());
}
public static void end() {
long time = System.currentTimeMillis();
System.out.print(Thread.currentThread().getStackTrace()[2] + " spend:");
System.out.println(time - t.get());
}
}
针对上面的内容创建MANIFEST.MF文件，并且进行打成jar包。
Manifest-Version: 1.0
Agent-Class: com.wsqandgy.asm.dynamic.DynamicPreMainAddTimeStatAgent
Can-Redefine-Classes: true
Can-Retransform-Classes: true
以上就是全部改写字节码文件的类和agentmain对应方法的类文件了。众所周知，我们需要对JVM进行注入就需要获取到全部的JVM运行信息，Java提供了一个方法可以获取到运行的JVM的相关信息。List<VirtualMachineDescriptor> list = VirtualMachine.list();
我们通过以上的方法获取到对应的JVM运行实例，进行attch操作。
public class AttachToolMain {
public static void main(String[] args) throws Exception {
List<VirtualMachineDescriptor> list = VirtualMachine.list();
for (VirtualMachineDescriptor machineDescriptor : list) {
if (machineDescriptor.displayName().equals("com.wsqandgy.asm.dynamic.RunLoopAccountMain")) {
VirtualMachine virtualMachine = VirtualMachine.attach(machineDescriptor.id());
System.out.println(new File("/Users/gongyan/Documents/home_code/tools/classes/artifacts/dynamicTimeAgent/dynamicTimeAgent.jar").exists());
virtualMachine.loadAgent("/Users/gongyan/Documents/home_code/tools/classes/artifacts/dynamicTimeAgent/dynamicTimeAgent.jar","argument for agent");
System.out.println("attach ok!");
virtualMachine.detach(); // 派遣操作，意思为生效
}
}
}
}
运行效果：
before
当我们运行AttachToolMain后，字节码重新生成已经增加了对应的响应时间了。
after
