package com.zhi.demo;

import java.lang.reflect.Field;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtConstructor;
import javassist.CtField;
import javassist.CtMethod;
import javassist.CtNewConstructor;
import javassist.CtNewMethod;
import javassist.Loader;
import javassist.Modifier;
import javassist.bytecode.AccessFlag;

/**
 * Javassist動(dòng)態(tài)編程測(cè)試
 *
 * @date 2019年03月11日23:00:33
 *
 */
public class JavassistTest {
  public static void main(String[] args) {
    try {
      test();
    } catch (Exception e) {
      e.printStackTrace();
    }
  }

  private static void test() throws Exception {
    System.out.println("-------------------新增類------------------");
    ClassPool pool = ClassPool.getDefault();
    // 創(chuàng)建類
    CtClass ct = pool.makeClass("com.zhi.Person");
    // 讓類實(shí)現(xiàn)Cloneable接口
    ct.setInterfaces(new CtClass[] { pool.makeInterface("java.lang.Cloneable") });

    // 添加一個(gè)int類型的共有屬性
    CtField fieldId = new CtField(CtClass.intType, "id", ct);
    fieldId.setModifiers(AccessFlag.PUBLIC);
    ct.addField(fieldId);

    // 添加一個(gè)默認(rèn)構(gòu)造器
    CtConstructor constructor1 = CtNewConstructor.make("public Person(){this.id=1;}", ct);
    ct.addConstructor(constructor1);

    // 添加方法
    CtMethod helloM = CtNewMethod
        .make("public void hello(String des){System.out.println(\"執(zhí)行hello方法，\"+des+\"，我的id是\"+this.id);}", ct);
    ct.addMethod(helloM);

    // 將生成的.class文件保存到磁盤(pán)
    ct.writeFile();

    // 加載目標(biāo)類，可用ct.toClass()或new Loader(pool).loadClass()
    Class<?> clazz = ct.toClass();
//    Class<?> clazz = new Loader(pool).loadClass("com.zhi.Person");

    // 輸出類基本信息
    System.out.println("包名：" + clazz.getPackageName());
    System.out.println("類名：" + clazz.getName());
    System.out.println("簡(jiǎn)要類名：" + clazz.getSimpleName());
    System.out.println("限定符：" + Modifier.toString(clazz.getModifiers()));
    System.out.println("繼承類：" + clazz.getSuperclass().getName());
    Field[] fields = clazz.getDeclaredFields();
    for (Field field : fields) {
      System.out.println("屬性名稱：" + field.getName() + "，屬性類型：" + field.getType() + "，限定符："
          + Modifier.toString(field.getModifiers()));
    }

    // 構(gòu)造一個(gè)對(duì)象，并執(zhí)行hello方法
    Object ob = clazz.getDeclaredConstructor().newInstance();
    clazz.getMethod("hello", String.class).invoke(ob, "張三");

    // 解凍（執(zhí)行toClass后會(huì)自動(dòng)凍結(jié)）
    ct.defrost();

    System.out.println("-------------------修改類------------------");

    // 添加一個(gè)String類型的私有屬性
    CtField fieldName = new CtField(pool.get(String.class.getName()), "name", ct);
    fieldName.setModifiers(AccessFlag.PRIVATE);
    ct.addField(fieldName);

    // 添加帶參的構(gòu)造函數(shù)
    CtConstructor constructor2 = new CtConstructor(new CtClass[] { pool.get(String.class.getName()) }, ct);
    constructor2.setModifiers(Modifier.PUBLIC);
    constructor2.setBody("{this.name=$1;}");
    ct.addConstructor(constructor2);

    ct.addMethod(CtNewMethod.make("public void setName(String name){this.name=name;}", ct));
    ct.addMethod(CtNewMethod.make("public String getName(){return this.name;}", ct));

    ct.writeFile();

    // 加載類，若前面已用ct.toClass()進(jìn)行加載，則這里不能再用ct.toClass()加載，否則會(huì)出錯(cuò)，同一個(gè)加載不能2次加載同一個(gè)類
    clazz = new Loader(pool).loadClass("com.zhi.Person");

    fields = clazz.getDeclaredFields();
    for (Field field : fields) {
      System.out.println("屬性名稱：" + field.getName() + "，屬性類型：" + field.getType() + "，限定符："
          + Modifier.toString(field.getModifiers()));
    }

    ob = clazz.getDeclaredConstructor(String.class).newInstance("馬云");
    System.out.println("執(zhí)行g(shù)etName方法得到的值為：" + clazz.getMethod("getName").invoke(ob));
  }
}

-------------------新增類------------------
包名：com.zhi
類名：com.zhi.Person
簡(jiǎn)要類名：Person
限定符：public
繼承類：java.lang.Object
屬性名稱：id，屬性類型：int，限定符：public
執(zhí)行hello方法，張三，我的id是1
-------------------修改類------------------
屬性名稱：id，屬性類型：int，限定符：public
屬性名稱：name，屬性類型：class java.lang.String，限定符：private
執(zhí)行g(shù)etName方法得到的值為：馬云

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