06_javassist

【简述】

Javassist是一个开源的java字节码操作工具，主要是对已经编译好class文件进行修改和处理，可以直接检查、修改、创建 java类。

【javassist实例】

package com.javassist;

import javassist.*;

import java.io.File;
import java.io.FileOutputStream;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;

/**
 * Created by cuixinjie on 2018/9/7.
 */
public class JavassistTest {

    public static void main(String[] args) throws Exception {
        //ClassPool：ctClass对象的容器
        ClassPool pool = ClassPool.getDefault();
        //通过ClassPool深沉给一个public的新类 User.java
        CtClass ctClass = pool.makeClass("com.javassist.User");

        //添加属性，首先添加age属性：private Integer age;
        CtField ageField = new CtField(pool.getCtClass("java.lang.Integer"), "age", ctClass);
        ageField.setModifiers(Modifier.PRIVATE);
        ctClass.addField(ageField);

        //添加属性，其次添加phone属性：private String phone;
        CtField phoneField = new CtField(pool.getCtClass("java.lang.String"), "phone", ctClass);
        phoneField.setModifiers(Modifier.PRIVATE);
        ctClass.addField(phoneField);

        //为age和phone添加getXxx/setXxx方法
        ctClass.addMethod(CtNewMethod.getter("getAge", ageField));
        ctClass.addMethod(CtNewMethod.setter("setAge", ageField));
        ctClass.addMethod(CtNewMethod.getter("getPhone", phoneField));
        ctClass.addMethod(CtNewMethod.setter("setPhone", phoneField));

        //添加构造方法
        CtConstructor ctConstructor = new CtConstructor(new CtClass[]{}, ctClass);
        //为构造方法添加方法体
        StringBuffer buffer1 = new StringBuffer();
        //注意：this.age = ($w)18这里加上$w,因为age是Integer类型，是一个包装类，$w代表一个包装类型
        buffer1.append("{\nthis.age = ($w)18;\nthis.phone = \"15966668888\";\n}");
        ctConstructor.setBody(buffer1.toString());
        //把构造方法添加到新的类ctClass中
        ctClass.addConstructor(ctConstructor);

        //添加自定义方法
        CtMethod ctMethod = new CtMethod(CtClass.voidType, "printInfo", new CtClass[]{}, ctClass);
        //为自定义的方法设置方法体
        StringBuffer buffer2 = new StringBuffer();
        buffer2.append("{\nSystem.out.println(\"I can get Alibaba's offer!\");\n");
        buffer2.append("System.out.println(\"my age is \"+age);\n");
        buffer2.append("System.out.println(\"my phone is \"+phone);\n");
        buffer2.append("}");
        ctMethod.setBody(buffer2.toString());
        ctClass.addMethod(ctMethod);

        //生成一个Class对象
        Class<?> clazz = ctClass.toClass();
        Object instance = clazz.newInstance();
        //反射 执行方法
        Method method = instance.getClass().getMethod("printInfo", new Class[]{});
        method.invoke(instance,null);

        // 把生成的class文件写入文件
        byte[] byteArr = ctClass.toBytecode();
        FileOutputStream fos = new FileOutputStream(new File("./User.class"));
        fos.write(byteArr);
        fos.close();
    }
}

【运行结果】