【Javassist】快速入门系列06 当检测到构造方法调用时替换方法调用的内容

系列文章目录

01 在方法体的开头或结尾插入代码
02 使用Javassist实现方法执行时间统计
03 使用Javassist实现方法异常处理
04 使用Javassist更改整个方法体
05 当有指定方法调用时替换方法调用的内容
06 当有构造方法调用时替换方法调用的内容

---

---

前言

上一章我们介绍了当有指定方法调用时替换方法调用的内容，学习了 method.instrument的用法。以及参数为MethodCall 的重载方法的含义。本章主要介绍当检测到构造方法调用时替换方法调用的内容。

---

引入Javassist jar包

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

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

当有指定方法调用时替换方法调用的内容

/**
 * 【Javassist】快速入门系列06 当构造方法调用时替换方法调用的内容
 * 公众号&B站：精致的王同学
 * @author wangfengxi1
 * @date 2022/12/24 21:59
 */
public class Basic06ConstructorCall {
    public static void main(String[] args) throws Exception{
        // 获取javassist默认类池
        ClassPool pool = ClassPool.getDefault();
        // 获取basic.Basic06Test类的ctClass对象
        CtClass ctClass = pool.get("basic.Basic06Test");
        // 获取basic.Basic06Test类的main方法
        CtMethod method = ctClass.getDeclaredMethod("main");
        // 获取basic.Basic06Test类参数为String类型的构造方法
        CtConstructor ctConstructor = ctClass.getDeclaredConstructor(new CtClass[]{pool.get("java.lang.String")});
        // 更改main方法的方法体 当有basic.Basic06Test的构造方法被调用时this() 或 super()
        ctConstructor.instrument(new ExprEditor(){
            @Override
            public void edit(ConstructorCall c) throws CannotCompileException {
                if (c.getClassName().equals("basic.Basic06Test")) {
                    c.replace("System.out.println(\"ConstructorCall\");$proceed($$);");
                }
            }
        });
        // 将类写成文件
        ctClass.writeFile();
        //获取basic.Basic06Test类修改后的class对象
        Class<?> clazz = ctClass.toClass();
        // 获取basic.Basic06Test类修改后的实例
        Object obj = clazz.newInstance();
        // 获取basic.Basic06Test类修改后的main方法
        Method main = clazz.getDeclaredMethod("main", String[].class);
        // 模拟调用修改后的main方法
        main.invoke(obj,(Object) new String[0]);
    }
}

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

/**
 * 第6节测试类
 *
 * @author wangfengxi1
 * @date 2022/12/24 22:00
 */
public class Basic06Test {
    private String name;

    public Basic06Test() {
        System.out.println("无参构造方法调用");
    }

    public Basic06Test(String name) {
        // 调用this方法被javassist监听
        this();
        System.out.println("有参构造方法调用");
        this.name = name;
    }

    public static void main(String[] args) {
        Basic06Test basic06Test = new Basic06Test("小明");
    }
}

该类中有一个main方法调用了该类的参数为name的构造方法，该构造方法中调用了this()无参构造方法。

回到Basic06ConstructorCall 的main方法，在获取到Basic06Test 类的ctClass的对象之后，获取其main方法的方法对象。然后调用ctClass.getDeclaredConstructor()方法获取其参数为String类型的构造器ctConstructor 。

接着调用ctConstructor.instrument(ExprEditor editor)方法搜索ctConstructor构造器内对其他构造方法的调用。判断如果其类是basic.Basic06Test则调用c.replace(“System.out.println(“ConstructorCall”);$proceed( $$);”);方法替换构造方法调用。

instrument方法接收一个ExprEditor 类型的对象，该类有很多重载的edit方法，其中参数为ConstructorCall的重载方法代表搜索构造方法内的其他构造方法调用。

由于任何构造函数都必须调用super类的构造函数或同类的另一个构造函数，因此替换语句必须包含构造函数调用，通常是对$proceed方法的调用。

$proceed 代表调用指定的方法，本例中为构造方法。

$$ 代表方法的全部参数。

最后模拟调用修改后的main方法结果如下：

其中Object obj = clazz.newInstance();调用了一次无参构造方法，$proceed($$)方法替换了有参构造方法的调用。所以屏幕上打印了两次无参方法调用。

总结

本篇文章介绍了使用Javassist当检测到构造方法调用时替换方法调用的内容，学习了 method.instrument的用法。以及参数为ConstructorCall的重载方法的含义。

说明