關于javassist使用方法

發(fā)布時間：2020-07-01 16:22:03 來源：億速云閱讀：158 作者：清晨欄目：開發(fā)技術

不懂關于javassist使用方法？其實想解決這個問題也不難，下面讓小編帶著大家一起學習怎么去解決，希望大家閱讀完這篇文章后大所收獲。

Java 字節(jié)碼以二進制的形式存儲在 .class 文件中，每一個 .class 文件包含一個 Java 類或接口。Javaassist 就是一個用來處理 Java 字節(jié)碼的類庫。它可以在一個已經編譯好的類中添加新的方法，或者是修改已有的方法，并且不需要對字節(jié)碼方面有深入的了解。同時也可以去生成一個新的類對象，通過完全手動的方式。

1. 使用 Javassist 創(chuàng)建一個 class 文件

首先需要引入jar包：

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

編寫創(chuàng)建對象的類：

package com.rickiyang.learn.javassist;

import javassist.*;

/**
 * @author rickiyang
 * @date 2019-08-06
 * @Desc
 */
public class CreatePerson {

 /**
  * 創(chuàng)建一個Person 對象
  *
  * @throws Exception
  */
 public static void createPseson() throws Exception {
  ClassPool pool = ClassPool.getDefault();

  // 1. 創(chuàng)建一個空類
  CtClass cc = pool.makeClass("com.rickiyang.learn.javassist.Person");

  // 2. 新增一個字段 private String name;
  // 字段名為name
  CtField param = new CtField(pool.get("java.lang.String"), "name", cc);
  // 訪問級別是 private
  param.setModifiers(Modifier.PRIVATE);
  // 初始值是 "xiaoming"
  cc.addField(param, CtField.Initializer.constant("xiaoming"));

  // 3. 生成 getter、setter 方法
  cc.addMethod(CtNewMethod.setter("setName", param));
  cc.addMethod(CtNewMethod.getter("getName", param));

  // 4. 添加無參的構造函數(shù)
  CtConstructor cons = new CtConstructor(new CtClass[]{}, cc);
  cons.setBody("{name = \"xiaohong\";}");
  cc.addConstructor(cons);

  // 5. 添加有參的構造函數(shù)
  cons = new CtConstructor(new CtClass[]{pool.get("java.lang.String")}, cc);
  // $0=this / $1,$2,$3... 代表方法參數(shù)
  cons.setBody("{$0.name = $1;}");
  cc.addConstructor(cons);

  // 6. 創(chuàng)建一個名為printName方法，無參數(shù)，無返回值，輸出name值
  CtMethod ctMethod = new CtMethod(CtClass.voidType, "printName", new CtClass[]{}, cc);
  ctMethod.setModifiers(Modifier.PUBLIC);
  ctMethod.setBody("{System.out.println(name);}");
  cc.addMethod(ctMethod);

  //這里會將這個創(chuàng)建的類對象編譯為.class文件
  cc.writeFile("/Users/yangyue/workspace/springboot-learn/java-agent/src/main/java/");
 }

 public static void main(String[] args) {
  try {
   createPseson();
  } catch (Exception e) {
   e.printStackTrace();
  }
 }
}

執(zhí)行上面的 main 函數(shù)之后，會在指定的目錄內生成 Person.class 文件：

//
// Source code recreated from a .class file by IntelliJ IDEA
// (powered by Fernflower decompiler)
//

package com.rickiyang.learn.javassist;

public class Person {
 private String name = "xiaoming";

 public void setName(String var1) {
  this.name = var1;
 }

 public String getName() {
  return this.name;
 }

 public Person() {
  this.name = "xiaohong";
 }

 public Person(String var1) {
  this.name = var1;
 }

 public void printName() {
  System.out.println(this.name);
 }
}

跟咱們預想的一樣。

在 Javassist 中，類 Javaassit.CtClass 表示 class 文件。一個 GtClass (編譯時類）對象可以處理一個 class 文件，ClassPool是 CtClass 對象的容器。它按需讀取類文件來構造 CtClass 對象，并且保存 CtClass 對象以便以后使用。

需要注意的是 ClassPool 會在內存中維護所有被它創(chuàng)建過的 CtClass，當 CtClass 數(shù)量過多時，會占用大量的內存，API中給出的解決方案是有意識的調用CtClass的detach()方法以釋放內存。

ClassPool需要關注的方法：

getDefault : 返回默認的ClassPool 是單例模式的，一般通過該方法創(chuàng)建我們的ClassPool；
appendClassPath, insertClassPath : 將一個ClassPath加到類搜索路徑的末尾位置或插入到起始位置。通常通過該方法寫入額外的類搜索路徑，以解決多個類加載器環(huán)境中找不到類的尷尬；
toClass : 將修改后的CtClass加載至當前線程的上下文類加載器中，CtClass的toClass方法是通過調用本方法實現(xiàn)。需要注意的是一旦調用該方法，則無法繼續(xù)修改已經被加載的class；
get , getCtClass : 根據(jù)類路徑名獲取該類的CtClass對象，用于后續(xù)的編輯。

CtClass需要關注的方法：

freeze : 凍結一個類，使其不可修改；
isFrozen : 判斷一個類是否已被凍結；
prune : 刪除類不必要的屬性，以減少內存占用。調用該方法后，許多方法無法將無法正常使用，慎用；
defrost : 解凍一個類，使其可以被修改。如果事先知道一個類會被defrost，則禁止調用 prune 方法；
detach : 將該class從ClassPool中刪除；
writeFile : 根據(jù)CtClass生成 .class 文件；
toClass : 通過類加載器加載該CtClass。

上面我們創(chuàng)建一個新的方法使用了CtMethod類。CtMthod代表類中的某個方法，可以通過CtClass提供的API獲取或者CtNewMethod新建，通過CtMethod對象可以實現(xiàn)對方法的修改。

CtMethod中的一些重要方法：

insertBefore : 在方法的起始位置插入代碼；
insterAfter : 在方法的所有 return 語句前插入代碼以確保語句能夠被執(zhí)行，除非遇到exception；
insertAt : 在指定的位置插入代碼；
setBody : 將方法的內容設置為要寫入的代碼，當方法被 abstract修飾時，該修飾符被移除；
make : 創(chuàng)建一個新的方法。

注意到在上面代碼中的：setBody()的時候我們使用了一些符號：

// $0=this / $1,$2,$3... 代表方法參數(shù)
cons.setBody("{$0.name = $1;}");

具體還有很多的符號可以使用，但是不同符號在不同的場景下會有不同的含義，所以在這里就不在贅述，可以看javassist 的說明文檔。http://www.javassist.org/tutorial/tutorial2.html

2. 調用生成的類對象

（1）. 通過反射的方式調用

上面的案例是創(chuàng)建一個類對象然后輸出該對象編譯完之后的 .class 文件。那如果我們想調用生成的類對象中的屬性或者方法應該怎么去做呢？javassist也提供了相應的api，生成類對象的代碼還是和第一段一樣，將最后寫入文件的代碼替換為如下：

// 這里不寫入文件，直接實例化
Object person = cc.toClass().newInstance();
// 設置值
Method setName = person.getClass().getMethod("setName", String.class);
setName.invoke(person, "cunhua");
// 輸出值
Method execute = person.getClass().getMethod("printName");
execute.invoke(person);

然后執(zhí)行main方法就可以看到調用了 printName方法。

（2）. 通過讀取 .class 文件的方式調用

ClassPool pool = ClassPool.getDefault();
// 設置類路徑
pool.appendClassPath("/Users/yangyue/workspace/springboot-learn/java-agent/src/main/java/");
CtClass ctClass = pool.get("com.rickiyang.learn.javassist.Person");
Object person = ctClass.toClass().newInstance();
// ...... 下面和通過反射的方式一樣去使用

（3）. 通過接口的方式

上面兩種其實都是通過反射的方式去調用，問題在于我們的工程中其實并沒有這個類對象，所以反射的方式比較麻煩，并且開銷也很大。那么如果你的類對象可以抽象為一些方法得合集，就可以考慮為該類生成一個接口類。這樣在newInstance()的時候我們就可以強轉為接口，可以將反射的那一套省略掉了。

還拿上面的Person類來說，新建一個PersonI接口類：

package com.rickiyang.learn.javassist;

/**
 * @author rickiyang
 * @date 2019-08-07
 * @Desc
 */
public interface PersonI {

 void setName(String name);

 String getName();

 void printName();

}

實現(xiàn)部分的代碼如下：

ClassPool pool = ClassPool.getDefault();
pool.appendClassPath("/Users/yangyue/workspace/springboot-learn/java-agent/src/main/java/");

// 獲取接口
CtClass codeClassI = pool.get("com.rickiyang.learn.javassist.PersonI");
// 獲取上面生成的類
CtClass ctClass = pool.get("com.rickiyang.learn.javassist.Person");
// 使代碼生成的類，實現(xiàn) PersonI 接口
ctClass.setInterfaces(new CtClass[]{codeClassI});

// 以下通過接口直接調用 強轉
PersonI person = (PersonI)ctClass.toClass().newInstance();
System.out.println(person.getName());
person.setName("xiaolv");
person.printName();

使用起來很輕松。

3. 修改現(xiàn)有的類對象

前面說到新增一個類對象。這個使用場景目前還沒有遇到過，一般會遇到的使用場景應該是修改已有的類。比如常見的日志切面，權限切面。我們利用javassist來實現(xiàn)這個功能。

有如下類對象：

package com.rickiyang.learn.javassist;

/**
 * @author rickiyang
 * @date 2019-08-07
 * @Desc
 */
public class PersonService {

 public void getPerson(){
  System.out.println("get Person");
 }

 public void personFly(){
  System.out.println("oh my god,I can fly");
 }
}

然后對他進行修改：

package com.rickiyang.learn.javassist;

import javassist.ClassPool;
import javassist.CtClass;
import javassist.CtMethod;
import javassist.Modifier;

import java.lang.reflect.Method;

/**
 * @author rickiyang
 * @date 2019-08-07
 * @Desc
 */
public class UpdatePerson {

 public static void update() throws Exception {
  ClassPool pool = ClassPool.getDefault();
  CtClass cc = pool.get("com.rickiyang.learn.javassist.PersonService");

  CtMethod personFly = cc.getDeclaredMethod("personFly");
  personFly.insertBefore("System.out.println(\"起飛之前準備降落傘\");");
  personFly.insertAfter("System.out.println(\"成功落地。。。。\");");


  //新增一個方法
  CtMethod ctMethod = new CtMethod(CtClass.voidType, "joinFriend", new CtClass[]{}, cc);
  ctMethod.setModifiers(Modifier.PUBLIC);
  ctMethod.setBody("{System.out.println(\"i want to be your friend\");}");
  cc.addMethod(ctMethod);

  Object person = cc.toClass().newInstance();
  // 調用 personFly 方法
  Method personFlyMethod = person.getClass().getMethod("personFly");
  personFlyMethod.invoke(person);
  //調用 joinFriend 方法
  Method execute = person.getClass().getMethod("joinFriend");
  execute.invoke(person);
 }

 public static void main(String[] args) {
  try {
   update();
  } catch (Exception e) {
   e.printStackTrace();
  }
 }
}

在personFly方法前后加上了打印日志。然后新增了一個方法joinFriend。執(zhí)行main函數(shù)可以發(fā)現(xiàn)已經添加上了。

另外需要注意的是：上面的insertBefore() 和 setBody()中的語句，如果你是單行語句可以直接用雙引號，但是有多行語句的情況下，你需要將多行語句用{}括起來。javassist只接受單個語句或用大括號括起來的語句塊。

感謝你能夠認真閱讀完這篇文章，希望小編分享關于javassist使用方法內容對大家有幫助，同時也希望大家多多支持億速云，關注億速云行業(yè)資訊頻道，遇到問題就找億速云，詳細的解決方法等著你來學習!

向AI問一下細節(jié)

關于javassist使用方法

猜你喜歡

最新資訊

相關推薦

相關標簽