【Javassist】快速入门系列04 使用Javassist更改整个方法体

系列文章目录

01 在方法体的开头或结尾插入代码
02 使用Javassist实现方法执行时间统计
03 使用Javassist实现方法异常处理
04 使用Javassist更改整个方法体

---

---

前言

上一章我们介绍了使用Javassist实现方法异常处理，学会了Javassist的addCatch()方法使用。本章主要介绍使用Javassist更改整个方法体，setBoday方法的使用。

---

引入Javassist jar包

在上几篇文章已经引入了javassist的jar包，如果你是第一次观看本系列文章，也可以复制以下maven依赖将jar包导入工程。

        <dependency>
            <groupId>org.javassist</groupId>
            <artifactId>javassist</artifactId>
            <version>3.28.0-GA</version>
        </dependency>

使用Javassist更改整个方法体

/**
 * 【Javassist】快速入门系列04 使用Javassist更改整个方法体
 * 公众号&B站：精致的王同学
 * @author 精致的王同学
 * @date 2022/12/21 16:25
 */
public class Basic04ChangeMethodBody {
    public static void main(String[] args) throws Exception{
        // 获取javassist默认类池
        ClassPool pool = ClassPool.getDefault();
        // 获取domain.Hello的CtClass对象
        CtClass ctClass = pool.get("domain.Hello");
        // 获取domain.Hello类的say方法
        CtMethod method = ctClass.getDeclaredMethod("say");
        // 更改整个方法体方法setBody
        method.setBody("{System.out.println(\"Javassist setBody!\");}");
        // 将类写成文件
        ctClass.writeFile();
        // 获取domain.Hello的clazz对象
        Class<?> clazz = ctClass.toClass();
        // 获取修改后的domain.Hello的实例
        Object obj = clazz.newInstance();
        // 获取修改后的say方法
        Method say = clazz.getDeclaredMethod("say");
        // 模拟调用修改后的say方法
        say.invoke(obj);
    }
}

以上Basic04ChangeMethodBody类创建了一个main方法，该方法中首先获取javassist的类池pool，然后调用pool.get(“domain.Hello”)方法获取到domain包下的Hello类。Hello类源码如下：

public class Hello {
    public void say() {
        System.out.println("Hello");
    }
}

该类中包含一个say方法，say方法调用println语句在屏幕上输出"Hello"。

回到Basic04ChangeMethodBody的main方法中，在获取到domain.Hello的ctClass对象后，获得了其声明的say方法的CtMethod对象。

然后通过调用method.setBody(“{System.out.println(“Javassist setBody!”);}”)方法更改say方法的方法体。其中setBody方法的参数为字符串类型的代码块。最后模拟调用修改后的say方法执行结果如下：

总结

本篇文章介绍了使用Javassist更改整个方法体，学习了Javassist的setBody语句的用法。通过调用CtMethod的setBody方法可以替换整个方法体。

说明