使用基于jvm-sandbox的對三層嵌套類型的改造

問題背景

先簡單介紹下基于jvm-sandbox的imock工具，是Java方法級別的mock，操作就是監聽指定方法，返回指定的mock內容。

jvm-sandbox 利用字節碼操作和自定義類加載器的技術，將原始方法替換為模擬代碼，從而在應用程序中實現方法級別的模擬。這種方法非常強大，但也需要對字節碼操作、類加載機制和 JVM 內部原理有一定的理解。

公司要搭建一個方法級別的后端mock平臺，因此我在imock的基礎上進行二次開發進行使用。

問題描述

在mock某個三方接口的方法時遇到報錯：ava.lang.ClassCastException: com.alibaba.fastjson.JSONObject cannot be cast to com.travelsky.angeldoe.output.PassengerFlightInfo

看樣子是本來應該是JSONObject 無法轉化成PassengerFlightInfo類型，通過日志排查問題，定位到報錯代碼。

PassengerFlightInfo?passengerFlightInfo?=?JSON.parseObject(out
?.getPassengerFlightInfoList().get(0).toString(),?PassengerFlightInfo.class);

線上服務沒有報錯，測試mock環境報錯，那么顯然是數據的問題，通過Arthas追蹤方法返回的bean對比發現，差異就是線上的PassengerFlightInfo是一個bean，測試的PassengerFlightInfo是一個object。差異由此出現。

那么問題的關鍵就在于，如何通過mock工具把object提前轉成bean。

解決方案

改造mock agent工具思路：通過我們的mock-module.jar實現。

1. 根據PsrInfoOutputBean初步解析returnObject，獲取list中的object
2. 將object解析成PassengerFlightInfo，再通過反射技術將bean反射回PsrInfoOutputBean

代碼實現

//針對cki特殊類型PsrInfoOutputBean
case?3:
????//獲取advice返回類型的類加載器
????ClassLoader?behaviorClassLoader?=?advice.getBehavior().getReturnType().getClassLoader();
????//加載最外層PsrInfoOutputBean
????Class<?>?targetClass?=?behaviorClassLoader.loadClass(ro.getClassNames()[0]);
????LogUtil.info2("targetClass=",?targetClass.toString());
????//根據目標類解析returnData
????Object?res1?=?JSON.parseObject(ro.getReturnData(),?targetClass);
????//賦值保存做對比
????Object?res0?=?res1;
????LogUtil.info2("res1-before=",?res1.toString());
????//?通過反射獲取passengerFlightInfoList
????List<Object>?passengerFlightInfoList?=?(List<Object>)?targetClass.getMethod("getPassengerFlightInfoList").invoke(res1);
????LogUtil.info2("passengerFlightInfoList=",?passengerFlightInfoList.toString());
????if?(!passengerFlightInfoList.isEmpty())?{
????????//?獲取?passengerFlightInfoList?列表中的第一個元素
????????Object?firstPassengerFlightInfoList?=?passengerFlightInfoList.get(0);
????????LogUtil.info2("firstPassengerFlightInfoList=",?firstPassengerFlightInfoList.toString());
????????//?將?firstFlightInfo?轉換成?JSON?字符串
????????String?firstFlightInfoJson?=?JSON.toJSONString(firstPassengerFlightInfoList);
????????//?獲取第三層額外目標?Bean?類的類名，使用同一類加載器
????????Class<?>?targetBeanClass?=?behaviorClassLoader.loadClass(ro.getClassNames()[2]);
????????LogUtil.info2("targetBeanClass=",?targetBeanClass.toString());
????????//根據類解析成bean
????????Object?targetBean?=?JSON.parseObject(firstFlightInfoJson,?targetBeanClass);
????????LogUtil.info2("targetBean=",?targetBean.toString());
????????//?創建一個新的passengerFlightInfoListNew?將?targetBean?添加到?passengerFlightInfoList?中
????????List<Object>?passengerFlightInfoListNew?=?new?ArrayList<>();
????????passengerFlightInfoListNew.add(targetBean);
????????//?設置?passengerFlightInfoList?屬性回?res1
????????try?{
????????????//?執行反射方法,把passengerFlightInfoListNew反射回res
????????????Method?method?=?targetClass.getMethod("setPassengerFlightInfoList",?List.class);
????????????method.invoke(res1,?passengerFlightInfoListNew);
????????}?catch?(Exception?e)?{
????????????//?捕獲異常并打印日志
????????????LogUtil.info2("Error?occurred?while?invoking?method:=",?e.getMessage()+"｜"+e);
????????}
????}
????LogUtil.info2("前后的兩個類equals嗎？=",?String.valueOf(res1.equals(res0)));
????LogUtil.info2("res1-after=",?res1.toString());
????ProcessController.returnImmediately(res1);
????break;

遇到的坑

外部獲取的類名不能直接通過Class.forName加載，如下代碼所示：

?//?使用目標?Bean?類名解析?JSON?字符串成目標?Bean
????????Class<?>?targetBeanClass?=?Class.forName(targetBeanClassName);

實際會報錯："message": "com.taobao.rigel.rap.model.PsrInfoOutputBean cannot be cast to com.taobao.rigel.rap.model.PsrInfoOutputBean", 原因是這兩個bean雖然名字一樣，但是類加載器不同，就導致bean的實際是不一樣的。類是否相同可以用equals進行判斷。

因此正確的做法是，先獲取advice返回類型的類加載器，然后加載我們所需要的類，這樣業務的代碼就會認得我們的bean了。

???//獲取advice返回類型的類加載器
????ClassLoader?behaviorClassLoader?=?advice.getBehavior().getReturnType().getClassLoader();
????//加載最外層PsrInfoOutputBean
????Class<?>?targetClass?=?behaviorClassLoader.loadClass(ro.getClassNames()[0]);

題外話：

為啥出現了這個錯誤？

出現這個報錯和開發的強轉類型也有關系，本地做了個小測試，同樣的數據。（但咱也沒發改開發的代碼，只能提提建議。 = =）

1、當前異常轉化：按照開發業務代碼中的list強轉對象

List<Object> list = JSON.*parseArray*(jsonString); PassengerFlightInfo passengerFlightInfo = (PassengerFlightInfo) list.get(0);

這是使用強制類型轉換的方式，直接將 list 中的第一個元素強制轉換為 PassengerFlightInfo 對象。這種方式在編譯時不會報錯，但如果 list 中的第一個元素不是 PassengerFlightInfo 對象，則會在運行時拋出 ClassCastException 異常。

2、正常轉化：優化過后用toJavaObject方法

PassengerFlightInfo passengerFlightInfo = ((JSONObject) list.get(0)).toJavaObject(PassengerFlightInfo.class);： 這是使用 FastJSON 提供的 toJavaObject 方法，將 JSONObject 類型轉換為 PassengerFlightInfo 對象。這種方式在運行時會檢查轉換是否可行，如果 JSONObject 不包含 PassengerFlightInfo 的屬性或結構不匹配，會拋出異常。這種方式更安全，因為它提供了更多的轉換檢查。

推薦使用第二種方式，因為它更加健壯和安全，能夠更好地處理可能出現的異常情況，并提供更好的錯誤信息。