系列文章目录

01 在方法体的开头或结尾插入代码
02 使用Javassist实现方法执行时间统计
03 使用Javassist实现方法异常处理
04 使用Javassist更改整个方法体
05 当有指定方法调用时替换方法调用的内容
06 当有构造方法调用时替换方法调用的内容
07 当检测到字段被访问时使用语句块替换访问
08 当检测到对象(不包括数组)创建时用代码块替换
09 当检测到对象(不包括数组)创建时用代码块替换

---

---

前言

上一章我们介绍了当检测到对象(不包括数组)创建时用代码块替换，学习了 method.instrument的用法。以及参数为NewExpr 的重载方法edit的含义。本章主要介绍当检测到数组创建时用代码块替换。

---

引入Javassist jar包

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

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

当检测到数组创建时用代码块替换

/**
 * 【Javassist】快速入门系列09 当检测到数组创建时用代码块替换
 * 公众号&B站：精致的王同学
 * @author wangfengxi1
 * @date 2022/12/26 17:11
 */
public class Basic09NewArray {
    public static void main(String[] args) throws Exception{
        // 获取javassist默认类池
        ClassPool pool = ClassPool.getDefault();
        //获取basic.Basic08Test的ctClass对象
        CtClass ctClass = pool.get("basic.Basic09Test");
        //获取basic.Basic08Test的main方法
        CtMethod method = ctClass.getDeclaredMethod("main");
        // 当检测到数组创建时用代码块替换
        method.instrument(new ExprEditor(){
            @Override
            public void edit(NewArray a) throws CannotCompileException {
                try {
                    CtClass componentType = a.getComponentType();
                    if (componentType.equals(pool.get("java.lang.Integer"))) {
                        a.replace("System.out.println(\"NewArray\");$_=$proceed($$);");
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        // 将类写成文件
        ctClass.writeFile();
        // 获取修改后的class对象
        Class<?> clazz = ctClass.toClass();
        // 获取修改后的实例
        Object obj = clazz.newInstance();
        // 获取修改后的方法
        Method main = clazz.getDeclaredMethod("main", String[].class);
        // 调用修改后的方法
        main.invoke(obj,(Object) new String[0]);
    }
}ethod main = clazz.getDeclaredMethod("main", String[].class);
        // 调用修改后的方法
        main.invoke(obj,(Object) new String[0]);
    }
}

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

/**
 * 第9节测试类
 *
 * @author wangfengxi1
 * @date 2022/12/26 17:30
 */
public class Basic09Test {
    public static void main(String[] args) {
        Integer[] integers = new Integer[1];
    }
}

该类中有一个main方法创建了一个Integer数组。

回到Basic09NewArray的main方法，在获取到Basic09Test 类的ctClass的对象之后，获取其main方法的方法对象。

接着调用method.instrument(ExprEditor editor)方法搜索method内的数组创建。调用a.getComponentType()方法获取数组元素的类型，判断如果是java.lang.Integer，代表该类型的数组被创建。a.replace(“System.out.println(“NewArray”);${}_{=}$proceed($$);”);表示替换数组创建为代码块的语句。

instrument方法接收一个ExprEditor 类型的对象，该类有很多重载的edit方法，其中参数为NewArray的重载方法代表搜索方法内的数组创建
在替换语句内必须包含${}_{=}$proceed($$);

在replace的代码块中，以下符号有特殊含义：

$0  null。
$1，$2，  …每个维度的大小。
$_  数组创建的结果值。新创建的数组必须存储在此变量中。
$r  创建的数组的类型。
$type  表示创建数组的类的java.lang.Class对象。
$proceed  执行原始数组创建的虚拟方法的名称。

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

总结

本篇文章介绍了使用Javassist当检测到数组创建时用代码块替换，学习了 method.instrument的用法。以及参数为NewArray的重载方法edit的含义。

说明