項目總結四：網絡請求中，NSURLProtocol添加請求頭的問題


項目背景：

最近在做一個項目，里面的網絡請求分為兩部分，一部分是根據第三方AFNetworking封裝出來的，另一個是webview自己請求顯示的。需求是返回的附件，附件是一個字典。每一個字典包含一張圖片的類型，請求地址，名稱等信息。附件的縮略圖是默認顯示第一張的圖片，點擊縮略圖，把所有的圖片信息顯示出來。

解決思路：一是根據總共返回的圖片個數，在控制器內顯示imageView,這個時候需要重新布局，根據SDImageView來獲取所有的圖片

二是，取出所有字典里面圖片的地址，放到html字符串里面，把這些地址直接當成html里面圖片的地址，然后直接加載，一個webview就搞定了。


- (void)loadHTMLString:(NSString*)string baseURL:(nullableNSURL*)baseURL;

但是，在不驗證請求頭的條件下，以上都是可以實現的，后來后臺加上了請求頭的驗證，對于第一種實現方式，直接在網絡請求的時候加上請求頭即可。

對于第二種方式，若是單次的一個webview請求，監聽webview的代理方法可以實現，但是只能用一次，對于多個的圖片，沒法在這里添加請求頭。（添加請求頭之后要重新請求，會造成循環）

- (BOOL)webView:(UIWebView*)webView shouldStartLoadWithRequest:(NSURLRequest*)request navigationType:(UIWebViewNavigationType)navigationType

因為程序中很多地方使用第二種方式， 基于改動最小的原則，經過分析，可以在webview,或者AFNetworking的上一層來添加請求頭，這樣，不管是哪種請求方式，都要經過這里和底層進行交互，也可以是url重定向，也可以解決DNS域名劫持問題，可以使用NSURLProtocol來解決。


NSURLProtocol

NSURLProtocol能夠讓你去重新定義蘋果的URL加載系統(URL Loading System)的行為，URL Loading System里有許多類用于處理URL請求，比如NSURL，NSURLRequest，NSURLConnection和NSURLSession等，當URL Loading System使用NSURLRequest去獲取資源的時候，它會創建一個NSURLProtocol子類的實例，你不應該直接實例化一個NSURLProtocol，NSURLProtocol看起來像是一個協議，但其實這是一個類，而且必須使用該類的子類，并且需要被注冊。

使用場景

不管你是通過UIWebView, NSURLConnection 或者第三方庫 (AFNetworking, MKNetworkKit等)，他們都是基于NSURLConnection或者 NSURLSession實現的，因此你可以通過NSURLProtocol做自定義的操作。

1.重定向網絡請求

2.忽略網絡請求，使用本地緩存

3.自定義網絡請求的返回結果

4.一些全局的網絡請求設置

5.攔截網絡請求

子類化NSURLProtocol并注冊

@interfaceCustomURLProtocol: NSURLProtocol@end

然后在application:didFinishLaunchingWithOptions:方法中注冊該CustomURLProtocol，一旦注冊完畢后，它就有機會來處理所有交付給URL Loading system的網絡請求。（也可以在具體的某一個控制器里面）

- (BOOL)application:(UIApplication*)application didFinishLaunchingWithOptions:(NSDictionary*)launchOptions {//注冊protocol[NSURLProtocolregisterClass:[CustomURLProtocolclass]];returnYES;}

實現CustomURLProtocol

注冊好了之后，現在可以開始實現NSURLProtocol的一些方法：

+canInitWithRequest:

這個方法主要是說明你是否打算處理對應的request，如果不打算處理，返回NO，URL Loading System會使用系統默認的行為去處理；如果打算處理，返回YES，然后你就需要處理該請求的所有東西，包括獲取請求數據并返回給 URL Loading System。網絡數據可以簡單的通過NSURLConnection去獲取，而且每個NSURLProtocol對象都有一個NSURLProtocolClient實例，可以通過該client將獲取到的數據返回給URL Loading System。

這里有個需要注意的地方，想象一下，當你去加載一個URL資源的時候，URL Loading System會詢問CustomURLProtocol是否能處理該請求，你返回YES，然后URL Loading System會創建一個CustomURLProtocol實例然后調用NSURLConnection去獲取數據，然而這也會調用URL Loading System，而你在+canInitWithRequest:中又總是返回YES，這樣URL Loading System又會創建一個CustomURLProtocol實例導致無限循環。我們應該保證每個request只被處理一次，可以通過+setProperty:forKey:inRequest:標示那些已經處理過的request，然后在+canInitWithRequest:中查詢該request是否已經處理過了，如果是則返回NO。

+ (BOOL)canInitWithRequest:(NSURLRequest*)request{//只處理http和https請求NSString*scheme = [[request URL] scheme];if( ([scheme caseInsensitiveCompare:@"http"] ==NSOrderedSame||? ? [scheme caseInsensitiveCompare:@"https"] ==NSOrderedSame))? ? {//看看是否已經處理過了，防止無限循環if([NSURLProtocolpropertyForKey:URLProtocolHandledKey inRequest:request]) {returnNO;? ? ? ? }returnYES;? ? }returnNO;}

+canonicalRequestForRequest:

通常該方法你可以簡單的直接返回request，但也可以在這里修改request，比如添加header，修改host等，并返回一個新的request，這是一個抽象方法，子類必須實現。

+ (NSURLRequest*) canonicalRequestForRequest:(NSURLRequest*)request {NSMutableURLRequest*mutableReqeust = [request mutableCopy];? ? mutableReqeust = [selfredirectHostInRequset:mutableReqeust];returnmutableReqeust;}+(NSMutableURLRequest*)redirectHostInRequset:(NSMutableURLRequest*)request{if([request.URL host].length ==0) {returnrequest;? ? }NSString*originUrlString = [request.URL absoluteString];NSString*originHostString = [request.URL host];NSRangehostRange = [originUrlString rangeOfString:originHostString];if(hostRange.location ==NSNotFound) {returnrequest;? ? }//定向到bing搜索主頁NSString*ip =@"cn.bing.com";// 替換域名NSString*urlString = [originUrlString stringByReplacingCharactersInRange:hostRange withString:ip];NSURL*url = [NSURLURLWithString:urlString];? ? request.URL = url;returnrequest;}

+requestIsCacheEquivalent:toRequest:

主要判斷兩個request是否相同，如果相同的話可以使用緩存數據，通常只需要調用父類的實現。

+ (BOOL)requestIsCacheEquivalent:(NSURLRequest *)atoRequest:(NSURLRequest *)b{return[superrequestIsCacheEquivalent:atoRequest:b];}

-startLoading -stopLoading

這兩個方法主要是開始和取消相應的request，而且需要標示那些已經處理過的request。

- (void)startLoading{NSMutableURLRequest*mutableReqeust = [[selfrequest] mutableCopy];//標示改request已經處理過了，防止無限循環[NSURLProtocolsetProperty:@YES forKey:URLProtocolHandledKey inRequest:mutableReqeust];self.connection = [NSURLConnectionconnectionWithRequest:mutableReqeust delegate:self];}- (void)stopLoading{? ? [self.connection cancel];}

NSURLConnectionDataDelegate方法

在處理網絡請求的時候會調用到該代理方法，我們需要將收到的消息通過client返回給URL Loading System。

- (void) connection:(NSURLConnection*)connection didReceiveResponse:(NSURLResponse*)response {? ? [self.client URLProtocol:selfdidReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];}- (void) connection:(NSURLConnection*)connection didReceiveData:(NSData*)data {? ? [self.client URLProtocol:selfdidLoadData:data];}- (void) connectionDidFinishLoading:(NSURLConnection*)connection {? ? [self.client URLProtocolDidFinishLoading:self];}- (void)connection:(NSURLConnection*)connection didFailWithError:(NSError*)error {? ? [self.client URLProtocol:selfdidFailWithError:error];}

參考文章：

http://www.cocoachina.com/ios/20141225/10765.html

http://www.lxweimin.com/p/7c89b8c5482a

http://www.lxweimin.com/p/f9ecdb697fd9

http://www.cnblogs.com/wobuyayi/p/6283599.html

最后編輯于 ：
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明：文章內容（如有圖片或視頻亦包括在內）由作者上傳并發布，文章內容僅代表作者本人觀點，簡書系信息發布平臺，僅提供信息存儲服務。

推薦閱讀更多精彩內容

  • 由于最近項目要區分是否為移動端，因此需要手動將URL請求的域名重定向到指定的IP地址，但是由于請求可能是通過NSU...
    All_Be_Alright閱讀 388評論 0贊 0
  • iOS開發系列--網絡開發 概覽 大部分應用程序都或多或少會牽扯到網絡開發，例如說新浪微博、微信等，這些應用本身可...
    lichengjin閱讀 3,721評論 2贊 7
  • 概覽 緩存組件應該說是每個客戶端程序必備的核心組件，試想對于每個界面的訪問都必須重新請求勢必降低用戶體驗。但是如何...
    默默_David閱讀 1,967評論 1贊 9
  • iOS網絡編程讀書筆記 Facade Tester客戶端門面模式的實例(被動版本化) 被動版本化，所以硬編碼URL...
    melouverrr閱讀 1,631評論 3贊 7
  • PS:附上原文，好好學習。 一、聽到不中聽的話：四種選擇 別人的行為可能會刺激我們，但并不是我們感受的根源。 非暴...
    歲月蓮上寫詩閱讀 264評論 0贊 0