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