【Javassist】快速入门系列10 当检测到instanceof表达式时用代码块替换

系列文章目录

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

---

---

前言

上一章我们介绍了当检测到数组创建时用代码块替换，学习了 method.instrument的用法。以及参数为NewArray 的重载方法edit的含义。本章主要介绍当检测到instanceof表达式时用代码块替换。

---

引入Javassist jar包

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

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

当检测到instanceof表达式时用代码块替换

/**
 * 【Javassist】快速入门系列10 当检测到instanceof表达式时用代码块替换
 * 公众号&B站：精致的王同学
 * @author wangfengxi1
 * @date 2022/12/27 17:29
 */
public class Basic10Instanceof {
    public static void main(String[] args) throws Exception{
        //获取javassist默认类池
        ClassPool pool = ClassPool.getDefault();
        // 获取basic.Basic10Test类的ctClass文件
        CtClass ctClass = pool.get("basic.Basic10Test");
        // 获取basic.Basic10Test类的main方法
        CtMethod method = ctClass.getDeclaredMethod("main");
        // 当检测到instanceof表达式时用代码块替换
        method.instrument(new ExprEditor(){
            @Override
            public void edit(Instanceof i) throws CannotCompileException {
                try {
                    CtClass type = i.getType();
                    if (type.equals(pool.get("domain.Animal"))) {
                        i.replace("$_=$proceed($$);System.out.println(\"Instanceof,result:\"+$_+\";value:\"+$1);");
                    }
                } catch (Exception e) {
                    e.printStackTrace();
                }
            }
        });
        // 获取修改后的class文件
        Class<?> clazz = ctClass.toClass();
        // 获取修改后的实例对象
        Object obj = clazz.newInstance();
        // 获取修改后的main方法
        Method main = clazz.getDeclaredMethod("main", String[].class);
        // 调用修改后的main方法
        main.invoke(obj,(Object) new String[0]);
    }
}

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

/**
 * 第10节测试类
 *
 * @author wangfengxi1
 * @date 2022/12/27 17:31
 */
public class Basic10Test {
    public static void main(String[] args) {
        People people = new People(1,"人");
        if (people instanceof Animal) {
            System.out.println(people.toString());
        }
    }
}

该类中有一个main方法，创建了一个People对象。然后判断people是否是Animal类或者其子类。如果是则在屏幕打印people对象。其中People类的源码如下：

/**
 * 人
 *
 * @author wangfengxi1
 * @date 2022/12/27 17:35
 */
public class People extends Animal{
    public People(Integer type, String typename) {
        super(type, typename);
    }
}

People继承了Animal类，其中Animal类的源码如下：

/**
 * 动物父类
 *
 * @author wangfengxi1
 * @date 2022/12/27 17:34
 */
public class Animal {
    private Integer type;
    private String typename;

    public Animal(Integer type, String typename) {
        this.type = type;
        this.typename = typename;
    }

    public Integer getType() {
        return type;
    }

    public void setType(Integer type) {
        this.type = type;
    }

    public String getTypename() {
        return typename;
    }

    public void setTypename(String typename) {
        this.typename = typename;
    }

    @Override
    public String toString() {
        return "Animal{" +
                "type=" + type +
                ", typename='" + typename + '\'' +
                '}';
    }
}

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

接着调用method.instrument(ExprEditor editor)方法搜索method内的instanceof表达式。调用i.getType()方法获取instanceof表达式类型，判断如果是domain.Animal的ctClass文件则调用i.replace(“${}_{=}$proceed($$);System.out.println(\ddot{I}nstanceof,result:\ddot{+}$_+”;value:“+$1);”);表示替换instanceof表达式。

instrument方法接收一个ExprEditor 类型的对象，该类有很多重载的edit方法，其中参数为Instanceof 的重载方法代表搜索方法内的instanceof表达式。

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

$0  null。
$1  原始instanceof运算符左侧的值。
$_  表达式的结果值。$_的类型是布尔型。
$r  instanceof运算符右侧的类型。
$type  表示instanceof运算符右侧类型的java.lang.Class对象。
$proceed  执行原始instanceof表达式的虚拟方法的名称。它接受一个参数（类型为java.lang.Object）并且如果参数值是instanceof操作符右侧类型的实例的话返回true。否则，返回false。

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

总结

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

说明