網(wǎng)絡(luò)請(qǐng)求框架設(shè)計(jì)
前言
我們產(chǎn)品線主要生產(chǎn)nas設(shè)備拾氓，nas設(shè)備有個(gè)核心功能是遠(yuǎn)程訪問势就。為了優(yōu)化用戶體驗(yàn)，前期嘗試了許多第三方連接方案系吭，導(dǎo)致網(wǎng)絡(luò)層架構(gòu)設(shè)計(jì)的十分復(fù)雜五嫂。其網(wǎng)絡(luò)請(qǐng)求流程圖如下：
我們可以看到，之前的方案邏輯很簡(jiǎn)單肯尺，都是簡(jiǎn)單的使用if判斷連接起來沃缘，但是代碼都是寫在一塊，看起來非吃蛞鳎混亂槐臀，是否配置都是采用block回調(diào)，block里面套用block氓仲，可讀性較差水慨，不方便維護(hù)得糜。并且后續(xù)如果決策要去除A、B連接方式晰洒，那代碼層面改動(dòng)非常大朝抖，還有可能會(huì)造成邏輯遺漏。綜合以上原因谍珊，我打算對(duì)連接模塊進(jìn)行一次重構(gòu)治宣。
重構(gòu)方案
前期連接方式多是因?yàn)楫a(chǎn)品希望通過嘗試不同廠商的方案尋找最適合的遠(yuǎn)程連接方式，目前是通過灰度開關(guān)進(jìn)行調(diào)節(jié)砌滞。當(dāng)后續(xù)功能穩(wěn)定后會(huì)把不適合的連接方式進(jìn)行去除侮邀。考慮到這層贝润，我打算采取比較靈活的編程方式AOP編程绊茧。經(jīng)過調(diào)研后，打算采用Aspects框架打掘。
思路：我們的應(yīng)用接口請(qǐng)求都是通過HTTP和HTTPS協(xié)議來傳輸按傅，主要方式是GET、POST和GET長(zhǎng)連接胧卤。目前的連接模塊已經(jīng)將這3個(gè)方法封裝，所有請(qǐng)求都會(huì)經(jīng)過這3個(gè)核心方法拼岳。如下圖：
連接方式A枝誊、B、C實(shí)際上只是把請(qǐng)求的URL和Port端口后替換成不同的服務(wù)器惜纸，從而達(dá)到走不同連接方式的目的叶撒，因此只要使用Aspects將網(wǎng)絡(luò)核心類中的方法替換，根據(jù)不同的連接情況將請(qǐng)求導(dǎo)到不同的連接模塊就能實(shí)現(xiàn)連接切換問題耐版。優(yōu)化后結(jié)構(gòu)如下：
這樣實(shí)現(xiàn)有如下優(yōu)點(diǎn)
- 沒有block"地獄嵌套",邏輯簡(jiǎn)單祠够，方法明了
- 輔助連接方式添加方便，可以快速刪除粪牲，不影響主連接方式的功能古瓤，整體實(shí)現(xiàn)更加靈活，滿足產(chǎn)品頻繁變更連接方式的業(yè)務(wù)需求
- 各個(gè)連接模塊之間無邏輯耦合
使用Aspects框架
Aspects是一個(gè)開源的的庫,面向切面編程,它能允許你在每一個(gè)類和每一個(gè)實(shí)例中存在的方法里面加入任何代碼腺阳÷渚可以在方法執(zhí)行之前或者之后執(zhí)行,也可以替換掉原有的方法。通過Runtime消息轉(zhuǎn)發(fā)實(shí)現(xiàn)Hook亭引。Aspects會(huì)自動(dòng)處理超類,比常規(guī)方法調(diào)用更容易使用,github上Star已經(jīng)超過6000多,目前已經(jīng)比較穩(wěn)定绎速。這里采用的就是它的替換方法，將原先的GET焙蚓、POST纹冤、active方法替換成AOP_GET洒宝、AOP_POST、AOP_active萌京。下面以hook GET請(qǐng)求為例：
首先利用Aspects的方法將GET方法替換成AOP_GET,將連接切換的代碼邏輯寫在AOP_GET方法中雁歌。Aspects hook后返回的是NSInvocation，這里需要利用反射的知識(shí)對(duì)其處理枫夺。為了完整將方法和參數(shù)"完全"拷貝到AOP_GET中将宪，我這里寫了一個(gè)通用的模板，如上圖所示橡庞。不過需要注意的是较坛，我這里只處理了參數(shù)為nil，NSNumber,BOOL扒最，long long丑勤，NSObject類型，因?yàn)槲覀僄ET方法的參數(shù)只有這些吧趣，如果要寫通用模板還需要補(bǔ)充其他參數(shù)類型的處理法竞。
經(jīng)過不斷踩坑，我這里總結(jié)下以下幾點(diǎn)：
[invocation setArgument:&result atIndex:i + 2];這里需要+2强挫，因?yàn)榈谝晃缓偷诙环謩e被self和方法簽名占用岔霸，參數(shù)是從第3位開始
參數(shù)為nil可以通過以下代碼傳參數(shù)
- 如果方法有返回值，比如我這是返回NSURLSessionDataTask俯渤，就需要先調(diào)用invoke 然后獲取return的值的地址呆细，再返給原先的originalInvocation
總結(jié)
從實(shí)戰(zhàn)中可以看出，通過Aspects進(jìn)行AOP變成可以很好的優(yōu)化代碼結(jié)構(gòu)八匠，對(duì)原代碼的侵入性低絮爷，是另一種比較巧妙的編程方式。
