iOS邊練邊學--NSURLSessionDataTask實現文件真正的斷點續傳

實現重點：

• NSURLSessionDataTask要設置請求頭，從路徑中獲取文件已經下載的長度（文件沒有下載過的話，長度為0）。通過這個長度設置請求的Range

如圖：

• 接收到請求的時候key：文件名（經過MD5加密過的URL，Url保證了文件名的唯一） Value：該文件已經下載過的長度。保存成plist文件，方便對下載文件的判斷
• 利用NSOutUpStream寫文件
• 在任務完成的代理方法里面，NSOutUpStream關閉並且清空，對應的task清空，對應的session清空

代碼如下：

#import "ViewController.h"
#import "NSString+Hash.h"

// 下載文件的URL
#define ChaosFileURL @"http://120.25.226.186:32812/resources/videos/minion_01.mp4"

// 根據文件唯一的URL MD5值 作為文件名
#define ChaosFileName ChaosFileURL.md5String

// 用來存儲文件總長度的plist文件 key:文件名的MD5值 value:文件總長度
#define ChaosDownloadFilesPlist [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:@"downloadFiles.plist"]

// 下載文件的全路徑
#define ChaosFileFullPath [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:ChaosFileName]

// 已經下載的文件長度
#define ChaosDownloadLength [[[NSFileManager defaultManager] attributesOfItemAtPath:ChaosFileFullPath error:nil][@"NSFileSize"] integerValue]

@interface ViewController () <NSURLSessionDataDelegate>

/** stream */
@property(nonatomic,strong) NSOutputStream *stream;

/** session */
@property(nonatomic,strong) NSURLSession *session;

/** task */
@property(nonatomic,strong) NSURLSessionDataTask *task;

/** totalLength */
@property(nonatomic,assign) NSInteger totalLength;

/** downloadLength */
@property(nonatomic,assign) NSInteger downloadLength;

@end

@implementation ViewController

- (NSURLSession *)session
{
    if (_session == nil) {
        
        _session = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration] delegate:self delegateQueue:[[NSOperationQueue alloc] init]];
    }
    
    return _session;
}

- (NSURLSessionDataTask *)task
{
    if (_task == nil) {
        
        // 獲得文件總長度
        NSInteger totalLength = [[NSDictionary dictionaryWithContentsOfFile:ChaosDownloadFilesPlist][ChaosFileName] integerValue];
        // 請求同一個文件，判斷下載文件長度；如果沒下載過此文件，totalLength = 0
        if (totalLength && ChaosDownloadLength == totalLength) {
            NSLog(@"文件已經下載過.");
            return nil;
        }
        
        NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:[NSURL URLWithString:ChaosFileURL]];
        
        // 設置請求頭 -- range 這次從哪里開始請求數據 格式：bytes=***-***（從指定開始到指定結束）  或者：bytes=***-（從指定位置到結束）
        NSString *range = [NSString stringWithFormat:@"bytes=%zd-",ChaosDownloadLength];
        
        [request setValue:range forHTTPHeaderField:@"Range"];
        
        _task = [self.session dataTaskWithRequest:request];
        
    }
    return _task;
}

// 開始
- (IBAction)startClick:(id)sender {
    
    [self.task resume];
}
// 暫停
- (IBAction)pauseClick:(id)sender {
    
    [self.task suspend];
}

- (void)viewDidLoad {
    [super viewDidLoad];
}

#pragma mark - <NSURLSessionDataDelegate>
/** 
 * 接收到響應的時候調用
 */
- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveResponse:(NSHTTPURLResponse *)response completionHandler:(void (^)(NSURLSessionResponseDisposition))completionHandler
{
    // 調用blcok，才能接受到數據
    completionHandler(NSURLSessionResponseAllow);
    // 初始化stream
    self.stream = [NSOutputStream outputStreamToFileAtPath:ChaosFileFullPath append:YES];
    [self.stream open];
    
    // 獲取文件總長度
    self.totalLength = [response.allHeaderFields[@"Content-Length"] integerValue] + ChaosDownloadLength;
    
    // 接收到服務器響應的時候存儲文件的總長度到plist,實現多文件下載，先取出字典，給字典賦值最后寫入。
    // 錯誤做法：直接寫入文件，會用這次寫入的信息覆蓋原來所有的信息
    NSMutableDictionary *dict = [NSMutableDictionary dictionaryWithContentsOfFile:ChaosDownloadFilesPlist];
    // 字典可能為空
    if (dict == nil) dict = [NSMutableDictionary dictionary];
    // 寫入文件
    dict[ChaosFileName] = @(self.totalLength);
    [dict writeToFile:ChaosDownloadFilesPlist atomically:YES];
}

/**
 * 接收到服務器發來的數據的時候調用 -- 有可能調用多次
 */
- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveData:(NSData *)data
{
    // 寫入數據
    [self.stream write:[data bytes] maxLength:data.length];
    // 獲取已經下載的長度
    self.downloadLength = ChaosDownloadLength;
    // 計算進度
    NSLog(@"%f",1.0 * self.downloadLength / self.totalLength);
}

/**
 * 任務完成的時候調用
 */
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error
{
    NSLog(@"----finish");
    [self.stream close];
    self.stream = nil;
    
    // 一個任務對應一個文件，用完清空
    self.task = nil;
}
@end