javassist ClassPool 动态修改代码

给自己记录一下:

1. javaassist 很有用:

  1. 可以动态生成 class ,好多框架,比如 dubbo 就有用 javaassist 动态生成代理类。

  2. 还可以在运行时修改现有类的字节码,比如新增方法、修改方法等。有些 jar 包里有 license、加水印等限制,有了 javaassist 就可以修改字节码的实现,重写 method 的逻辑。

  3. 还可以修改 jar 里的代码实现,修改后重新打包,见https://blog.csdn.net/shfqbluestone/article/details/107411956

2. 使用例子:

  1. 增加 maven 依赖:
    
        
            org.javassist
            javassist
            3.19.0-GA
            true
        
    
  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

你可能感兴趣的:(javassist,java)