系列文章目录

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

文章目录

系列文章目录
前言
引入Javassist jar包
- 当检测到对象(不包括数组)创建时用代码块替换
总结
说明

前言

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

引入Javassist jar包

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

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

当检测到对象(不包括数组)创建时用代码块替换

/**
 * 【Javassist】快速入门系列08 当检测到对象(不包括数组)创建时用代码块替换
 * 公众号&B站：精致的王同学
 * @author wangfengxi1
 * @date 2022/12/25 18:02
 */
public class Basic08NewExpr {
    public static void main(String[] args) throws Exception{
        // 获取javassist默认类池
        ClassPool pool = ClassPool.getDefault();
        //获取basic.Basic08Test的ctClass对象
        CtClass ctClass = pool.get("basic.Basic08Test");
        //获取basic.Basic08Test的main方法
        CtMethod method = ctClass.getDeclaredMethod("main");
        // 当检测到对象创建时用代码块替换
        method.instrument(new ExprEditor(){
            @Override
            public void edit(NewExpr e) throws CannotCompileException {
                if (e.getClassName().equals("java.lang.Integer")) {
                    e.replace("System.out.println(\"NewExpr\");$_=$proceed($$);");
                }
            }
        });
        // 将类写成文件
        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]);
    }
}

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

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

该类中有一个main方法创建了一个Integer 对象

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

接着调用method.instrument(ExprEditor editor)方法搜索method内的对象创建。判断如果其类名是java.lang.Integer，代表该类型的对象被创建。e.replace(“System.out.println(“NewExpr”); $_=$ proceed($$);”);表示替换对象创建为代码块的语句。

instrument方法接收一个ExprEditor 类型的对象，该类有很多重载的edit方法，其中参数为NewExpr 的重载方法代表搜索方法内的对象创建（不包含数组创建）。

在替换语句内必须包含 $_=$ proceed($$);

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

$0 null。
$1，$2，  …构造函数的参数。
$_  对象创建的结果值。新创建的对象必须存储在此变量中。
$r  创建对象的类型。
$sig  表示形式参数类型的java.lang.Class对象数组。
$type  表示创建对象的类的java.lang.Class对象。
$proceed  执行原始对象创建的虚拟方法的名称。