项目总结四:网络请求中,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.jianshu.com/p/7c89b8c5482a

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

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

推荐阅读更多精彩内容

  • 由于最近项目要区分是否为移动端,因此需要手动将URL请求的域名重定向到指定的IP地址,但是由于请求可能是通过NSU...
    谪守京都阅读 107评论 0 0
  • iOS开发系列--网络开发 概览 大部分应用程序都或多或少会牵扯到网络开发,例如说新浪微博、微信等,这些应用本身可...
    lichengjin阅读 1,964评论 0 6
  • iOS架构设计-URL缓存(上) 2017-07-15崔江涛Cocoa开发者社区 转载自崔江涛(KenshinCu...
    C9090阅读 667评论 0 1
  • 概览 缓存组件应该说是每个客户端程序必备的核心组件,试想对于每个界面的访问都必须重新请求势必降低用户体验。但是如何...
    默默_David阅读 946评论 1 9
  • iOS网络编程读书笔记 Facade Tester客户端门面模式的实例(被动版本化) 被动版本化,所以硬编码URL...
    melouverrr阅读 1,171评论 3 7