java.lang.instrument实现aop_jvm instrument aop-优快云博客

本文链接：https://blog.youkuaiyun.com/licheng989/article/details/45567587

本文介绍如何利用JDK 1.5后的instrument包进行字节码增强，包括实现ClassFileTransformer接口并使用javassist操作字节码，还探讨了cglib、动态代理等方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在jdk1.5之后引入了instrument包，关于这个包的介绍可以看官方文档http://docs.oracle.com/javase/8/docs/api/java/lang/instrument/package-summary.html

JVM运行的是二进制文件，我们可以通过instrument包的使用，在JVM运行之前修改JVM本要运行的二进制文件。

首先需要编写一个转换器类，它实现java.lang.instrument.ClassFileTransformer类。我们将这个类命名为MyClassFileTransformer，实现transform方法，同时写一个premain方法，有了这个方法，jvm就会在执行main之前按照transform来修改二进制文件。

我们还需要将这个类打成jar包命名为MyClassFileTransformer.jar，我将这个jar放到D:/workspace/agentDemo1/agentDemo1/WebContent/WEB-INF/lib/下。很重要的一点是修改jar包里面的MANIFEST.MF文件，在文件中加入Premain-Class: MyClassFileTransformer.jar。在运行main方法时配置-javaagent:D:/workspace/agentDemo1/agentDemo1/WebContent/WEB-INF/lib/MyClassFileTransformer.jar

下面是代码示例：

首先转换器类：

package org.aop.instrument;

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

import javassist.CannotCompileException;
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.NotFoundException;

public class MyClassFileTransformer implements ClassFileTransformer {

	/**
	 * 字节码加载到虚拟机前会进入这个方法
	 */
	public byte[] transform(ClassLoader loader, String className,
			Class<?> classBeingRedefined, ProtectionDomain protectionDomain,
			byte[] classfileBuffer) throws IllegalClassFormatException {
		System.out.println(className);
		// 如果加载Business类才拦截
        if (!"org/aop/model/Business".equals(className)) {   
            return null;   
        }   
  
        //javassist的包名是用点分割的，需要转换下   
        if (className.indexOf("/") != -1) {   
            className = className.replaceAll("/", ".");   
        }   
        try {   
            //通过包名获取类文件   
            CtClass cc = ClassPool.getDefault().get(className);   
            //获得指定方法名的方法   
            CtMethod m = cc.getDeclaredMethod("doSomething");   
            //在方法执行前插入代码   
            m.insertBefore("{System.out.println(\"记录日志\"); }");   
            return cc.toBytecode();   
        } catch (NotFoundException e) {   
        } catch (CannotCompileException e) {   
        } catch (IOException e) {   
            //忽略异常处理   
        }   
		return null;
	}

    public static void premain(String options, Instrumentation ins) {   
        //注册我自己的字节码转换器   
        ins.addTransformer(new MyClassFileTransformer());   
    }
	
}

驱动类：

package org.aop.instrument;

import org.aop.model.Business;

public class Driver {

	public static void main(String args[]) {
		new Business().doSomething();
		new Business().doSomething2();
	}
}

这是model类的接口

package org.aop.model;

public interface IBusiness {

	public void doSomething();
}

package org.aop.model;

public interface IBusiness2 {

	public void doSomething2();
}

这是model的实现类

package org.aop.model;

public class Business implements IBusiness, IBusiness2 {

	public void doSomething2() {
		System.out.println("执行业务逻辑2");
	}

	public void doSomething() {
		System.out.println("执行业务逻辑");
	}

}

最后的运行结果如图：