javassist ClassPool 动态修改代码

给自己记录一下:

1. javaassist 很有用:

  1. 可以动态生成 class ,好多框架,比如 dubbo 就有用 javaassist 动态生成代理类。
  2. 还可以在运行时修改现有类的字节码,比如新增方法、修改方法等。有些 jar 包里有 license、加水印等限制,有了 javaassist 就可以修改字节码的实现,重写 method 的逻辑。
  3. 还可以修改 jar 里的代码实现,修改后重新打包,见

2. 使用例子:

  1. 增加 maven 依赖:
<dependencies>
        <dependency>
            <groupId>org.javassist</groupId>
            <artifactId>javassist</artifactId>
            <version>3.19.0-GA</version>
            <optional>true</optional>
        </dependency>
    </dependencies>
  1. 定义一个想要修改的目标类:
public class SayHelloService {
          
   
    public void sayHello() {
          
   
        System.out.println("hello");
    }
}
  1. 定义一个动态修改 sayHello() 方法实现的类:
import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.LoaderClassPath;

public class TestJavassist {
          
   
    public static void main(String[] args) throws Exception {
          
   
        // 注意这里需要插入 class path ,这里的 TestJavassist 不能是 SayHelloService 。因为一个类只能被加载一次,否则的话在
        // ctClass.toClass(); 会加载类,会报错的。
        ClassPool.getDefault().insertClassPath(new LoaderClassPath(TestJavassist.class.getClassLoader()));
        CtClass ctClass = ClassPool.getDefault().getCtClass("SayHelloService");
        CtMethod zzv = ctClass.getDeclaredMethod("sayHello");
        String methodBody = "{System.out.println("hello, changed");}";
        zzv.setBody(methodBody);
        // 调用之后会加载修改后的类
        ctClass.toClass();

        SayHelloService service = new SayHelloService();
        service.sayHello();
    }
}
  1. 运行 main() 方法,观察结果,发现确实方法的实现被改动了。
hello, changed
经验分享 程序员 微信小程序 职场和发展