網(wǎng)絡(luò)請(qǐng)求框架設(shè)計(jì)
前言
我們產(chǎn)品線(xiàn)主要生產(chǎn)nas設(shè)備，nas設(shè)備有個(gè)核心功能是遠(yuǎn)程訪(fǎng)問(wèn)。為了優(yōu)化用戶(hù)體驗(yàn)，前期嘗試了許多第三方連接方案，導(dǎo)致網(wǎng)絡(luò)層架構(gòu)設(shè)計(jì)的十分復(fù)雜。其網(wǎng)絡(luò)請(qǐng)求流程圖如下：
我們可以看到，之前的方案邏輯很簡(jiǎn)單，都是簡(jiǎn)單的使用if判斷連接起來(lái)，但是代碼都是寫(xiě)在一塊，看起來(lái)非常混亂，是否配置都是采用block回調(diào)，block里面套用block，可讀性較差，不方便維護(hù)。并且后續(xù)如果決策要去除A、B連接方式，那代碼層面改動(dòng)非常大，還有可能會(huì)造成邏輯遺漏。綜合以上原因，我打算對(duì)連接模塊進(jìn)行一次重構(gòu)。
重構(gòu)方案
前期連接方式多是因?yàn)楫a(chǎn)品希望通過(guò)嘗試不同廠(chǎng)商的方案尋找最適合的遠(yuǎn)程連接方式，目前是通過(guò)灰度開(kāi)關(guān)進(jìn)行調(diào)節(jié)。當(dāng)后續(xù)功能穩(wěn)定后會(huì)把不適合的連接方式進(jìn)行去除?？紤]到這層，我打算采取比較靈活的編程方式AOP編程。經(jīng)過(guò)調(diào)研后，打算采用Aspects框架。
思路：我們的應(yīng)用接口請(qǐng)求都是通過(guò)HTTP和HTTPS協(xié)議來(lái)傳輸，主要方式是GET、POST和GET長(zhǎng)連接。目前的連接模塊已經(jīng)將這3個(gè)方法封裝，所有請(qǐng)求都會(huì)經(jīng)過(guò)這3個(gè)核心方法。如下圖：
連接方式A、B、C實(shí)際上只是把請(qǐng)求的URL和Port端口后替換成不同的服務(wù)器，從而達(dá)到走不同連接方式的目的，因此只要使用Aspects將網(wǎng)絡(luò)核心類(lèi)中的方法替換，根據(jù)不同的連接情況將請(qǐng)求導(dǎo)到不同的連接模塊就能實(shí)現(xiàn)連接切換問(wèn)題。優(yōu)化后結(jié)構(gòu)如下：
這樣實(shí)現(xiàn)有如下優(yōu)點(diǎn)
- 沒(méi)有block"地獄嵌套",邏輯簡(jiǎn)單，方法明了
- 輔助連接方式添加方便，可以快速刪除，不影響主連接方式的功能，整體實(shí)現(xiàn)更加靈活，滿(mǎn)足產(chǎn)品頻繁變更連接方式的業(yè)務(wù)需求
- 各個(gè)連接模塊之間無(wú)邏輯耦合
使用Aspects框架
Aspects是一個(gè)開(kāi)源的的庫(kù),面向切面編程,它能允許你在每一個(gè)類(lèi)和每一個(gè)實(shí)例中存在的方法里面加入任何代碼?？梢栽诜椒▓?zhí)行之前或者之后執(zhí)行,也可以替換掉原有的方法。通過(guò)Runtime消息轉(zhuǎn)發(fā)實(shí)現(xiàn)Hook。Aspects會(huì)自動(dòng)處理超類(lèi),比常規(guī)方法調(diào)用更容易使用,github上Star已經(jīng)超過(guò)6000多,目前已經(jīng)比較穩(wěn)定。這里采用的就是它的替換方法，將原先的GET、POST、active方法替換成AOP_GET、AOP_POST、AOP_active。下面以hook GET請(qǐng)求為例：
首先利用Aspects的方法將GET方法替換成AOP_GET,將連接切換的代碼邏輯寫(xiě)在AOP_GET方法中。Aspects hook后返回的是NSInvocation，這里需要利用反射的知識(shí)對(duì)其處理。為了完整將方法和參數(shù)"完全"拷貝到AOP_GET中，我這里寫(xiě)了一個(gè)通用的模板，如上圖所示。不過(guò)需要注意的是，我這里只處理了參數(shù)為nil，NSNumber,BOOL，long long，NSObject類(lèi)型，因?yàn)槲覀僄ET方法的參數(shù)只有這些，如果要寫(xiě)通用模板還需要補(bǔ)充其他參數(shù)類(lèi)型的處理。
經(jīng)過(guò)不斷踩坑，我這里總結(jié)下以下幾點(diǎn)：
[invocation setArgument:&result atIndex:i + 2];這里需要+2，因?yàn)榈谝晃缓偷诙环謩e被self和方法簽名占用，參數(shù)是從第3位開(kāi)始
參數(shù)為nil可以通過(guò)以下代碼傳參數(shù)
- 如果方法有返回值，比如我這是返回NSURLSessionDataTask，就需要先調(diào)用invoke 然后獲取return的值的地址，再返給原先的originalInvocation
總結(jié)
從實(shí)戰(zhàn)中可以看出，通過(guò)Aspects進(jìn)行AOP變成可以很好的優(yōu)化代碼結(jié)構(gòu)，對(duì)原代碼的侵入性低，是另一種比較巧妙的編程方式。
