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

3110861-42f511634a28dc33.png
效果图.png
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值