写在前面

源码 。
本文一起来看下，如何使用javassist来生成一个helloworld程序。

1：程序

package com.dahuyou.javassist.helloworld;

import javassist.*;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.lang.reflect.Method;

public class GenerateHelloWorld {

    public static void main(String[] args) throws Exception {
        ClassPool pool = ClassPool.getDefault();

        // 创建类
        CtClass ctClass = pool.makeClass("com.dahuyou.javassist.helloworld.Hello_byjavassist");

        // 创建main方法 设置返回类型，方法名称，参数等
        CtMethod mainMethod = new CtMethod(CtClass.voidType, "main", new CtClass[]{pool.get(String[].class.getName())}, ctClass);
        // 设置main方法方法修饰符
        mainMethod.setModifiers(Modifier.PUBLIC + Modifier.STATIC);
        // 设置方法体
        mainMethod.setBody("{System.out.println(\"javassist say hello by dahuyou\");}");
        // 方法添加到类
        ctClass.addMethod(mainMethod);

        // 创建无参数的构造函数
        CtConstructor ctConstructor = new CtConstructor(new CtClass[]{}, ctClass);
        ctConstructor.setBody("{}");
        // 构造函数添加到类
        ctClass.addConstructor(ctConstructor);

        // 写到文件
        ctClass.writeFile();

        // 反射运行测试
        Class aClass = ctClass.toClass();
        Object obj = aClass.newInstance();
        Method main = aClass.getDeclaredMethod("main", String[].class);
        main.invoke(obj, new String[1]);

//        outputClazz(ctClass.toBytecode(), aClass.getSimpleName());
    }

    private static void outputClazz(byte[] bytes, String className) {
        // 输出类字节码
        FileOutputStream out = null;
        try {
            String pathName = GenerateHelloWorld.class.getResource("/").getPath() + className + "_after_instrument.class";
            out = new FileOutputStream(new File(pathName));
            System.out.println("插桩后代码输出路径：" + pathName);
            out.write(bytes);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (null != out) try {
                out.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
    }

}

运行：

javassist say hello by dahuyou

Process finished with exit code 0

在项目路径看下生成的字节码：

写在后面

确实是asm版本的helloworld 简单多了😓😓😓。

参考文章列表