系列文章目录

01 在方法体的开头或结尾插入代码
02 使用Javassist实现方法执行时间统计

前言

上一章我们介绍了使用Javassist实现一个简单的Hello World程序，学会了Javassist创建类和新增方法。本章主要介绍如何使用Javassist实现一个无侵入的方法执行时间统计

引入Javassist jar包

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

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

使用Javassist实现方法执行时间统计

/**
 * 【Javassist】快速入门系列02 使用Javassist实现方法执行时间统计
 * 公众号&B站：精致的王同学
 * @author 精致的王同学
 * @date 2022/12/19 17:23
 */
public class Basic02ExecTime {
    public static void main(String[] args) throws Exception{
        // 获取javassist的类池
        ClassPool pool = ClassPool.getDefault();
        // 获取basic.StockService javassist的ctClass文件
        CtClass ctClass = pool.get("basic.StockService");
        // 获取声明的queryStockNum方法
        CtMethod method = ctClass.getDeclaredMethod("queryStockNum");
        // 为StockService类新增stime和etime字段
        CtField stime = new CtField(CtClass.longType, "stime", ctClass);
        CtField etime = new CtField(CtClass.longType, "etime", ctClass);
        ctClass.addField(stime);
        ctClass.addField(etime);
        // 在方法体之前插入代码
        method.insertBefore("stime = System.currentTimeMillis();");
        // 在方法体之后插入代码
        method.insertAfter("etime = System.currentTimeMillis();\n" +
                "System.out.println(\"执行时长：\"+(etime - stime)+\"毫秒\");");
        // 将类写成文件
        ctClass.writeFile();
        // 获取生成后的类文件
        Class<?> clazz = ctClass.toClass();
        // 创建新的实例
        Object obj = clazz.newInstance();
        // 获取生成后的queryStockNum方法
        Method queryStockNum = clazz.getDeclaredMethod("queryStockNum", Long.class);
        // 调用生成后的queryStockNum方法
        queryStockNum.invoke(obj,(Object) 1l);
    }
}

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

/**
 * 库存业务类
 * 公众号&B站：精致的王同学
 * @author 精致的王同学
 * @date 2022/12/19 17:23
 */
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;
    }
}