java学习笔记----javaAsist入门

javaAsist入门

java动态性的常用实现方式有两种，反射和javaAsist，javaAsist性能要优于反射。可以直接编辑和生成Java生成的字节码，以达到对.class文件进行动态修改的效果。

maven依赖：

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

• 利用javaAsist去创建一个类

public class JavaAsistTest {
    public static void main(String[] args) throws CannotCompileException, NotFoundException, IOException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        ClassPool pool=ClassPool.getDefault();
        //创建类
        CtClass ctPerson=pool.makeClass("com.dl.Person");

        //创建属性
        CtField ctField1=CtField.make("private int age;", ctPerson);
        CtField ctField2=CtField.make("private String name;", ctPerson);

        //添加属性
        ctPerson.addField(ctField1);
        ctPerson.addField(ctField2);

        //添加方法
        ctPerson.addMethod(CtMethod.make("public String getName(){return name;}", ctPerson));
        ctPerson.addMethod(CtMethod.make("public void setName(String name){this.name=name;}", ctPerson));

        //添加构造器   进行参数传递，ctClass自带的只有8大基本类型，其他类型通过pool.get("java.lang.String")获得
        CtConstructor ctConstructor=new CtConstructor(new CtClass[]{CtClass.intType,pool.get("java.lang.String")},ctPerson);
        //$1代表第一个参数
        ctConstructor.setBody("{this.age=$1;this.name=$2;}");//设置方法体
        ctPerson.addConstructor(ctConstructor);
        //添加无参构造器
        CtConstructor ctConstructor1=new CtConstructor(null,ctPerson);
        ctConstructor1.setBody("{}");
        ctPerson.addConstructor(ctConstructor1);

        //写到本地
        //当CtClass 调用writeFile()、toClass()、toBytecode() 这些方法的时候，Javassist会冻结CtClass Object
        //调用ctPerson.defrost()进行解冻
        ctPerson.writeFile("D:/test");
        //byte[] b=cc.toBytecode(); 输出成二进制格式
    }
}

• 加载生成好的类，并动态添加方法，以及调用

public class JavaAsistTest2 {
    public static void main(String[] args) throws CannotCompileException, NotFoundException, IOException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        ClassPool pool=ClassPool.getDefault();
        pool.insertClassPath("D:/test");  //从file加载classpath

        //读取写好的类
        CtClass person=pool.get("com.dl.Person");
        System.out.println(person.getName());
        //再在这个类的基础上去添加一个方法,
        //CtMethod cm=CtMethod.make("public int add(int a,int b){return a+b;}", person);
        //同样可以采用其他方法进行创建=new CtMethod(返回值, 方法名, 参数类型，类)
        CtMethod cm=new CtMethod(CtClass.intType, "add", 
                new CtClass[]{CtClass.intType,CtClass.intType},person);
        cm.setModifiers(Modifier.PUBLIC);//设置访问权限
        //$1代表第一个参数
        cm.setBody("{return $1+$2;}");//设置方法体
        person.addMethod(cm);

        //通过反射去调用新生成的方法
        Class clazz= person.toClass();
        Object obj=clazz.newInstance();//调用无参构造器，创建Person对象
        //通过反射调用add方法，参数类型为（int，int）
        Method method=clazz.getDeclaredMethod("add",int.class,int.class);
        System.out.println(method.invoke(obj, 1,2));
    }
}

运行结果：
com.dl.Person
3

• 去操作方法内部，进行添加代码

public class JavaAsistTest3 {
    public static void main(String[] args) throws CannotCompileException, NotFoundException, IOException, NoSuchMethodException, SecurityException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException {
        ClassPool pool=ClassPool.getDefault();
        //读取类
        CtClass person=pool.get("com.dl.Person");;
        CtMethod cm=person.getDeclaredMethod("setName",new CtClass[]{pool.get("java.lang.String")});
        //在方法前加入代码
        cm.insertBefore("System.out.println( \"setName (name=\"+$1+\")   Start!\");");
        cm.insertAfter("System.out.println( \"setName end!\");");
        //在指定行数前加入代码
        cm.insertAt(24, "System.out.println( \"insert at 24\");");

        //通过反射去调用新生成的方法
        Class clazz= person.toClass();
        Object obj=clazz.newInstance();//调用无参构造器，创建Person对象
        Method method=clazz.getDeclaredMethod("setName",String.class);
        method.invoke(obj, "张三");
    }
}

运行结果：
setName (name=张三) Start!
insert at 24
setName end!