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動態編程測試
 *
 * @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();
    // 創建類
    CtClass ct = pool.makeClass("com.zhi.Person");
    // 讓類實現Cloneable接口
    ct.setInterfaces(new CtClass[] { pool.makeInterface("java.lang.Cloneable") });

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

    // 添加一個默認構造器
    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(\"執行hello方法，\"+des+\"，我的id是\"+this.id);}", ct);
    ct.addMethod(helloM);

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

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

    // 輸出類基本信息
    System.out.println("包名：" + clazz.getPackageName());
    System.out.println("類名：" + clazz.getName());
    System.out.println("簡要類名：" + 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()));
    }

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

    // 解凍（執行toClass后會自動凍結）
    ct.defrost();

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

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

    // 添加帶參的構造函數
    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()進行加載，則這里不能再用ct.toClass()加載，否則會出錯，同一個加載不能2次加載同一個類
    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("執行getName方法得到的值為：" + clazz.getMethod("getName").invoke(ob));
  }
}

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

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