【Javassist】快速入门系列05 当有指定方法调用时替换方法调用的内容

系列文章目录

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

---

---

前言

上一章我们介绍了使用Javassist更改整个方法体，学会了Javassist的setBody()方法使用。本章主要介绍当检测到指定方法调用时替换方法调用的内容，method.instrument方法的使用。

---

引入Javassist jar包

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

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

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

/**
 * 【Javassist】快速入门系列05 当有指定方法调用时替换方法调用的内容
 * 公众号&B站：精致的王同学
 * @author 精致的王同学
 * @date 2022/12/21 16:25
 */
public class Basic05MethodCall {
    public static void main(String[] args) throws Exception{
        // 获取javassist默认类池
        ClassPool pool = ClassPool.getDefault();
        // 获取basic.Basic05Test的CtClass对象
        CtClass ctClass = pool.get("basic.Basic05Test");
        // 获取Basic05Test类的main方法
        CtMethod method = ctClass.getDeclaredMethod("main");
        // 更改main方法的方法体
        method.instrument(new ExprEditor(){
            @Override
            public void edit(MethodCall m) throws CannotCompileException {
                if (m.getClassName().equals("basic.StockService") && m.getMethodName().equals("queryStockNum")) {
                    m.replace("{System.out.println(\"MethodCall\");$_ = $proceed($$);}");
                }
            }
        });
        // 将类写成文件
        ctClass.writeFile();
        // 获取basic.Basic05Test的clazz对象
        Class<?> clazz = ctClass.toClass();
        // 获取修改后的basic.Basic05Test的实例
        Object obj = clazz.newInstance();
        // 获取修改后的main方法
        Method main = clazz.getDeclaredMethod("main",String[].class);
        // 模拟调用修改后的main方法
        main.invoke(obj,(Object) new String[0]);
    }
}

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

/**
 * 第5节测试类
 * 公众号&B站：精致的王同学
 * @author 精致的王同学
 * @date 2022/12/21 17:24
 */
public class Basic05Test {
    public static void main(String[] args) throws Exception{
        StockService stockService = new StockService();
        stockService.queryStockNum(1l);
    }
}

该类中调用了StockService 的stockService方法。StockService 源码如下：

/**
 * 库存业务类
 * 公众号&B站：精致的王同学
 * @author 精致的王同学
 * @date 2022/12/19 17:25
 */
public class StockService {
    public Integer queryStockNum(Long skuId) throws Exception {
        // 调用库存数量接口
        Random random = new Random();
        Long mills = Long.valueOf(random.nextInt(5)*1000);
        // 模拟接口调用耗时
        Thread.sleep(mills);
        return 0;
    }
}

StockService 的queryStockNum方法用于模拟调用获取sku库存数量接口的调用。

回到Basic05MethodCall 的main方法，在获取到Basic05Test类的ctClass的对象之后，获取其main方法的方法对象。然后调用method.instrument(ExprEditor editor)方法搜索类型为方法调用的语句。

instrument方法接收一个ExprEditor 类型的对象，该类有很多重载的edit方法，其中参数为MethodCall 的重载方法代表搜索method方法内的方法调用。然后判断如果是basic.StockService类的queryStockNum方法调用，则将此方法调用替换为指定代码块的内容。

ExprEditor 类重载的edit方法如下：

如果指定方法的调用有返回值的话，必须指定其内容。$_ 代表方法调用的返回值。

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

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

new ExprEditor() 代码块中的内容所做的事情为搜索main方法中所有的方法调用，当方法调用是basic.StockService类的queryStockNum方法时在屏幕上打印MethodCall。

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

总结

本篇文章介绍了使用Javassist当有指定方法调用时替换方法调用的内容，学习了 method.instrument的用法。以及参数为MethodCall 的重载方法的含义。

说明