利用NSURLSession实现断点下载
思路:
1将文件名,存放路径,文件总长度,以及文件已经下载的长度先规定好。
2开始下载时 判断是否下载过,如果下载过且完成了,则提示下载完成,下载过未完成
则从上次下载的位置开始下载 通过设置NSMutableURLRequest的头
NSString *range = [NSString stringWithFormat:@"bytes=%zd-", XMGDownloadLength];
[request setValue:range forHTTPHeaderField:@"Range"];
3在didReceiveResponse收到响应时,打开NSOutputStream文件流,存储文件的总长度,并且允许下载
4在didReceiveData中不断的通过NSOutputStream将数据写入到沙盒中,记录下载进度
5在didCompleteWithError中完成 关闭NSOutputStream 清除任务
通过以上代码就可以完成断点下载功能
// 所需要下载的文件的URL
#define XMGFileURL @"http://120.25.226.186:32812/resources/videos/minion_01.mp4"
// 文件名(沙盒中的文件名)
#define XMGFilename XMGFileURL.md5String
// 文件的存放路径(caches)
#define XMGFileFullpath [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:XMGFilename]
// 存储文件总长度的文件路径(caches)
#define XMGTotalLengthFullpath [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:@"totalLength.xmg"]
// 文件的已下载长度
#define XMGDownloadLength [[[NSFileManager defaultManager] attributesOfItemAtPath:XMGFileFullpath error:nil][NSFileSize] integerValue]
#import "ViewController.h"
#import "NSString+Hash.h"
#import "UIImageView+WebCache.h"
@interface ViewController () <NSURLSessionDataDelegate>
/** 下载任务 */
@property (nonatomic, strong) NSURLSessionDataTask *task;
/** session */
@property (nonatomic, strong) NSURLSession *session;
/** 写文件的流对象 */
@property (nonatomic, strong) NSOutputStream *stream;
/** 文件的总长度 */
@property (nonatomic, assign) NSInteger totalLength;
@end
@implementation ViewController
- (NSURLSession *)session
{
if (!_session) {
_session = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration] delegate:self delegateQueue:[[NSOperationQueue alloc] init]];
}
return _session;
}
- (NSOutputStream *)stream
{
if (!_stream) {
_stream = [NSOutputStream outputStreamToFileAtPath:XMGFileFullpath append:YES];
}
return _stream;
}
- (NSURLSessionDataTask *)task
{
if (!_task) {
NSInteger totalLength = [[NSDictionary dictionaryWithContentsOfFile:XMGTotalLengthFullpath][XMGFilename] integerValue];
if (totalLength && XMGDownloadLength == totalLength) {
NSLog(@"----文件已经下载过了");
return nil;
}
// 创建请求
NSMutableURLRequest *request = [NSMutableURLRequest requestWithURL:[NSURL URLWithString:@"http://120.25.226.186:32812/resources/videos/minion_01.mp4"]];
// 设置请求头
// Range : bytes=xxx-xxx
NSString *range = [NSString stringWithFormat:@"bytes=%zd-", XMGDownloadLength];
[request setValue:range forHTTPHeaderField:@"Range"];
// 创建一个Data任务
_task = [self.session dataTaskWithRequest:request];
}
return _task;
}
- (void)viewDidLoad
{
[super viewDidLoad];
NSLog(@"%@", XMGFileFullpath);
}
/**
* 开始下载
*/
- (IBAction)start:(id)sender {
// 启动任务
[self.task resume];
}
/**
* 暂停下载
*/
- (IBAction)pause:(id)sender {
[self.task suspend];
}
#pragma mark - <NSURLSessionDataDelegate>
/**
* 1.接收到响应
*/
- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveResponse:(NSHTTPURLResponse *)response completionHandler:(void (^)(NSURLSessionResponseDisposition))completionHandler
{
// 打开流
[self.stream open];
// 获得服务器这次请求 返回数据的总长度
self.totalLength = [response.allHeaderFields[@"Content-Length"] integerValue] + XMGDownloadLength;
// 存储总长度
NSMutableDictionary *dict = [NSMutableDictionary dictionaryWithContentsOfFile:XMGTotalLengthFullpath];
if (dict == nil) dict = [NSMutableDictionary dictionary];
dict[XMGFilename] = @(self.totalLength);
[dict writeToFile:XMGTotalLengthFullpath atomically:YES];
// 接收这个请求,允许接收服务器的数据
completionHandler(NSURLSessionResponseAllow);
}
/**
* 2.接收到服务器返回的数据(这个方法可能会被调用N次)
*/
- (void)URLSession:(NSURLSession *)session dataTask:(NSURLSessionDataTask *)dataTask didReceiveData:(NSData *)data
{
// 写入数据
[self.stream write:data.bytes maxLength:data.length];
// 下载进度
NSLog(@"%f", 1.0 * XMGDownloadLength / self.totalLength);
}
/**
* 3.请求完毕(成功\失败)
*/
- (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error
{
// 关闭流
[self.stream close];
self.stream = nil;
// 清除任务
self.task = nil;
}