iOS Downloader断点续传

1. 最简单的方式

1). 创建一个下载工具类MTDownloader
MTDownloader.h文件

#import <Foundation/Foundation.h>

@interface MTDownloader : NSObject

- (void)download:(NSString *)netUrl;

@end

MTDownloader.m文件

// 下载,传入下载链接
- (void)download:(NSString *)netUrl {
    NSURL *url = [NSURL URLWithString:netUrl];
    NSURLRequest *request = [NSURLRequest requestWithURL:url];
    [NSURLConnection sendAsynchronousRequest:request queue:[NSOperationQueue mainQueue] completionHandler:
     ^(NSURLResponse * _Nullable response, NSData * _Nullable data, NSError * _Nullable connectionError) {
         if (connectionError) {
             NSLog(@"连接错误 %@", connectionError);
             return;
         }
         NSHTTPURLResponse *httpResponse = (NSHTTPURLResponse *)response;
         if (httpResponse.statusCode == 200 || httpResponse.statusCode == 304) {
             // 解析数据
             // 存在的问题,内存暴涨,无法获取下载进度
             [data writeToFile:@"/Users/mazaiting/Desktop/1.zip" atomically:YES];
             NSLog(@"下载成功");
         } else {
             NSLog(@"服务器内部错误");
         }
     }];
}

这种下载方式存在的问题是下载时内存暴涨,因为文件读取到内存中,并且无法获取下载进度。
2). 在ViewController.m文件中使用

#import "ViewController.h"
#import "MTDownloader.h"
@interface ViewController ()

@end

@implementation ViewController

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

// 触摸屏幕时
- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event {
    MTDownloader *downloader = [MTDownloader new];
    [downloader download:@"http://127.0.0.1/HBuilder.zip"];
}

@end

2. 解决获取下载进度问题

改造MTDownloader.m文件


#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 记录文件的总长度,因为在获取下载进度的时候需要文件总长度,所以定义为属性
@property (nonatomic, assign) long long expectedContentLength;
// 记录当前文件的长度
@property (nonatomic, assign) long long currentFileSize;
// 用来记录总共下载的数据
@property (nonatomic, strong) NSMutableData *mutableData;
@end

@implementation MTDownloader

// 懒加载
- (NSMutableData *)mutableData {
    if (_mutableData == nil) {
        _mutableData = [NSMutableData data];
    }
    return _mutableData;
}

- (void)download:(NSString *)netUrl {
    // 创建url
    NSURL *url = [NSURL URLWithString:netUrl];
    // 创建下载请求
    NSURLRequest *request = [NSURLRequest requestWithURL:url];
    // 设置代理, 实现NSURLConnectionDataDelegate协议
    [[NSURLConnection alloc] initWithRequest:request delegate:self];
}

#pragma mark- 代理方法
// 接收到响应头
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
    NSLog(@"接收到响应 %@", response);
    // 获取到文件的总长度
    self.expectedContentLength = response.expectedContentLength;
}

// 接收到数据--一点点接收
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // 将每次接收到的字节长度与已有的文件长度相加
    self.currentFileSize += data.length;
    // 添加接收的数据
    [self.mutableData appendData:data];
    // 当前已下载的文件长度 * 1.0 / 文件的总长度 = 当前的进度
    float process = self.currentFileSize * 1.0 / self.expectedContentLength;
    NSLog(@"接收数据: %f", process);
}

// 下载完成
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
    NSLog(@"下载完成");
    // 下载完成后将字节数组写入文件
    [self.mutableData writeToFile:@"/Users/mazaiting/Desktop/1.zip" atomically:YES];
}

// 下载出错
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
    NSLog(@"下载出错");
}

@end

改造完成后,可以获取下载文件的进度,但仍然后出现内存暴涨的问题。

3. 解决文件内存暴涨--NSFileHandle/NSOutputStream

将第二步中记录下载数据的mutableData属性删除,在此处不再需要,因为是将文件一点一点的写入。
在接收一点点接收数据的地方进行数据的存储。

1). NSFileHandle解决

#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 记录文件的总长度,因为在获取下载进度的时候需要文件总长度,所以定义为属性
@property (nonatomic, assign) long long expectedContentLength;
// 记录当前文件的长度
@property (nonatomic, assign) long long currentFileSize;
@end

@implementation MTDownloader

- (void)download:(NSString *)netUrl {
    // 创建url
    NSURL *url = [NSURL URLWithString:netUrl];
    // 创建下载请求
    NSURLRequest *request = [NSURLRequest requestWithURL:url];
    // 设置代理, 实现NSURLConnectionDataDelegate协议
    [[NSURLConnection alloc] initWithRequest:request delegate:self];
}

#pragma mark- 代理方法
// 接收到响应头
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
    NSLog(@"接收到响应 %@", response);
    // 获取到文件的总长度
    self.expectedContentLength = response.expectedContentLength;
}

// 接收到数据--一点点接收
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // 将每次接收到的字节长度与已有的文件长度相加
    self.currentFileSize += data.length;
    // 当前已下载的文件长度 * 1.0 / 文件的总长度 = 当前的进度
    float process = self.currentFileSize * 1.0 / self.expectedContentLength;
    NSLog(@"接收数据: %f", process);
    // 保存数据
    [self saveData:data];
}

// 下载完成
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
    NSLog(@"下载完成");
}

// 下载出错
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
    NSLog(@"下载出错");
}

// 保存数据
// data 每次下载的二进制数据
- (void)saveData:(NSData *)data {
    // 保存文件的路径
    NSString *filePath = @"/Users/mazaiting/Desktop/1.zip";
    // 根据路径获取NSFileHandle, 它不会自动创建文件,而是第0个字节添加数据
    NSFileHandle *fileHandle = [NSFileHandle fileHandleForWritingAtPath:filePath];
    // 如果fileHandle存在,则让fileHandle的offset偏移量指向文件的末尾,进行数据的添加
    // 如果fileHandle不存在,则创建文件
    if (fileHandle) {
        // fileHandle存在
        // 将文件指针指向文件的末尾
        [fileHandle seekToEndOfFile];
        // 写数据
        [fileHandle writeData:data];
        // 关闭文件
        [fileHandle closeFile];
    } else {
        // fileHandle不存在
        [data writeToFile:filePath atomically:YES];
    }
}

@end

2). NSOutputStream解决
流操作时要注意打开流,关闭流,并且要注意初始化流的位置。文件写入时推荐使用流操作。

#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 记录文件的总长度,因为在获取下载进度的时候需要文件总长度,所以定义为属性
@property (nonatomic, assign) long long expectedContentLength;
// 记录当前文件的长度
@property (nonatomic, assign) long long currentFileSize;
// NSOutputStream用来保存每次下载的二进制数据
@property (nonatomic, strong) NSOutputStream *outputStream;
@end

@implementation MTDownloader

- (void)download:(NSString *)netUrl {
    // 创建url
    NSURL *url = [NSURL URLWithString:netUrl];
    // 创建下载请求
    NSURLRequest *request = [NSURLRequest requestWithURL:url];
    // 设置代理, 实现NSURLConnectionDataDelegate协议
    [[NSURLConnection alloc] initWithRequest:request delegate:self];
}

#pragma mark- 代理方法
// 接收到响应头
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
    NSLog(@"接收到响应 %@", response);
    // 获取到文件的总长度
    self.expectedContentLength = response.expectedContentLength;
    // 保存文件的路径
    NSString *filePath = @"/Users/mazaiting/Desktop/1.zip";
    // 接收到响应头的时候初始化outputStream
    self.outputStream = [NSOutputStream outputStreamToFileAtPath:filePath append:YES];
    // 打开流
    [self.outputStream open];
}

// 接收到数据--一点点接收
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // 将每次接收到的字节长度与已有的文件长度相加
    self.currentFileSize += data.length;
    // 当前已下载的文件长度 * 1.0 / 文件的总长度 = 当前的进度
    float process = self.currentFileSize * 1.0 / self.expectedContentLength;
    NSLog(@"接收数据: %f", process);
    // 保存数据
    [self saveData:data];
}

// 下载完成
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
    NSLog(@"下载完成");
    // 下载完成的时候,关闭流
    [self.outputStream close];}

// 下载出错
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
    NSLog(@"下载出错");
    // 判断流是否为空,下载出错的时候,关闭流
    if (self.outputStream) {
        [self.outputStream close];
    }
}

// NSOutputStream保存数据
// data 每次下载的二进制文件
- (void)saveData:(NSData *)data {
    // 写数据
    [self.outputStream write:data.bytes maxLength:data.length];
}

// NSFileHandle保存数据
// data 每次下载的二进制数据
- (void)saveData1:(NSData *)data {
    // 保存文件的路径
    NSString *filePath = @"/Users/mazaiting/Desktop/1.zip";
    // 根据路径获取NSFileHandle, 它不会自动创建文件,而是第0个字节添加数据
    NSFileHandle *fileHandle = [NSFileHandle fileHandleForWritingAtPath:filePath];
    // 如果fileHandle存在,则让fileHandle的offset偏移量指向文件的末尾,进行数据的添加
    // 如果fileHandle不存在,则创建文件
    if (fileHandle) {
        // fileHandle存在
        // 将文件指针指向文件的末尾
        [fileHandle seekToEndOfFile];
        // 写数据
        [fileHandle writeData:data];
        // 关闭文件
        [fileHandle closeFile];
    } else {
        // fileHandle不存在
        [data writeToFile:filePath atomically:YES];
    }
}

@end

此时,一个下载的操作工具类就已完成,但目前无法使用断点下载。

4. 文件的断点下载操作

断点续传思路
I. 首先发送一个head请求,获取到服务器文件的大小和名称
II. 判断本地文件是否存在
1). 不存在,从0开始下载
2). 存在
a. 本地文件大小 == 服务器文件大小 不需要下载
b. 本地文件大小 < 服务器文件大小 从当前位置下载
c. 本地文件打下 > 服务器文件大小 删除文件,从0下载

//
// 断点续传思路
// 1. 首先发送一个head请求,获取到服务器文件的大小和名称
// 2. 判断本地文件是否存在
//      1). 不存在,从0开始下载
//      2). 存在
//          a. 本地文件大小 == 服务器文件大小  不需要下载
//          b. 本地文件大小 < 服务器文件大小   从当前位置下载
//          c. 本地文件打下 > 服务器文件大小   删除文件,从0下载
#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 记录文件的总长度,因为在获取下载进度的时候需要文件总长度,所以定义为属性
@property (nonatomic, assign) long long expectedContentLength;
// 记录当前文件的长度
@property (nonatomic, assign) long long currentFileSize;
// NSOutputStream用来保存每次下载的二进制数据
@property (nonatomic, strong) NSOutputStream *outputStream;
// 文件的存储路径
@property (nonatomic, strong) NSString *filePath;
@end

@implementation MTDownloader

- (void)download:(NSString *)netUrl {
    // 创建url
    NSURL *url = [NSURL URLWithString:netUrl];
    // 获取服务器文件信息, 获取当前文件长度
    [self getServerFileInfo:url];
    // 获取本地文件信息,并且和服务器做比较
    self.currentFileSize = [self getLocalFileInfo];
    // 判断当前文件大小是否是-1,如果是-1,说明文件已经下载完成
    if (self.currentFileSize == -1) {
        NSLog(@"文件已下载完成,请不要重复下载!!!");
        return;
    }
    
    // 从指定位置处下载文件
    [self downloadFile: url];
}

// 从指定位置处下载文件
- (void)downloadFile:(NSURL *)url {
    // 创建一个请求
    NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
    // 请求头
    // Range:bytes=x-y 从x个字节下载到y个字节
    // Range:bytes=x-  从x个字节下载到最后
    // Range:bytes=-y  从开始下载到y个字节
    [request setValue:[NSString stringWithFormat:@"bytes=%lld-", self.currentFileSize] forHTTPHeaderField:@"Range"];
    
    // 设置代理, 实现NSURLConnectionDataDelegate协议
    [[NSURLConnection alloc] initWithRequest:request delegate:self];
}

// 检测本地文件大小
- (long long)getLocalFileInfo {
    // 文件的长度, 如果文件不存在,则为0
    long long fileSize = 0;
    // 获取文件管理者对象
    NSFileManager *fileManager = [NSFileManager defaultManager];
    // 判断文件是否存在
    if ([fileManager fileExistsAtPath:self.filePath]) {
        // 文件存在
        // 获取本地文件的大小
        NSDictionary *attrs = [fileManager attributesOfItemAtPath:self.filePath error:NULL];
        // 赋值
        fileSize = attrs.fileSize;
        // 文件大小等于服务器文件长度
        if (fileSize == self.expectedContentLength) {
            fileSize = -1;
        }
        // 文件大小大于服务器文件长度
        if (fileSize > self.expectedContentLength) {
            fileSize = 0;
            // 删除文件
            [fileManager removeItemAtPath:self.filePath error:NULL];
        }
    }
    return fileSize;
}

// 获取服务器文件信息
// 文件的url
- (void)getServerFileInfo:(NSURL *)url {
    // 创建一个请求
    NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
    // 设置请求方法
    request.HTTPMethod = @"head";
    // 同步请求响应体
    NSURLResponse *response = nil;
    // 发送一个同步请求
    [NSURLConnection sendSynchronousRequest:request returningResponse:&response error:NULL];
    
    // 设置文件的总长度,注意:在下载时- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response;方法中为当前文件总长度赋值的语句要删除
    self.expectedContentLength = response.expectedContentLength;
    // 设置文件的存储路径, 存储在临时目录下,拼接文件名
    self.filePath = [NSTemporaryDirectory() stringByAppendingPathComponent:response.suggestedFilename];
    NSLog(@"%@", self.filePath);
}

#pragma mark- 代理方法
// 接收到响应头
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
    NSLog(@"接收到响应 %@", response);
    // 接收到响应头的时候初始化outputStream
    self.outputStream = [NSOutputStream outputStreamToFileAtPath:self.filePath append:YES];
    // 打开流
    [self.outputStream open];
}

// 接收到数据--一点点接收
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // 将每次接收到的字节长度与已有的文件长度相加
    self.currentFileSize += data.length;
    // 当前已下载的文件长度 * 1.0 / 文件的总长度 = 当前的进度
    float process = self.currentFileSize * 1.0 / self.expectedContentLength;
    NSLog(@"接收数据: %f", process);
    // 保存数据
    [self saveData:data];
}

// 下载完成
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
    NSLog(@"下载完成");
    // 下载完成的时候,关闭流
    [self.outputStream close];}

// 下载出错
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
    NSLog(@"下载出错");
    // 判断流是否为空,下载出错的时候,关闭流
    if (self.outputStream) {
        [self.outputStream close];
    }
}

// NSOutputStream保存数据
// data 每次下载的二进制文件
- (void)saveData:(NSData *)data {
    // 写数据
    [self.outputStream write:data.bytes maxLength:data.length];
}

// NSFileHandle保存数据
// data 每次下载的二进制数据
- (void)saveData1:(NSData *)data {
    // 保存文件的路径
    NSString *filePath = @"/Users/mazaiting/Desktop/1.zip";
    // 根据路径获取NSFileHandle, 它不会自动创建文件,而是第0个字节添加数据
    NSFileHandle *fileHandle = [NSFileHandle fileHandleForWritingAtPath:filePath];
    // 如果fileHandle存在,则让fileHandle的offset偏移量指向文件的末尾,进行数据的添加
    // 如果fileHandle不存在,则创建文件
    if (fileHandle) {
        // fileHandle存在
        // 将文件指针指向文件的末尾
        [fileHandle seekToEndOfFile];
        // 写数据
        [fileHandle writeData:data];
        // 关闭文件
        [fileHandle closeFile];
    } else {
        // fileHandle不存在
        [data writeToFile:filePath atomically:YES];
    }
}

@end

5. 下载回调设置

MTDownloader.h 中在download方法中增加成功、进度、失败的对调block。

#import <Foundation/Foundation.h>

@interface MTDownloader : NSObject
// 下载方法
// netUrl 下载的网络连接
// successBlock 成功回调block
// processBlock 进度对调block
// errorBlock 异常回调block
- (void)download:(NSString *)netUrl successBlock:(void(^)(NSString *path))successBlock processBlock:(void(^)(float process))processBlock errorBlock:(void(^)(NSError *error))errorBlock;

@end

MTDownloader.m 中实现该方法,并将下载文件的操作重新开启一个新的线程来执行下载任务。

//
// 断点续传思路
// 1. 首先发送一个head请求,获取到服务器文件的大小和名称
// 2. 判断本地文件是否存在
//      1). 不存在,从0开始下载
//      2). 存在
//          a. 本地文件大小 == 服务器文件大小  不需要下载
//          b. 本地文件大小 < 服务器文件大小   从当前位置下载
//          c. 本地文件打下 > 服务器文件大小   删除文件,从0下载
#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 记录文件的总长度,因为在获取下载进度的时候需要文件总长度,所以定义为属性
@property (nonatomic, assign) long long expectedContentLength;
// 记录当前文件的长度
@property (nonatomic, assign) long long currentFileSize;
// NSOutputStream用来保存每次下载的二进制数据
@property (nonatomic, strong) NSOutputStream *outputStream;
// 文件的存储路径
@property (nonatomic, strong) NSString *filePath;
// 下载成功后的回调
@property (nonatomic, copy) void (^successBlock)(NSString *path);
// 下载进度回调
@property (nonatomic, copy) void (^processBlock)(float process);
// 下载异常的回调
@property (nonatomic, copy) void (^errorBlock)(NSError *error);
@end

@implementation MTDownloader
// 下载
- (void)download:(NSString *)netUrl successBlock:(void (^)(NSString *))successBlock processBlock:(void (^)(float))processBlock errorBlock:(void (^)(NSError *))errorBlock {
    self.successBlock = successBlock;
    self.processBlock = processBlock;
    self.errorBlock = errorBlock;
    
    // 创建url
    NSURL *url = [NSURL URLWithString:netUrl];
    // 获取服务器文件信息, 获取当前文件长度
    [self getServerFileInfo:url];
    // 获取本地文件信息,并且和服务器做比较
    self.currentFileSize = [self getLocalFileInfo];
    // 判断当前文件大小是否是-1,如果是-1,说明文件已经下载完成
    if (self.currentFileSize == -1) {
//        NSLog(@"文件已下载完成,请不要重复下载!!!");
        // 如果下载回调不为空
        if (self.successBlock) {
            // 分发消息到主线程,让主线程来处理
            dispatch_async(dispatch_get_main_queue(), ^{
                self.successBlock(self.filePath);
            });
        }
        return;
    }
    
    // 从指定位置处下载文件
    [self downloadFile: url];
}

// 从指定位置处下载文件
- (void)downloadFile:(NSURL *)url {
    // 创建一个新的线程,并添加到队列中
    [[NSOperationQueue new] addOperationWithBlock:^{
        // 创建一个请求
        NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
        // 请求头
        // Range:bytes=x-y 从x个字节下载到y个字节
        // Range:bytes=x-  从x个字节下载到最后
        // Range:bytes=-y  从开始下载到y个字节
        [request setValue:[NSString stringWithFormat:@"bytes=%lld-", self.currentFileSize] forHTTPHeaderField:@"Range"];
        
        // 设置代理, 实现NSURLConnectionDataDelegate协议
        [[NSURLConnection alloc] initWithRequest:request delegate:self];
        // 开启消息循环
        [[NSRunLoop currentRunLoop] run];
    }];
    
}

// 检测本地文件大小
- (long long)getLocalFileInfo {
    // 文件的长度, 如果文件不存在,则为0
    long long fileSize = 0;
    // 获取文件管理者对象
    NSFileManager *fileManager = [NSFileManager defaultManager];
    // 判断文件是否存在
    if ([fileManager fileExistsAtPath:self.filePath]) {
        // 文件存在
        // 获取本地文件的大小
        NSDictionary *attrs = [fileManager attributesOfItemAtPath:self.filePath error:NULL];
        // 赋值
        fileSize = attrs.fileSize;
        // 文件大小等于服务器文件长度
        if (fileSize == self.expectedContentLength) {
            fileSize = -1;
        }
        // 文件大小大于服务器文件长度
        if (fileSize > self.expectedContentLength) {
            fileSize = 0;
            // 删除文件
            [fileManager removeItemAtPath:self.filePath error:NULL];
        }
    }
    return fileSize;
}

// 获取服务器文件信息
// 文件的url
- (void)getServerFileInfo:(NSURL *)url {
    // 创建一个请求
    NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
    // 设置请求方法
    request.HTTPMethod = @"head";
    // 同步请求响应体
    NSURLResponse *response = nil;
    // 发送一个同步请求
    [NSURLConnection sendSynchronousRequest:request returningResponse:&response error:NULL];
    
    // 设置文件的总长度,注意:在下载时- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response;方法中为当前文件总长度赋值的语句要删除
    self.expectedContentLength = response.expectedContentLength;
    // 设置文件的存储路径, 存储在临时目录下,拼接文件名
    self.filePath = [NSTemporaryDirectory() stringByAppendingPathComponent:response.suggestedFilename];
//    NSLog(@"%@", self.filePath);
}

#pragma mark- 代理方法
// 接收到响应头
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
//    NSLog(@"接收到响应 %@", response);
    // 接收到响应头的时候初始化outputStream
    self.outputStream = [NSOutputStream outputStreamToFileAtPath:self.filePath append:YES];
    // 打开流
    [self.outputStream open];
}

// 接收到数据--一点点接收
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // 将每次接收到的字节长度与已有的文件长度相加
    self.currentFileSize += data.length;
    // 当前已下载的文件长度 * 1.0 / 文件的总长度 = 当前的进度
    float process = self.currentFileSize * 1.0 / self.expectedContentLength;
//    NSLog(@"接收数据: %f", process);
    // 保存数据
    [self saveData:data];
    // 判断进度回调是否为空
    if (self.processBlock) {
        // 如果不为空则回到主线程
        dispatch_async(dispatch_get_main_queue(), ^{
            self.processBlock(process);
        });
    }
}

// 下载完成
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
//    NSLog(@"下载完成");
    // 下载完成的时候,关闭流
    [self.outputStream close];
    
    // 如果下载回调不为空
    if (self.successBlock) {
        // 分发消息到主线程,让主线程来处理
        dispatch_async(dispatch_get_main_queue(), ^{
            self.successBlock(self.filePath);
        });
    }
}

// 下载出错
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
//    NSLog(@"下载出错");
    // 判断流是否为空,下载出错的时候,关闭流
    if (self.outputStream) {
        [self.outputStream close];
    }
    // 判断异常回调是否为空
    if (self.errorBlock) {
        // 不为空回到主线程
        dispatch_async(dispatch_get_main_queue(), ^{
            self.errorBlock(error);
        });
    }
}

// NSOutputStream保存数据
// data 每次下载的二进制文件
- (void)saveData:(NSData *)data {
    // 写数据
    [self.outputStream write:data.bytes maxLength:data.length];
}

// NSFileHandle保存数据
// data 每次下载的二进制数据
- (void)saveData1:(NSData *)data {
    // 保存文件的路径
    NSString *filePath = @"/Users/mazaiting/Desktop/1.zip";
    // 根据路径获取NSFileHandle, 它不会自动创建文件,而是第0个字节添加数据
    NSFileHandle *fileHandle = [NSFileHandle fileHandleForWritingAtPath:filePath];
    // 如果fileHandle存在,则让fileHandle的offset偏移量指向文件的末尾,进行数据的添加
    // 如果fileHandle不存在,则创建文件
    if (fileHandle) {
        // fileHandle存在
        // 将文件指针指向文件的末尾
        [fileHandle seekToEndOfFile];
        // 写数据
        [fileHandle writeData:data];
        // 关闭文件
        [fileHandle closeFile];
    } else {
        // fileHandle不存在
        [data writeToFile:filePath atomically:YES];
    }
}

@end

ViewController.m中的调用

#import "ViewController.h"
#import "MTDownloader.h"
@interface ViewController ()

@end

@implementation ViewController

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

- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event {
    MTDownloader *downloader = [MTDownloader new];
    NSString *url = @"http://127.0.0.1/HBuilder.zip";
    [downloader download:url successBlock:^(NSString *path) {
        NSLog(@"下载成功");
    } processBlock:^(float process) {
        NSLog(@"下载进度: %f", process);
    } errorBlock:^(NSError *error) {
        NSLog(@"下载出错 %@", error);
    }];
}

@end

6. 进度条显示--- 自定义圆形进度条

#import "ViewController.h"
#import "MTDownloader.h"
#import "MTProcessView.h"
@interface ViewController ()
@property (weak, nonatomic) IBOutlet MTProcessView *processView;

@end

@implementation ViewController

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

- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event {
    MTDownloader *downloader = [MTDownloader new];
    NSString *url = @"http://127.0.0.1/HBuilder.zip";
    [downloader download:url successBlock:^(NSString *path) {
        NSLog(@"下载成功");
    } processBlock:^(float process) {
        self.processView.process = process;
        NSLog(@"下载进度: %f", process);
    } errorBlock:^(NSError *error) {
        NSLog(@"下载出错 %@", error);
    }];
}

@end

7. 暂停下载

在MTDownloader.h中添加暂停下载方法

#import <Foundation/Foundation.h>

@interface MTDownloader : NSObject
// ... 省略其他方法
// 暂停下载
- (void)pause;
@end

在MTDownloader.m中实现该方法, 以下内容为该类修改的方法。

#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 下载时创建的NSURLConnection,可用于取消下载
@property (nonatomic, strong) NSURLConnection *urlConnection;
@end

@implementation MTDownloader
// 从指定位置处下载文件
- (void)downloadFile:(NSURL *)url {
    // 创建一个新的线程,并添加到队列中
    [[NSOperationQueue new] addOperationWithBlock:^{
        // 创建一个请求
        NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
        // 请求头
        // Range:bytes=x-y 从x个字节下载到y个字节
        // Range:bytes=x-  从x个字节下载到最后
        // Range:bytes=-y  从开始下载到y个字节
        [request setValue:[NSString stringWithFormat:@"bytes=%lld-", self.currentFileSize] forHTTPHeaderField:@"Range"];
        
        // 设置代理, 实现NSURLConnectionDataDelegate协议
        self.urlConnection = [[NSURLConnection alloc] initWithRequest:request delegate:self];
        // 开启消息循环
        [[NSRunLoop currentRunLoop] run];
    }];
    
}
// 暂停下载
- (void)pause {
    // 取消下载
    [self.urlConnection cancel];
}

8. 下载类的封装

MTDownloadManager.h


#import <Foundation/Foundation.h>

@interface MTDownloadManager : NSObject
// 单例
+ (instancetype)sharedManager;
// 下载方法
// netUrl 下载的网络连接
// successBlock 成功回调block
// processBlock 进度对调block
// errorBlock 异常回调block
- (void)download:(NSString *)netUrl successBlock:(void(^)(NSString *path))successBlock processBlock:(void(^)(float process))processBlock errorBlock:(void(^)(NSError *error))errorBlock;

// 暂停下载
- (void)pause:(NSString *)netUrl;
@end

MTDownloadManager.m

#import "MTDownloadManager.h"
#import "MTDownloader.h"

@interface MTDownloadManager ()
// 下载操作缓存池
@property (nonatomic, strong) NSMutableDictionary *downloadCache;
@end

@implementation MTDownloadManager
// 单例
+ (instancetype)sharedManager {
    static id instance = nil;
    
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        instance = [self new];
    });
    return instance;
}

// 懒加载
- (NSMutableDictionary *)downloadCache {
    if (_downloadCache == nil) {
        _downloadCache = [NSMutableDictionary dictionaryWithCapacity:10];
    }
    return _downloadCache;
}
// 下载
- (void)download:(NSString *)netUrl successBlock:(void (^)(NSString *))successBlock processBlock:(void (^)(float))processBlock errorBlock:(void (^)(NSError *))errorBlock {
    // 判断缓存池中是否有下载操作
    if (self.downloadCache[netUrl]) {
        NSLog(@"正在拼命下载中...");
        return;
    }
    // 新建下载
    MTDownloader *downloader = [MTDownloader download:netUrl successBlock:^(NSString *path) {
        // 移除下载操作
        [self.downloadCache removeObjectForKey:netUrl];
        if (successBlock) {
            successBlock(path);
        }
    } processBlock:processBlock errorBlock:^(NSError *error) {
        // 移除下载操作
        [self.downloadCache removeObjectForKey:netUrl];
        if (errorBlock) {
            errorBlock(error);
        }
    }];
    [[NSOperationQueue new] addOperation:downloader];
    
    // 添加到缓存池中
    [self.downloadCache setObject:downloader forKey:netUrl];
}
// 暂停
- (void)pause:(NSString *)netUrl {
    MTDownloader *downloader = self.downloadCache[netUrl];
    // 判断是否为空,如果不为空,则暂停;如果为空,则什么都不做
    if (downloader == nil) {
        NSLog(@"没有下载操作,无法暂停");
        return;
    }
    
    [downloader pause];
    
    // 删除缓存池中的下载操作
    [self.downloadCache removeObjectForKey:netUrl];
}

@end

MTDownloader.h


#import <Foundation/Foundation.h>

@interface MTDownloader : NSOperation
// 下载方法
// netUrl 下载的网络连接
// successBlock 成功回调block
// processBlock 进度对调block
// errorBlock 异常回调block
+ (instancetype)download:(NSString *)netUrl successBlock:(void(^)(NSString *path))successBlock processBlock:(void(^)(float process))processBlock errorBlock:(void(^)(NSError *error))errorBlock;

// 暂停下载
- (void)pause;
@end

MTDownloader.m

// 断点续传思路
// 1. 首先发送一个head请求,获取到服务器文件的大小和名称
// 2. 判断本地文件是否存在
//      1). 不存在,从0开始下载
//      2). 存在
//          a. 本地文件大小 == 服务器文件大小  不需要下载
//          b. 本地文件大小 < 服务器文件大小   从当前位置下载
//          c. 本地文件打下 > 服务器文件大小   删除文件,从0下载
#import "MTDownloader.h"
// 分类
@interface MTDownloader () <NSURLConnectionDataDelegate>
// 记录文件的总长度,因为在获取下载进度的时候需要文件总长度,所以定义为属性
@property (nonatomic, assign) long long expectedContentLength;
// 记录当前文件的长度
@property (nonatomic, assign) long long currentFileSize;
// NSOutputStream用来保存每次下载的二进制数据
@property (nonatomic, strong) NSOutputStream *outputStream;
// 下载时创建的NSURLConnection,可用于取消下载
@property (nonatomic, strong) NSURLConnection *urlConnection;
// 文件的存储路径
@property (nonatomic, strong) NSString *filePath;
// 下载成功后的回调
@property (nonatomic, copy) void (^successBlock)(NSString *path);
// 下载进度回调
@property (nonatomic, copy) void (^processBlock)(float process);
// 下载异常的回调
@property (nonatomic, copy) void (^errorBlock)(NSError *error);
// 下载链接
@property (nonatomic, copy) NSString *urlString;
@end

@implementation MTDownloader
// 下载
+ (instancetype)download:(NSString *)netUrl successBlock:(void (^)(NSString *))successBlock processBlock:(void (^)(float))processBlock errorBlock:(void (^)(NSError *))errorBlock {
    MTDownloader *downloader = [MTDownloader new];
    downloader.successBlock = successBlock;
    downloader.processBlock = processBlock;
    downloader.errorBlock = errorBlock;
    downloader.urlString = netUrl;
    return downloader;
}

- (void)main {
    @autoreleasepool {
        // 创建url
        NSURL *url = [NSURL URLWithString:self.urlString];
        // 获取服务器文件信息, 获取当前文件长度
        [self getServerFileInfo:url];
        // 获取本地文件信息,并且和服务器做比较
        self.currentFileSize = [self getLocalFileInfo];
        // 判断当前文件大小是否是-1,如果是-1,说明文件已经下载完成
        if (self.currentFileSize == -1) {
            //        NSLog(@"文件已下载完成,请不要重复下载!!!");
            // 如果下载回调不为空
            if (self.successBlock) {
                // 分发消息到主线程,让主线程来处理
                dispatch_async(dispatch_get_main_queue(), ^{
                    self.successBlock(self.filePath);
                });
            }
            return;
        }
        
        // 从指定位置处下载文件
        [self downloadFile: url];
    }
}

// 从指定位置处下载文件
- (void)downloadFile:(NSURL *)url {
    // 创建一个新的线程,并添加到队列中
    [[NSOperationQueue new] addOperationWithBlock:^{
        // 创建一个请求
        NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
        // 请求头
        // Range:bytes=x-y 从x个字节下载到y个字节
        // Range:bytes=x-  从x个字节下载到最后
        // Range:bytes=-y  从开始下载到y个字节
        [request setValue:[NSString stringWithFormat:@"bytes=%lld-", self.currentFileSize] forHTTPHeaderField:@"Range"];
        
        // 设置代理, 实现NSURLConnectionDataDelegate协议
        self.urlConnection = [[NSURLConnection alloc] initWithRequest:request delegate:self];
        // 开启消息循环
        [[NSRunLoop currentRunLoop] run];
    }];
    
}

// 检测本地文件大小
- (long long)getLocalFileInfo {
    // 文件的长度, 如果文件不存在,则为0
    long long fileSize = 0;
    // 获取文件管理者对象
    NSFileManager *fileManager = [NSFileManager defaultManager];
    // 判断文件是否存在
    if ([fileManager fileExistsAtPath:self.filePath]) {
        // 文件存在
        // 获取本地文件的大小
        NSDictionary *attrs = [fileManager attributesOfItemAtPath:self.filePath error:NULL];
        // 赋值
        fileSize = attrs.fileSize;
        // 文件大小等于服务器文件长度
        if (fileSize == self.expectedContentLength) {
            fileSize = -1;
        }
        // 文件大小大于服务器文件长度
        if (fileSize > self.expectedContentLength) {
            fileSize = 0;
            // 删除文件
            [fileManager removeItemAtPath:self.filePath error:NULL];
        }
    }
    return fileSize;
}

// 获取服务器文件信息
// 文件的url
- (void)getServerFileInfo:(NSURL *)url {
    // 创建一个请求
    NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:url];
    // 设置请求方法
    request.HTTPMethod = @"head";
    // 同步请求响应体
    NSURLResponse *response = nil;
    // 发送一个同步请求
    [NSURLConnection sendSynchronousRequest:request returningResponse:&response error:NULL];
    
    // 设置文件的总长度,注意:在下载时- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response;方法中为当前文件总长度赋值的语句要删除
    self.expectedContentLength = response.expectedContentLength;
    // 设置文件的存储路径, 存储在临时目录下,拼接文件名
    self.filePath = [NSTemporaryDirectory() stringByAppendingPathComponent:response.suggestedFilename];
//    NSLog(@"%@", self.filePath);
}

// 暂停下载
- (void)pause {
    // 取消下载
    [self.urlConnection cancel];
    // 取消自定义操作
    [self cancel];
}

#pragma mark- 代理方法
// 接收到响应头
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
//    NSLog(@"接收到响应 %@", response);
    // 接收到响应头的时候初始化outputStream
    self.outputStream = [NSOutputStream outputStreamToFileAtPath:self.filePath append:YES];
    // 打开流
    [self.outputStream open];
}

// 接收到数据--一点点接收
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
    // 将每次接收到的字节长度与已有的文件长度相加
    self.currentFileSize += data.length;
    // 当前已下载的文件长度 * 1.0 / 文件的总长度 = 当前的进度
    float process = self.currentFileSize * 1.0 / self.expectedContentLength;
//    NSLog(@"接收数据: %f", process);
    // 保存数据
    [self saveData:data];
    // 判断进度回调是否为空
    if (self.processBlock) {
        // 如果不为空则回到主线程
        dispatch_async(dispatch_get_main_queue(), ^{
            self.processBlock(process);
        });
    }
}

// 下载完成
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
//    NSLog(@"下载完成");
    // 下载完成的时候,关闭流
    [self.outputStream close];
    
    // 如果下载回调不为空
    if (self.successBlock) {
        // 分发消息到主线程,让主线程来处理
        dispatch_async(dispatch_get_main_queue(), ^{
            self.successBlock(self.filePath);
        });
    }
}

// 下载出错
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
//    NSLog(@"下载出错");
    // 判断流是否为空,下载出错的时候,关闭流
    if (self.outputStream) {
        [self.outputStream close];
    }
    // 判断异常回调是否为空
    if (self.errorBlock) {
        // 不为空回到主线程
        dispatch_async(dispatch_get_main_queue(), ^{
            self.errorBlock(error);
        });
    }
}

// NSOutputStream保存数据
// data 每次下载的二进制文件
- (void)saveData:(NSData *)data {
    // 写数据
    [self.outputStream write:data.bytes maxLength:data.length];
}

// NSFileHandle保存数据
// data 每次下载的二进制数据
- (void)saveData1:(NSData *)data {
    // 保存文件的路径
    NSString *filePath = @"/Users/mazaiting/Desktop/1.zip";
    // 根据路径获取NSFileHandle, 它不会自动创建文件,而是第0个字节添加数据
    NSFileHandle *fileHandle = [NSFileHandle fileHandleForWritingAtPath:filePath];
    // 如果fileHandle存在,则让fileHandle的offset偏移量指向文件的末尾,进行数据的添加
    // 如果fileHandle不存在,则创建文件
    if (fileHandle) {
        // fileHandle存在
        // 将文件指针指向文件的末尾
        [fileHandle seekToEndOfFile];
        // 写数据
        [fileHandle writeData:data];
        // 关闭文件
        [fileHandle closeFile];
    } else {
        // fileHandle不存在
        [data writeToFile:filePath atomically:YES];
    }
}

@end

效果图.png
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 212,718评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,683评论 3 385
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,207评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,755评论 1 284
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,862评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,050评论 1 291
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,136评论 3 410
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,882评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,330评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,651评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,789评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,477评论 4 333
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,135评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,864评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,099评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,598评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,697评论 2 351

推荐阅读更多精彩内容

  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,858评论 25 707
  • iOS开发系列--网络开发 概览 大部分应用程序都或多或少会牵扯到网络开发,例如说新浪微博、微信等,这些应用本身可...
    lichengjin阅读 3,648评论 2 7
  • ## 可重入函数 ### 可重入性的理解 若一个程序或子程序可以安全的被并行执行,则称其为可重入的;即当该子程序正...
    夏至亦韵阅读 700评论 0 0
  • 在上一篇文章开发只懂 AFN ?搞定 NSURLSession 才是硬道理[https://www.jianshu...
    BWLi420阅读 3,896评论 18 63
  • 网络下载是我们在项目中经常要用到的功能,如果是小文件的下载,比如图片和文字之类的,我们可以直接请求源地址,然后一次...
    liang_1阅读 2,433评论 0 53