iOS 客户端基于 WebP 图片格式的流量优化（下）

在iOS 客户端基于 WebP 图片格式的流量优化（上）这篇文章中，已经介绍了WebP格式图片的下载使用，仅仅只有这样还远远不够，还需要对已经下载的图片数据进行缓存。

曾经有句名言『计算机世界有两大难题，第一是起名字，第二是写一个缓存』，鄙人不能同意更多。

在iOS上，重写一份图片缓存是不现实的，而直接修改SDWebImage框架也是不太好的。所以，在SDWebImage的基础上添加一个中间层CacheManager比较好。

我感觉，缓存的难度在于，如何准确命中。的确在开发的时候，一大半时间都是在测试缓存命中情况，测试本身就挺麻烦，需要在模拟器的沙盒里面看文件，同时断网测试，需要一些调试技巧，很多技巧并么有办法详尽表述出来，需要所谓的悟性去理解。

一、SDWebImage缓存处理

这一部分，由于SD下载图片的方法中，url被替换，所以要看懂SD本身的代码，是什么时候给缓存一个确定的key。发现在

- (id <SDWebImageOperation>)downloadImageWithURL:(NSURL *)urloptions:(SDWebImageOptions)optionsprogress:(SDWebImageDownloaderProgressBlock)progressBlockcompleted:(SDWebImageCompletionWithFinishedBlock)completedBlock {if ([url isKindOfClass:NSString.class]) {url = [NSURL URLWithString:(NSString *)url];}if (![url isKindOfClass:NSURL.class]) {url = nil;}url = [url qd_replaceToWebPURLWithScreenWidth];......

方法中，确定了缓存的key值

    NSString *key = [self cacheKeyForURL:url];operation.cacheOperation = [self.imageCache queryDiskCacheForKey:key done:^(UIImage *image, SDImageCacheType cacheType) {

这也就是之前，为什么要在这个方法的最前面把URL替换掉，这样，SD的key值已经是保护WebP格式的图片URL，这一部分的缓存都可以正常使用，不需要修改。

所以，难度还是在WebView的图片缓存中，因为之前虽然是用SD托管WebView中WebP图片的下载，然而WebView读缓存却不能自动从SDImageCache中读取。这样，需要用NSURLCache来接管WebView的图片缓存。

二、WebView图片缓存

关于WebView的缓存，系统提供了一个类，NSURLCache。这个类可以在所有的网络请求前查看缓存，并且决定是否缓存（注意：是所有请求）。具体的NSURLCache用法，动动勤劳的小手Google一下，很多文章可以参考。

我们自己的实现，直接上代码

@implementation QDURLCache/***  请求完成决定是否要将response进行存储*/
- (void)storeCachedResponse:(NSCachedURLResponse *)cachedResponse forRequest:(NSURLRequest *)request {NSString* ua = [request valueForHTTPHeaderField:@"User-Agent"];if (!EmptyString(ua) && [ua lf_containsSubString:@"AppleWebKit"]) {//判断本次请求是不是请求图片if ([[QDCacheManager defaultManager] isImageRequest:request]) {[[QDCacheManager defaultManager] storeImageResponse:cachedResponse forRequest:request];return;}//其他请求if ([[QDCacheManager defaultManager] shouldCacheExceptImageResponseForRequest:request]) {if (![[QDCacheManager defaultManager] storeCachedResponse:cachedResponse forRequest:request]) {[super storeCachedResponse:cachedResponse forRequest:request];return;} else {return;}}}[super storeCachedResponse:cachedResponse forRequest:request];
}/***  每次发请求之前会调此方法,查看本地是否有缓存*/
- (NSCachedURLResponse *)cachedResponseForRequest:(NSURLRequest *)request {NSString* ua = [request valueForHTTPHeaderField:@"User-Agent"];if (!EmptyString(ua) && [ua lf_containsSubString:@"AppleWebKit"]) {if ([[QDCacheManager defaultManager] isImageRequest:request]) { //图片//从本地取图片NSCachedURLResponse *imageCacheResponse = [[QDCacheManager defaultManager] retrieveImageCacheResponseForRequest:request];if (imageCacheResponse) {return imageCacheResponse;} else {return [super cachedResponseForRequest:request];}}if ([[QDCacheManager defaultManager] shouldCacheExceptImageResponseForRequest:request]) { //其它缓存的东西//判断本地自定义缓存目录是否存在if (![[QDCacheManager defaultManager] cacheAvaliableForRequest:request]) {NSCachedURLResponse *response = [super cachedResponseForRequest:request];//判断本地系统缓存目录是否存在if (response.data) {BOOL contentLengthValid = [((NSHTTPURLResponse *)response.response) expectedContentLength] == [response.data length];//判断是否是有效的文件if (!contentLengthValid) {return response;}//将系统缓存放到自定义的缓存目录中[[QDCacheManager defaultManager] storeCachedResponse:response forRequest:request];} else {}return response;}//从本地缓存中取出对应的缓存NSCachedURLResponse *cachedResponse = [[QDCacheManager defaultManager] retrieveCachedResponseForRequest:request];if (cachedResponse) {return cachedResponse;}}}return [super cachedResponseForRequest:request];
}- (void)removeCachedResponseForRequest:(NSURLRequest *)request {if ([[QDCacheManager defaultManager] cacheAvaliableForRequest:request]) {if (![[QDCacheManager defaultManager] removeCachedResponseForRequest:request]) {LogI(@"Failed to remove local cache for request: %@", request.URL);}} else {[super removeCachedResponseForRequest:request];}
}@end

这段代码并没有多么难以理解的地方，可以看出来，我们是新建了一个中间层QDCacheManager，来管理WebView的所有缓存。

而且，既然是全局影响，肯定要用UA包起来，防止误伤其他缓存。

这一段代码在调试的时候有个技巧，就是所有super方法的调用，在测试阶段，全部直接return，防止WebView自身的缓存干扰调试结果。这个方法在很多缓存处理的地方都需要注意，别的地方但凡出现了调用super方法的，调试中也一律是直接return的。

既然已经用QDCacheManager托管了缓存，URLCache类的任务就已经完成，储存Response由

- (void)storeCachedResponse:(NSCachedURLResponse *)cachedResponse forRequest:(NSURLRequest *)request

而下面：

- (NSCachedURLResponse *)cachedResponseForRequest:(NSURLRequest *)request

在NSURLProtocol的startLoading方法执行之前，就调用了。很好理解，因为这个方法就是取缓存的方法，自然是先取，没有再去Loading。

这里的逻辑，必须通过大量调试，反复验证，不能简单套用别人的结论，甚至官方文档也要怀疑的态度来看。因为，很多第三方框架，会影响NSURLCache类，我在调试时，就发现，JSPatch，React Native还有我们的一个放劫持服务，都有可能影响这个类中方法的调用。

下面就转入我们自己的缓存管理方法中去，由于现在关注的是WebP图片问题，所以，其他缓存处理就不再展开。

三、中间层CacheManager处理

关于这个中间层，主要处理的实际就是缓存key的问题，因为请求的时候，request里的URL仍然是没有替换WebP的，所以，需要先用之前qd_defultWebPURLCacheKey方法来获取真实图片缓存key值。思路的关键就是换key，再取cache，代码本身就只能靠功底了。

直接上代码，没什么好解释的。

- (BOOL)isImageRequest:(NSURLRequest *)request {if (![request.URL.absoluteString qd_isQdailyHost]) {return NO;}NSArray *extensions = @[@".jpg", @".jpeg", @".png", @".gif"];for (NSString *extension in extensions) {if ([request.URL.absoluteString.lowercaseString lf_containsSubString:extension]){return YES;}}return NO;
}- (void)storeImageResponse:(NSCachedURLResponse *)cachedResponse forRequest:(NSURLRequest *)request {NSString *key = [request.URL qd_defultWebPURLCacheKey];if ([_imageCache imageFromDiskCacheForKey:key]) {return;}dispatch_async([_imageCache currentIOQueue], ^{// 硬盘缓存直接存data，webp格式；内存缓存为UIImage，可以直接使用[_imageCache storeImageDataToDisk:cachedResponse.data forKey:key];});
}- (NSCachedURLResponse *)retrieveImageCacheResponseForRequest:(NSURLRequest *)request {NSString *key = [request.URL qd_defultWebPURLCacheKey];NSString *defaultPath = [_imageCache defaultCachePathForKey:key];NSData *data = nil;if ([_imageCache imageFromMemoryCacheForKey:key]) {UIImage * image = [_imageCache imageFromMemoryCacheForKey:key];if ([key lf_containsSubString:@".png"]) {data = UIImagePNGRepresentation(image);} else {data = UIImageJPEGRepresentation(image, 1.0);}}if (data &&  data.length != 0) {NSURLResponse *response = [[NSURLResponse alloc] initWithURL:request.URLMIMEType:[request.URL.absoluteString qd_MIMEType]expectedContentLength:data.lengthtextEncodingName:nil];return [[NSCachedURLResponse alloc] initWithResponse:response data:data];}data = [NSData dataWithContentsOfFile:defaultPath];if (data == nil) {data = [NSData dataWithContentsOfFile:[defaultPath stringByDeletingPathExtension]];}if (data == nil || data.length == 0) {[_imageCache removeImageForKey:key fromDisk:YES];return nil;}NSURLResponse *response = [[NSURLResponse alloc] initWithURL:request.URLMIMEType:[request.URL.absoluteString qd_MIMEType]expectedContentLength:data.lengthtextEncodingName:nil];return [[NSCachedURLResponse alloc] initWithResponse:response data:data];
}

其中currentIOQueue方法，是修改了一下SDImageCache，暴露这个IOQueue，原来的框架是没有这个方法的。

至于为什么图片硬盘缓存直接用data，因为这里考虑是性能问题，取缓存的时候，返回的NSURLResponse所携带的，肯定还是NSData，如果当时存了UIImage格式，内部一样是转码成了NSData，而取的时候，还是按UIImage格式取，再转成NSData返回，相当于多了两次转码。

内存缓存却没有这个问题，因为SD的内存缓存，用的NSCache，存的就是UIImage对象，可以直接取出来用。

这里其实仍然并没有什么好讲的，还是基本的逻辑问题，需要比较严谨地处理。

四、其他情况的特别处理

我们的app是实现了wifi预加载了，然而这一部分也需要与上面完成的缓存体系通用，不然，wifi预加载的意义就不大。

首先，我们的wifi预加载，是自己写了一个URLSession，所以在下载前替换URL就可以

 for (NSString *urlString in resourcesArray) {if ([urlString isKindOfClass:[NSString class]]) {NSURL *theURL = [NSURL URLWithString:urlString];if ([[QDCacheManager defaultManager] isImageRequest:[NSURLRequest requestWithURL:theURL]]) {theURL = [theURL qd_replaceToWebPURLWithScreenWidth];}if(![[QDCacheManager defaultManager] cacheAvaliableForURL:theURL] && ![[SDImageCache sharedImageCache] diskImageExistsWithKey:theURL.absoluteString]) {__weak QDPrefetcher* weakSelf = self;[self.session getTasksWithCompletionHandler:^(NSArray *dataTasks, NSArray *uploadTasks, NSArray *downloadTasks) {......

部分代码如上，关键也在于替换URL时机和判断缓存情况。而下载之后的文件存到哪，是需要处理的。

#pragma mark NSURLSession Delegate- (void)URLSession:(NSURLSession *)session downloadTask:(NSURLSessionDownloadTask *)downloadTask didFinishDownloadingToURL:(NSURL *)location {NSError *error = nil;NSFileManager *fileManager = [NSFileManager defaultManager];NSString *destinationPath = nil;if ([[QDCacheManager defaultManager] isImageRequest:downloadTask.originalRequest]) {NSString *key = downloadTask.originalRequest.URL.absoluteString;destinationPath = [[SDImageCache sharedImageCache] defaultCachePathForKey:key];} else {destinationPath = [[QDCacheManager defaultManager] localCachedWebContentPathWithRequest:downloadTask.originalRequest];}if ([fileManager fileExistsAtPath:destinationPath]) {[fileManager removeItemAtPath:destinationPath error:nil];}[fileManager copyItemAtPath:location.path toPath:destinationPath error:&error];}

我是在finish的方法里面，把图片下载的目录直接copy给SDImageCache的缓存目录。这样，SD的缓存里面就有了这些WebP格式的NSData，与之前的代码逻辑统一，格式统一。

总结

首先有了一个心得，看上去很复杂的功能，可能实际代码并不需要自己写多少，学会在前人的基础上再加工，比如我们现在这套WebP适配，底层仍然是SDWebImage的基本逻辑，我们只不过在上层，加一些判断和处理，来适应业务层丰富的功能。

而且，代码是一步步写出来的，提前设想的方案，并不一定能实现，先实现功能，再优化架构，才是正确的方向。当时在WebURLProtocol里面，绕了很大的弯子，甚至还涉及到了多线程问题，不小心发现了iOS8，9，10三个版本的内部实现都在变化，绕开了一个个坑，才逐步清晰了整个逻辑。

总结整个方案的逻辑，其实比较清晰：

首先确定是不是需要被替换的图片URL，然后所有的替换都采用统一方法，与之配套的key，也用这套方法处理得到他被替换后的URL，保证命中。
然后，无论Native请求还是WebView请求，都用SD托管，避免两套处理逻辑造成的种种不确定性；
而WebView的缓存，通过一个中间层处理，再交给SDImageCache，使之与Native请求的数据统一，让两种图片请求公用一套缓存，进一步重用。

思路大致如此，其他的问题，就需要靠代码能力了。