攔截webP的網(wǎng)絡(luò)請求焕参，利用SDWebImage進(jìn)行轉(zhuǎn)碼，然后加載

//
//  WebPLoadProtocol.m
//  WCFriend
//
//  Created by apple on 2016/11/18.
//  Copyright ? 2016年 apple. All rights reserved.
//

/*
 
 在需要加載webP的控制器加入此句
 [NSURLProtocol registerClass:[WebPLoadProtocol class]];
 
 */

#import "WebPLoadProtocol.h"
#import "UIImage+WebP.h"

static NSString *URLProtocolHandledKey = @"URLHasHandle";

@interface WebPLoadProtocol ()<NSURLSessionDelegate>

@property(nonatomic, strong) NSURLSession *session;

@end

@implementation WebPLoadProtocol

#pragma mark - 初始化請求

+ (BOOL)canInitWithRequest:(NSURLRequest *)request
{
    //只處理http和https請求
    NSString *scheme = [[request URL] scheme];
    if ( ([scheme caseInsensitiveCompare:@"http"] == NSOrderedSame ||
          [scheme caseInsensitiveCompare:@"https"] == NSOrderedSame))
    {
        if ([NSURLProtocol propertyForKey:URLProtocolHandledKey inRequest:request]) {
            return NO;
        }
        return YES;
    }
    return NO;
}

+ (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request {
    return request;
}

#pragma mark - 通信協(xié)議內(nèi)容實(shí)現(xiàn)

- (void)startLoading
{
    NSMutableURLRequest *mutableReqeust = [[self request] mutableCopy];
    //標(biāo)示改request已經(jīng)處理過了没讲，防止無限循環(huán)
    [NSURLProtocol setProperty:@YES forKey:URLProtocolHandledKey inRequest:mutableReqeust];
    
    if (![[self.request.URL absoluteString] hasSuffix:@"webp"]) {
        
        NSLog(@"是一個(gè)正常的,%@",self.request.URL);
        
    }else{
        NSLog(@"檢測到了 webp---%@,攔截它",self.request.URL);
    }
    
    NSURLSessionConfiguration *config = [NSURLSessionConfiguration defaultSessionConfiguration];
    self.session = [NSURLSession sessionWithConfiguration:config delegate:self delegateQueue:[NSOperationQueue currentQueue]];
    [[self.session dataTaskWithRequest:mutableReqeust] resume];
    
}

- (void)stopLoading
{
    [self.session invalidateAndCancel];
}

#pragma mark dataDelegate
- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveResponse:(NSURLResponse *)response completionHandler:(void (^)(NSURLSessionResponseDisposition disposition))completionHandler
{
    [self.client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
    
    if (completionHandler) {
        completionHandler(NSURLSessionResponseAllow);
    }
    
}

- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveData:(NSData *)data{
    
    NSData *transData = data;
    
    if ([dataTask.currentRequest.URL.absoluteString hasSuffix:@"webp"]) {
        NSLog(@"webp---%@---替換它",dataTask.currentRequest.URL);
        //采用 SDWebImage 的轉(zhuǎn)換方法
        UIImage *imgData = [UIImage sd_imageWithWebPData:data];
        transData = UIImageJPEGRepresentation(imgData, 1.0f);
    }
    
    [self.client URLProtocol:self didLoadData:transData];
}

- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(nullable NSError *)error{
    
    if (error) {
        
        [self.client URLProtocol:self didFailWithError:error];
    }else{
        
        [self.client URLProtocolDidFinishLoading:self];
    }
    
}
@end