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. 一些全局的網絡請求設置

 接觸過iOS系統中URL Loading System都知道，NSURLProtocol是如此地強大，可以攔截應用內幾乎所有的網絡請求（除了WKWebView），并可以修改請求頭，返回client任意自定義的數據等等，據說很多做網絡緩存都是利用這個類的。

那么，首先講解一下NSURLProtocol怎么使用吧。

1. 定義一個NSURLProtocol的子類

在繼承NSURLProtocol中，我們需要實現

+ (BOOL)canInitWithRequest:(NSURLRequest *)request, 定義攔截請求的URL規則

- (void)startLoading, 對于攔截的請求，系統創建一個NSURLProtocol對象執行startLoading方法開始加載請求

- (void)stopLoading，對于攔截的請求，NSURLProtocol對象在停止加載時調用該方法

+ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request，可選方法，對于需要修改請求頭的請求在該方法中修改

下面代碼定義了一個專門攔截https請求的NSURLProtocol子類，并通過CFHttpMessageRef重新請求

@interface CFHttpMessageURLProtocol () <NSStreamDelegate> {   NSMutableURLRequest *curRequest;   NSRunLoop *curRunLoop;   NSInputStream *inputStream; }  @end  @implementation CFHttpMessageURLProtocol  /**  * 是否攔截處理指定的請求  *  * @param request 指定的請求  *  * @return 返回YES表示要攔截處理，返回NO表示不攔截處理  */ + (BOOL)canInitWithRequest:(NSURLRequest *)request {      /* 防止無限循環，因為一個請求在被攔截處理過程中，也會發起一個請求，這樣又會走到這里，如果不進行處理，就會造成無限循環 */   if ([NSURLProtocol propertyForKey:protocolKey inRequest:request]) {     return NO;   }      NSString *url = request.URL.absoluteString;      // 如果url以https開頭，則進行攔截處理，否則不處理   if ([url hasPrefix:@"https"]) {     return YES;   }   return NO; }  /**  * 如果需要對請求進行重定向，添加指定頭部等操作，可以在該方法中進行  */ + (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {   return request; }  /**  * 開始加載，在該方法中，加載一個請求  */ - (void)startLoading {   NSMutableURLRequest *request = [self.request mutableCopy];   // 表示該請求已經被處理，防止無限循環   [NSURLProtocol setProperty:@(YES) forKey:protocolKey inRequest:request];   curRequest = request;   [self startRequest]; }  /**  * 取消請求  */ - (void)stopLoading {   if (inputStream.streamStatus == NSStreamStatusOpen) {     [inputStream removeFromRunLoop:curRunLoop forMode:NSRunLoopCommonModes];     [inputStream setDelegate:nil];     [inputStream close];   }   [self.client URLProtocol:self didFailWithError:[[NSError alloc] initWithDomain:@"stop loading" code:-1 userInfo:nil]]; } 

以上代碼中的startRequest方法是通過復制原始請求頭，使用CFHttpMessageRef重新發起請求的，關于這部分的代碼由于跟本文章內容關系不大，這里就先不放，有興趣的朋友可以參考我的下一篇博客。

2. 在網絡請求前注冊NSURLProtocol

 // 注冊攔截請求的NSURLProtocol [NSURLProtocol registerClass:[CFHttpMessageURLProtocol class]]; 

對于NSURLSession的請求，注冊NSURLProtocol的方式稍有不同，是通過NSURLSessionConfiguration注冊的

// NSURLSession例子 NSURLSessionConfiguration *configuration = [NSURLSessionConfiguration defaultSessionConfiguration]; NSArray *protocolArray = @[ [CFHttpMessageURLProtocol class] ]; configuration.protocolClasses = protocolArray; NSURLSession *session = [NSURLSession sessionWithConfiguration:configuration delegate:self delegateQueue:[NSOperationQueue mainQueue]]; NSURLSessionTask *task = [session dataTaskWithRequest:_request]; [task resume]; 

3. 請求結束后注銷NSURLProtocol

[NSURLProtocol unregisterClass:[CFHttpMessageURLProtocol class]]; 

好了，到這里NSURLProtocol的使用方法大家應該有所了解了。下面主要講一下NSURLProtocol在使用過程中可能會遇到的坑，給自己以及需要的朋友留個提醒。

1. 上面一開始就已經說了，對于WebView的請求，目前NSURLProtocol還不能攔截WKWebView的請求，只能攔截UIWebview的，但后者好像AppStore已經不讓審核通過了（尷尬臉）。

2. NSURLProtocol在攔截NSURLSession的POST請求時不能獲取到Request中的HTTPBody，這個貌似早就國外的論壇上傳開了，但國內好像還鮮有人知，據蘋果官方的解釋是Body是NSData類型，即可能為二進制內容，而且還沒有大小限制，所以可能會很大，為了性能考慮，索性就攔截時就不拷貝了（內流滿面臉）。為了解決這個問題，我們可以通過把Body數據放到Header中，不過Header的大小好像是有限制的，我試過2M是沒有問題，不過超過10M就直接Request timeout了。。。而且當Body數據為二進制數據時這招也沒轍了，因為Header里都是文本數據，另一種方案就是用一個NSDictionary或NSCache保存沒有請求的Body數據，用URL為key，最后方法就是別用NSURLSession，老老實實用古老的NSURLConnection算了。。。

3. 使用NSURLProtocol時，在那兩個類方法可以發送同步網絡請求，而實例方法，如startLoading則進入死鎖，直至超時，原因是執行實例方法所在的線程并沒有啟動runloop，而NSURLConnection這些網絡請求需要依賴于runloop的，因此這些請求根本發不出去，所以必須使用異步請求，NSURLConnection/NSURLSession的異步請求的線程保證啟動了runloop。

以上就是我目前發現的坑，歡迎大家補充，也希望對大家開發有所幫助哈~所幸的是NSURLProtocol對于大量并發的請求支持的還不錯，不然就要棄用了~希望對大家的學習有所幫助，也希望大家多多支持VEVB武林網。