字节码编程，Javassist篇三《使用Javassist在运行时重新加载类「替换原方法输出不一样的结果」》

小傅哥

发布于 2020-07-14 15:51:54

1.3K0

发布于 2020-07-14 15:51:54

文章被收录于专栏：CodeGuide | 程序员编码指南

作者：小傅哥

热部署常用在生产环境中，主要由于这样的系统不能频繁启停且启动耗时较长的应用。
另外一些组件化风控模型包，给外部使用。当模型包进行升级时并不需要外部重新部署，甚至不需要让你知道升级了。
再者会用于开发、调试中，可以非常有效的提升编码效率，解放码农的「右手」和左手。

「人的大脑」很难创造未知的事物，所以需要学习。请多看小傅哥的码文，少搞CRUD

关于字节编程中所有涉及的代码，都可以通过关注公众号：「bugstack虫洞栈」，回复：源码，进行获取。

二、开发环境

JDK 1.8.0
jdk1.8.0_161\lib\tools.jar - 需要使用到 jdi 包
javassist 3.12.1.GA

三、案例目标

为了让案例目标更具色彩，我们模拟一个「谢飞机老婆」，通过系统查询自己男朋友前女友数量的 「危机」 方法，需要紧急处理的过程。

为了保障家庭的和谐化解危机，我们通过动态重新加载类，将谢飞机前女友数量修改为0并返回。依次安定家庭和谐。~~最终谢飞机会给我钱，当做报酬~~

德莱联盟，王牌工程师，申请出栈

「让谢飞机很慌的方法」

public class ApiTest {

    public String queryGirlfriendCount(String boyfriendName) {
        return boyfriendName + "的前女友数量：" + (new Random().nextInt(10) + 1) + " 个";
    }

}

「可预见的结果」；

你到底几个前女友！！！
谢飞机的前女友数量：3 个
谢飞机的前女友数量：5 个
谢飞机的前女友数量：8 个

四、技术实现

1. HotSwapper 操作类热加载


public class GenerateClazzMethod {

    public static void main(String[] args) throws Exception {

        ApiTest apiTest = new ApiTest();
        System.out.println("你到底几个前女友！！！");

        // 模拟谢飞机老婆一顿查询
        new Thread(() -> {
            while (true){
                System.out.println(apiTest.queryGirlfriendCount("谢飞机"));
                try {
                    Thread.sleep(100);
                } catch (InterruptedException e) {
                    e.printStackTrace();
                }
            }
        }).start();

        
        // 监听 8000 端口,在启动参数里设置
        // java -agentlib:jdwp=transport=dt_socket,server=y,suspend=n,address=8000
        HotSwapper hs = new HotSwapper(8000);

        ClassPool pool = ClassPool.getDefault();
        CtClass ctClass = pool.get(ApiTest.class.getName());

        // 获取方法
        CtMethod ctMethod = ctClass.getDeclaredMethod("queryGirlfriendCount");
        // 重写方法
        ctMethod.setBody("{ return $1 + \"的前女友数量：\" + (0L) + \" 个\"; }");

        // 加载新的类
        System.out.println(":: 执行HotSwapper热插拔，修改谢飞机前女友数量为0个！");
        hs.reload(ApiTest.class.getName(), ctClass.toBytecode());

    }

}

2. 知识点讲解

多线程模拟循环调用，这个方法会一直执行查询。在后续修改类之后输出的结果信息会有不同。
javassist.tools.HotSwapper，是 javassist 的包中提供的热加载替换类操作。在执行时需要启用 JPDA（Java平台调试器体系结构）。
ctMethod.setBody，重写方法的内容在上面两个章节已经很清楚的描述了。$1 是获取方法中的第一个入参，大括号{}里是具体执行替换的方法体。
最后使用 hs.reload 执行热加载替换操作，这里的 ctClass.toBytecode() 获取的是处理后类的字节码。

五、测试结果

1. 引入tools.jar

Open Module Settings，引入tools.jar

2. 配置-agentlib

-agentlib:jdwp=transport=dt_socket,server=y,suspend=n,address=8000

VM options，配置-agentlib

3. 执行测试

Listening for transport dt_socket at address: 8000
你到底几个前女友！！！
谢飞机的前女友数量：3 个
谢飞机的前女友数量：5 个
谢飞机的前女友数量：8 个
:: 执行HotSwapper热插拔，修改谢飞机前女友数量为0个！
谢飞机的前女友数量：4 个
谢飞机的前女友数量：5 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
谢飞机的前女友数量：0 个
...

Process finished with exit code -1

「当」看到前女友数量为 0 时，谢飞机露出了羞涩的微笑，并兑现了承诺，将4毛钱给了王牌工程师小傅哥。

来自谢飞机的4毛钱收入

4. 效果演示

热加载救火，成功拿到4毛钱

六、总结

关于热加载修改类的操作，在实际场景中还是蛮多的，但一般都是比较苛刻的场景诉求。在平时开发中还是比较少遇到的，并且CRUD开发不会遇到。
Javassist 对 ASM 这样的字节码操作封装起来提供的API确实很好操作，在一些场景下也不需要考虑 JVM 中局部变量和操作数栈。但如果需要更高的性能，可以考虑使用 ASM。

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2020-04-22，如有侵权请联系 cloudcommunity@tencent.com 删除

java

腾讯云测试服务

本文分享自 bugstack虫洞栈微信公众号，前往查看

如有侵权，请联系 cloudcommunity@tencent.com 删除。

本文参与腾讯云自媒体同步曝光计划，欢迎热爱写作的你一起参与！

java

腾讯云测试服务

登录后参与评论

0 条评论

热度