iOS 实现断点续传

网络下载是我们在项目中经常要用到的功能,如果是小文件的下载,比如图片和文字之类的,我们可以直接请求源地址,然后一次下载完毕。但是如果是下载较大的音频和视频文件,不可能一次下载完毕,用户可能下载一段时间,关闭程序,回家接着下载。这个时候,就需要实现断点续传的功能。让用户可以随时暂停下载,下次开始下载,还能接着上次的下载的进度

网络下载是我们在项目中经常要用到的功能,如果是小文件的下载,比如图片和文字之类的,我们可以直接请求源地址,然后一次下载完毕。但是如果是下载较大的音频和视频文件,不可能一次下载完毕,用户可能下载一段时间,关闭程序,回家接着下载。这个时候,就需要实现断点续传的功能。让用户可以随时暂停下载,下次开始下载,还能接着上次的下载的进度。

今天我们来看看如何自己简单的封装一个断点续传的类,实现如下功能。

使用者只需要调用一个接口即可以下载,同时可以获取下载的进度。

下载成功,可以获取文件存储的位置

下载失败,给出失败的原因

可以暂停下载,下次开始下载,接着上次的进度继续下载

原理讲解

要实现断点续传的功能,通常都需要客户端记录下当前的下载进度,并在需要续传的时候通知服务端本次需要下载的内容片段。

在HTTP1.1协议(RFC2616)中定义了断点续传相关的HTTP头的Range和Content-Range字段,一个最简单的断点续传实现大概如下:

客户端下载一个1024K的文件,已经下载了其中512K

网络中断,客户端请求续传,因此需要在HTTP头中申明本次需要续传的片段:

Range:bytes=512000-

这个头通知服务端从文件的512K位置开始传输文件

服务端收到断点续传请求,从文件的512K位置开始传输,并且在HTTP头中增加:

Content-Range:bytes 512000-/1024000

并且此时服务端返回的HTTP状态码应该是206,而不是200。

难点说明

1. 客户端如何获取已经下载的文件字节数

客户端这边,我们需要记录下每次用户每次下载的文件大小,然后实现原理讲解中步骤1的功能。

那么如何记载呢?

其实我们可以直接获取指定路径下文件的大小,iOS已经提供了相关的功能,实现代码如下,

[[[NSFileManager defaultManager] attributesOfItemAtPath: FileStorePath error:nil][NSFileSize] integerValue]

2.如何获取被下载文件的总字节数

上一步,我们获取了已经下载文件的字节数,这里我们需要获取被下载文件的总字节数,有了这两个值,我们就可以算出下载进度了。

那么如何获取呢?这里我们需要用到http 头部的conten-length字段,先来看看该字段的含义

Content-Length用于描述HTTP消息实体的传输长度the transfer-length of the message-body。在HTTP协议中,消息实体长度和消息实体的传输长度是有区别,比如说gzip压缩下,消息实体长度是压缩前的长度,消息实体的传输长度是gzip压缩后的长度。

简单点说,content-length表示被下载文件的字节数。

对比原理讲解的第三步,我们可以看到如果要计算出文件的总字节数,那么必须把已经下载的字节数 加上 content-length。

我们需要把每个被下载文件的总字节数存储起来,这里我们选择使用plist文件来记载,plist文件包含一个字典。设置文件名为键值,已经下载的文件字节数为值。

文件名为了防止重复,这里我们设置文件名为下载url的hash值,可以保证不重重。

实现代码如下:

- (void)URLSession:(NSURLSession*)session dataTask:(NSURLSessionDataTask*)dataTask didReceiveResponse:(NSHTTPURLResponse*)response completionHandler:(void(^)(NSURLSessionResponseDisposition))completionHandler{

self.totalLength = [response.allHeaderFields[@"Content-Length"] integerValue] +  DownloadLength;NSMutableDictionary*dict = [NSMutableDictionarydictionaryWithContentsOfFile: TotalLengthPlist];

if(dict ==nil) dict = [NSMutableDictionarydictionary];   

 dict[ Filename] = @(self.totalLength);    [dict writeToFile: TotalLengthPlist atomically:YES];}

上述NSSessionDelegate方法会在请求收到回应的时候调用一次,我们可以在该方法中获取回应信息,取出content-length字段。

3.封装一个方法,实现下载进度,成功,失败提示

我们可以模仿AFNetwork,把下载封装到一个方法,然后使用不同的block来实现下载进度,成功,失败后的回调。

定义如下:

-(void)downLoadWithURL:(NSString*)URL   progress:(progressBlock)progressBlock              success:(successBlock)successBlock   faile:(faileBlock)faileBlock{

self.successBlock = successBlock;self.failedBlock = faileBlock;self.progressBlock = progressBlock;self.downLoadUrl = URL;    [self.task resume];}

上面的三个block都采用宏定义方式,这样看起来比较简洁,具体代码参考下面的完整代码。

然后我们可以在NSURLSessionDataDelegate的对应的代理方法中去实现三个block的调用,然后传入相应的参数。这样当其他人调用我们的方法,就可以在相应的block中实现回调。具体代码参考下面的完整代码

完整代码实现

下面是完整的代码实现

#import

typedefvoid(^successBlock) (NSString*fileStorePath);

typedefvoid(^faileBlock) (NSError*error);

typedefvoid(^progressBlock) (floatprogress);

@interfaceDownLoadManager:NSObject

@property(copy) successBlock  successBlock;

@property(copy) faileBlock      failedBlock;

@property(copy) progressBlock    progressBlock;

-(void)downLoadWithURL:(NSString*)URL  progress:(progressBlock)progressBlock              success:(successBlock)successBlock faile:(faileBlock)faileBlock;+ (instancetype)sharedInstance;

-(void)stopTask;

@end

#import"DownLoadManager.h"

#import"NSString+Hash.h"

@interfaceDownLoadManager()/** 下载任务 */@property(nonatomic,strong)NSURLSessionDataTask*task;/** session */@property(nonatomic,strong)NSURLSession*session;/** 写文件的流对象 */@property(nonatomic,strong)NSOutputStream*stream;/** 文件的总大小 */@property(nonatomic,assign)NSIntegertotalLength;@property(nonatomic,strong)NSString*downLoadUrl;

@end

// 文件名(沙盒中的文件名),使用md5哈希url生成的,这样就能保证文件名唯一#define  Filename  self.downLoadUrl.md5String// 文件的存放路径(caches)#define  FileStorePath [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent: Filename]// 使用plist文件存储已经下载的文件大小#define  TotalLengthPlist [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:@"totalLength.plist"]// 文件的已被下载的大小#define  DownloadLength [[[NSFileManager defaultManager] attributesOfItemAtPath: FileStorePath error:nil][NSFileSize] integerValue]@implementationDownLoadManager#pragma mark  - 创建单例staticid_instance;+ (instancetype)allocWithZone:(struct_NSZone *)zone{staticdispatch_once_tonceToken;dispatch_once(&onceToken, ^{        _instance = [superallocWithZone:zone];    });return_instance;}+ (instancetype)sharedInstance{staticdispatch_once_tonceToken;dispatch_once(&onceToken, ^{        _instance = [[selfalloc] init];    });return_instance;}- (id)copyWithZone:(NSZone*)zone{return_instance;}- (id)mutableCopyWithZone:(NSZone*)zone {return_instance;}#pragma mark - 公开方法-(void)downLoadWithURL:(NSString*)URL              progress:(progressBlock)progressBlock              success:(successBlock)successBlock                faile:(faileBlock)faileBlock{self.successBlock = successBlock;self.failedBlock = faileBlock;self.progressBlock = progressBlock;self.downLoadUrl = URL;    [self.task resume];}-(void)stopTask{    [self.task suspend ];}#pragma mark  - getter方法- (NSURLSession*)session{if(!_session) {        _session = [NSURLSessionsessionWithConfiguration:[NSURLSessionConfigurationdefaultSessionConfiguration] delegate:selfdelegateQueue:[[NSOperationQueuealloc] init]];    }return_session;}- (NSOutputStream*)stream{if(!_stream) {        _stream = [NSOutputStreamoutputStreamToFileAtPath: FileStorePath append:YES];    }return_stream;}- (NSURLSessionDataTask*)task{if(!_task) {NSIntegertotalLength = [[NSDictionarydictionaryWithContentsOfFile: TotalLengthPlist][ Filename] integerValue];if(totalLength &&  DownloadLength == totalLength) {NSLog(@"######文件已经下载过了");returnnil;        }// 创建请求NSMutableURLRequest*request = [NSMutableURLRequestrequestWithURL:[NSURLURLWithString:self.downLoadUrl]];// 设置请求头// Range : bytes=xxx-xxx,从已经下载的长度开始到文件总长度的最后都要下载NSString*range = [NSStringstringWithFormat:@"bytes=%zd-",  DownloadLength];        [request setValue:range forHTTPHeaderField:@"Range"];// 创建一个Data任务_task = [self.session dataTaskWithRequest:request];    }return_task;}#pragma mark -/**

* 1.接收到响应

*/- (void)URLSession:(NSURLSession*)session dataTask:(NSURLSessionDataTask*)dataTask didReceiveResponse:(NSHTTPURLResponse*)response completionHandler:(void(^)(NSURLSessionResponseDisposition))completionHandler{// 打开流[self.stream open];/*

(Content-Length字段返回的是服务器对每次客户端请求要下载文件的大小)

比如首次客户端请求下载文件A,大小为1000byte,那么第一次服务器返回的Content-Length = 1000,

客户端下载到500byte,突然中断,再次请求的range为 “bytes=500-”,那么此时服务器返回的Content-Length为500

所以对于单个文件进行多次下载的情况(断点续传),计算文件的总大小,必须把服务器返回的content-length加上本地存储的已经下载的文件大小

*/self.totalLength = [response.allHeaderFields[@"Content-Length"] integerValue] +  DownloadLength;// 把此次已经下载的文件大小存储在plist文件NSMutableDictionary*dict = [NSMutableDictionarydictionaryWithContentsOfFile: TotalLengthPlist];if(dict ==nil) dict = [NSMutableDictionarydictionary];    dict[ Filename] = @(self.totalLength);    [dict writeToFile: TotalLengthPlist atomically:YES];// 接收这个请求,允许接收服务器的数据completionHandler(NSURLSessionResponseAllow);}/**

* 2.接收到服务器返回的数据(这个方法可能会被调用N次)

*/- (void)URLSession:(NSURLSession*)session dataTask:(NSURLSessionDataTask*)dataTask didReceiveData:(NSData*)data{// 写入数据[self.stream write:data.bytes maxLength:data.length];floatprogress =1.0*  DownloadLength /self.totalLength;if(self.progressBlock) {self.progressBlock(progress);    }// 下载进度}/**

* 3.请求完毕(成功\失败)

*/- (void)URLSession:(NSURLSession*)session task:(NSURLSessionTask*)task didCompleteWithError:(NSError*)error{if(error) {if(self.failedBlock) {self.failedBlock(error);        }self.stream =nil;self.task =nil;    }else{if(self.successBlock) {self.successBlock(FileStorePath);        }// 关闭流[self.stream close];self.stream =nil;// 清除任务self.task =nil;    }}@end

如何调用

@interfaceViewController()@end@implementationViewController/**

* 开始下载

*/- (IBAction)start:(id)sender {

// 启动任务NSString* downLoadUrl =@"http://audio.xmcdn.com/group11/M01/93/AF/wKgDa1dzzJLBL0gCAPUzeJqK84Y539.m4a";    [[DownLoadManager sharedInstance]downLoadWithURL:downLoadUrl progress:^(floatprogress) {NSLog(@"###%f",progress);    } success:^(NSString*fileStorePath) {NSLog(@"###%@",fileStorePath);    } faile:^(NSError*error) {NSLog(@"###%@",error.userInfo[NSLocalizedDescriptionKey]);    }];}/**

* 暂停下载

*/- (IBAction)pause:(id)sender {    [[DownLoadManager sharedInstance]stopTask];}@end

总结

这里只能实现单个任务下载,大家可以自己想想办法,看如何实现多任务下载,并且实现断点续传功能。

并且为了更加便于操作,建议把存储信息换成使用数据库存储。

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

推荐阅读更多精彩内容