早早就說好要分享一下HandyJSON的實現(xiàn)思路了,今天總算有點時間配乱,這里就記錄一下吧。
背景
我所在iOS團隊是從去年11月份切Swift的。我們服務(wù)端和客戶端數(shù)據(jù)交互格式一直用的是JSON败京,而當時Swift中處理JSON名氣比較大的庫貌似只有SwiftyJSON,工程切到Swift后梦染,我們也用了這個庫赡麦。用上之后,需求是滿足了帕识,但是對一些復(fù)雜的Model泛粹,代碼寫得看起來非常糟糕,因為每次取值都需要 json["akey"]["bkey"]["ckey"].value
形式肮疗,寫的時候?qū)χ臋n沒覺得啥問題晶姊,但過后在脫離文檔的情況下,通篇都是字符串表達的key伪货,很難從代碼中感覺出Model結(jié)構(gòu)们衙。所以我們都會把一段sample數(shù)據(jù)寫在注釋里。但仍然比較凌亂碱呼,另外key寫錯了debug起來也費勁蒙挑,一個大小寫問題有時候debug半天。
于是我們進化了一下巍举,先寫好Model,然后Model類中寫convert函數(shù)凝垛,也用上了KVC遍歷key賦值懊悯。寫起來舒服多了,但還是麻煩梦皮,而且要求每一個類都繼承自NSObject炭分。不久后,我們認識了ObjectMapper庫剑肯,二話不說捧毛,就換了上去。世界頓時干凈多了让网。
但還是感覺差了一點呀忧,因為ObjectMapper需要自己指明映射關(guān)系。通常JSON中key和Model中字段名都是一致的溃睹,每次都要額外寫一坨東西而账,總覺得多余,字段有改動的時候也費勁因篇。新來剛接觸Swift的同事泞辐,也表示不太舒服笔横,因為他們之前使用的JSON反序列化庫,無論Java中還是Objective-C中咐吼,都是自然使用Model字段名去取值的吹缔。
所以就想著研究一下,Swift中能不能做到這種效果锯茄。
Swift中存在的限制
無論是Java或者Objective-C中的JSON反序列化庫厢塘,通常都是,在運行時獲取Model的字段名集合撇吞,遍歷該集合俗冻,拿Key去JSON中取值并完成賦值。這些步驟牍颈,Java依賴反射機制可以實現(xiàn)迄薄,Objective-C通過class_copyPropertyList
方法加上KVC機制,也能輕松實現(xiàn)煮岁。而Swift會卡在最后一步: 無法賦值讥蔽。
Swift的反射是只讀的,就是說画机,我們能在運行時獲取一個Model實例的所有字段冶伞、字段值,但卻無法給它賦值步氏。事實上响禽,我們拿到的value是原值的一個只讀拷貝,即使獲取到這個拷貝的地址寫入新值荚醒,也是無效的芋类。
class Animal {
var name: String?
}
Mirror(reflecting: Animal()).children.forEach { (child) in
print(child.label ?? "", child.value) // working correctly
child.value = "cat" // error,不能直接賦值
}
而且迄今界阁,蘋果官網(wǎng)文檔上對實現(xiàn)反射機制的Mirror類仍然是這么描述: Mirrors are used by playgrounds and the debugger
侯繁,態(tài)度非常含糊,似乎不太鼓勵泡躯,但生產(chǎn)中很多類庫都用上了贮竟。只能說,蘋果不會輕易撤下這個能力较剃,但期待它對這個能力做出改進(比如支持運行時賦值)咕别,是希望渺茫的。
如何繞過限制
最簡單的方式写穴,就是在Swift中定義Model時繼承NSObject顷级,讓這個Model的實例存在于objc運行時中,上述的class_copyPropertyList
方法和KVC就能用上了确垫。目前看見的Swift中不需要指明映射關(guān)系的JSON庫弓颈,都是這種方式帽芽。
然后就是以ObjectMapper
為代表的庫,通過運算符重載翔冀,在指定映射關(guān)系時完成賦值导街。走這一類實現(xiàn)的庫也非常多了。
但我想做到的是纤子,既支持運行在Swift運行時的純Swift類搬瑰,又不需要顯示指定每一個字段的映射關(guān)系。那么控硼,不能走反射賦值泽论,那就直接寫入內(nèi)存吧。
具體實現(xiàn)
Swift中卡乾,一個類實例的內(nèi)存布局是有規(guī)律的:
- 32位機器上翼悴,類前面有4+8個字節(jié)存儲meta信息,64位機器上幔妨,有8+8個字節(jié)鹦赎;
- 內(nèi)存中,字段從前往后有序排列误堡;
- 如果該類繼承自某一個類古话,那么父類的字段在前;
- Optional會增加一個字節(jié)來存儲.None/.Some信息锁施;
- 每個字段需要考慮內(nèi)存對齊陪踩;
這方面尚未從官方的資料找到參考,上述規(guī)律一些是從網(wǎng)上其他大神的總結(jié)中收集悉抵,一些從Clang的一些說明文檔中挖掘肩狂,加上自己的反復(fù)驗證得到。
有法子計算內(nèi)存布局基跑,剩下的事情就比較簡單了描焰。對一個實例:
- 獲取它的起始指針,移動到有效起點;
- 通過Mirror獲取每一個字段的字段名和字段類型吕喘;
- 根據(jù)字段名在JSON中取值,轉(zhuǎn)換為和字段一樣的類型拱礁,通過指針寫入;
- 根據(jù)本字段類型的占位大小和下一個字段類型計算下一個字段的對齊起點;
- 移動指針,繼續(xù)處理;
獲取類實例的起始指針
Swift中鹏浅,獲取struct實例起始指針和獲取class實例起始指針的方法是不一樣的,和語言版本也相關(guān)。在Swift3中:
// 獲取struct實例起始指針
mutating func headPointerOfStruct() -> UnsafeMutablePointer<Byte> {
return withUnsafeMutablePointer(to: &self) {
return UnsafeMutableRawPointer($0).bindMemory(to: Byte.self, capacity: MemoryLayout<Self>.stride)
}
}
// 獲取class實例起始指針
mutating func headPointerOfClass() -> UnsafeMutablePointer<Byte> {
let opaquePointer = Unmanaged.passUnretained(self as AnyObject).toOpaque()
let mutableTypedPointer = opaquePointer.bindMemory(to: Byte.self, capacity: MemoryLayout<Self>.stride)
return UnsafeMutablePointer<Byte>(mutableTypedPointer)
}
通過Mirror獲取字段名峰尝、類型
Mirror(reflecting: Animal()).children.forEach { (child) in
print(child.label ?? "") // 獲取字段名
print(type(of: child.value)) // 獲取字段類型
}
計算Model的每個屬性字段占位大小
Swift3暴露了兩個接口用于計算類型占位大新滓狻:MemoryLayout.size(ofValue: T)
和MemoryLayout<T>.size
括勺。這兩者都沒辦法直接用乱豆,因為:
- 對于每個屬性,我們目前只持有它的起始指針宛裕,而不是它的實例瑟啃,第一個接口用不上;
- 對于每個屬性揩尸,我們是在運行時中獲取到它的類型蛹屿,已經(jīng)沒辦法再實例化出泛型類型MemoryLayout<T>來計算size。所以岩榆,我引入了
HandyJSON
類错负,在擴展中實現(xiàn)函數(shù):
protocol HandyJSON {
}
extension HandyJSON {
static func size() -> Int {
return MemoryLayout<Self>.size
}
}
于是,對于每一個實現(xiàn)HandyJSON
協(xié)議的Model類T勇边,直接調(diào)用 T.size() 就能獲取到T的size了犹撒。
內(nèi)存對齊的影響
類實例的屬性并不是直接按照各自占位大小依次往下排列的,不然事情就簡單了粒褒。和C/C++一樣识颊,Swift中實例內(nèi)存布局也考慮了內(nèi)存對齊。翻閱了Swift的docs和LLVM的一些資料奕坟,MemoryLayout
提供了一個接口:MemoryLayout<T>.alignment
祥款,對齊的規(guī)則為,每個字段的起始地址必須為alignment值的整數(shù)倍执赡。細節(jié)的出處我一下子找不到了镰踏。那么函筋,HandyJSON
中計算下一個字段起始地址的函數(shù)為:
// Returns the offset to the next integer that is greater than
// or equal to Value and is a multiple of Align. Align must be
// non-zero.
static func offsetToAlignment(value: Int, align: Int) -> Int {
let m = value % align
return m == 0 ? 0 : (align - m)
}
其他情況
基本類型按照上述方法處理就可以了沙合,還有可選類型、數(shù)組類型跌帐、字典類型首懈,通過遍歷绊率、遞歸解析等方式,處理方法也類似究履。如數(shù)組:
extension Array: ArrayTypeProtocol {
static func getWrappedType() -> Any.Type {
return Element.self
}
static func castArrayType(arr: [Any]) -> Array<Element> {
return arr.map({ (p) -> Element in
return p as! Element
})
}
}
獲取到Array泛型實參類型滤否,然后構(gòu)造出該類型的一個數(shù)組,完成賦值就可以了最仑。
結(jié)語
主要流程就是這樣了藐俺,也比較簡單,剩下處理繼承泥彤、組合等情況欲芹,只是實現(xiàn)問題,就不再贅述了吟吝×飧福總覺得自己對Swift指針這一套設(shè)施理解還不是很到位,也許有更好的用法剑逃,比如說浙宜,完全不需要空的init()
函數(shù)就可以初始化出一個類的實例。有同學(xué)在這方面有更深入理解蛹磺,有什么意見或者建議的粟瞬,歡迎交流~
參考
- The Swift Reflection API and what you can do with it
- Getting to Know Enums, Structs and Classes in Swift
- Behind Swift Object
- Unsafe Swift: For Fun & Profit
- Becoming less afraid of unsafe mutable pointers in Swift
- Swift對C指針實現(xiàn)淺析
- Swift 中的指針使用
- Advanced & Practical Enum usage in Swift
- SwiftJSON
- ObjectMapper