吐槽 o(╯□╰)o
最近在写的一个项目里面涉及到资源包的下载操作,由于用户在使用过程中有可能会同时下载多个资源包,那么就需要对下载功能进行设计和封装。最开始使用的是使用AFN的下载方法来进行实现的,不过当时项目需求要求不是太高,所以没有处理一些特殊情况,比如任务数量的控制,优先级还有断点下载等等。就简单的在一个单例Manager里面添加了一个字典,将下载资源包的URL地址作为Key,对应的downloadTask作为Value,这样就可以在想要的地方获取到相应的下载进度了。
1 | @property (nonatomic, strong, nonnull) NSMutableDictionary <NSString *, NSURLSessionDownloadTask *> *downloadTaskDict; |
由于现在需要统一管理下载的资源,而且为了表示“用户至上”,比如节约流量需要使用断点下载,包括暂停和继续、应用重启时的断点下载;显示下载速度和比例等等。同时为了控制下载的速度也对最大下载数量进行了限制,默认3个等等。so…..
先说下原理
这里使用了AFN的下载方法来完成下载功能,只不过需要对其进行一些扩展。因此我将每个下载任务封装成了一个模型
1 |
|
这样可以将每一个下载的资源封装成一个Model,这样在使用管理类来控制下载的时候就会显得比较轻松。
在使用AFN之前需要补充说明一下NSURLSessionTask的一些知识:
抽象类(父类):NSURLSessionTask 包含了一些属性,其中使用到的属性包括:
1 | //请求信息,包括URL地址等。 |
同时父类也提供了几个方法
1 | - (void)cancel;//取消下载任务 |
理论上说,如果不做应用重启后的断点下载,通过上面的步骤已经可以完成大部分的下载操作了,但是在程序重启时并不能够获取到对应的已下载的进度(downloadTask会将下载的资源放入tmp路径下面,以’CFNetworking’开头),那这个时候执行下载任务那么会重新下载。考虑到用户的流量,就必须在重启后的进行断点下载,因此我们需要使用到子类NSURLSessionDownloadTask的一个方法
1 | - (void)cancelByProducingResumeData:(void (^)(NSData * __nullable resumeData))completionHandler; |
这个方法可以讲当前的task已下载的数据使用一个data保存起来, 然后使用NSURLSession的方法可以根据这个data从服务器获取后续的数据
1 | - (NSURLSessionDownloadTask *)downloadTaskWithResumeData:(NSData *)resumeData; |
/* 这个方法是经常用到的一个,生成一个NSURLRequest,然后根据这个请求开始下载任务可以追踪到下载的进度,需要设置下载完成后的文件移动到的位置,最后是完成后的回调信息 /
-(NSURLSessionDownloadTask )downloadTaskWithRequest:(NSURLRequest )request
progress:(nullable void (^)(NSProgress downloadProgress)) downloadProgressBlock
destination:(nullable NSURL (^)(NSURL targetPath, NSURLResponse response))destination
completionHandler:(nullable void (^)(NSURLResponse response, NSURL _Nullable filePath, NSError * _Nullable error))completionHandler;
/* 该方法根据一个resumeData来进行断点下载,我这里断点下载的实现就是根据这个来的,resumeData已经在模型里面了。
在下载cancel或者suspend的时候只需要将model的resumeData赋值,然后写入本地plist文件,下次下载的时候根据plist信息获取下载的一些信息就OK了 /
-(NSURLSessionDownloadTask )downloadTaskWithResumeData:(NSData )resumeData
progress:(nullable void (^)(NSProgress downloadProgress)) downloadProgressBlock
destination:(nullable NSURL (^)(NSURL targetPath, NSURLResponse response))destination
completionHandler:(nullable void (^)(NSURLResponse response, NSURL _Nullable filePath, NSError * _Nullable error))completionHandler;
1 |
|
#import <Foundation/Foundation.h>
#import “AFNetworking.h”
#import “FKNetworkingDownloadModel.h”
NS_ASSUME_NONNULL_BEGIN
NS_CLASS_AVAILABLE_IOS(7_0) @interface FKNetworkingManager : NSObject
/**
- download root file path. default is ‘~/cache/forkid.networking.manager1.0’
/
@property (nonatomic, copy, readonly) NSString downloadDirectory;
/**
- contains the models whose is downloading.
/
@property (nonatomic, strong, readonly) NSMutableArray <__kindof FKNetworkingDownloadModel > *downloadingModels;
/**
- contains the models whose is waiting for download.
/
@property (nonatomic, strong, readonly) NSMutableArray <__kindof FKNetworkingDownloadModel > *waitingModels;
/**
- max download mission number.
*/
@property (nonatomic, assign) NSInteger maxDownloadCount;
/**
- first in and fisrt out.
*/
@property (nonatomic, assign) BOOL resumeTaskFIFO;
/**
- ignore maxDownloadCount. manager will resume all downloadTask.
*/
@property (nonatomic, assign, getter=isBatchDownload) BOOL batchDownload;
/**
- singleton
/
+(FKNetworkingManager )shareManager;
/*
===============> Description of download start <=============
= when you wanna download a file from serverce, you can use next methods as you need.
= there have some methods such as start, resume, cancel and so on.
= We add the ‘cancel’ method to pause a task instead of suspend, because we need to resume the task when the App restart.
= all methods will use a download model who is subclass of the ‘FKNetworkingDownloadModel’ for convenience.
= and every model use a URL string as the primary key to mark a download mission or task.
===============> Description of download end. <=============
/
/**
- this method used to start a download mission with a download model, notice that the download model can’t be nil, or the download mission will not execute.
* - @param downloadModel download model
- @param progress progress of the download, track to refresh UI and …
- @param completionHandler you can doing something after download mission completed, such as refresh your UI and move the file and so on.
/
-(void)fk_startDownloadWithDownloadModel:(FKNetworkingDownloadModel )downloadModelprogress:(void (^)(FKNetworkingDownloadModel *downloadModel))progress completionHandler:(void (^)(FKNetworkingDownloadModel *downloadModel, NSError * _Nullable error))completionHandler;
/**
- resume a download task with a download model, it will use the download model’s ‘resumeData’
* - @param downloadModel download model
/
-(void)fk_resumeDownloadWithDownloadModel:(FKNetworkingDownloadModel )downloadModel;
/**
- suspend or cancel a download task
* - @param downloadModel download model
/
-(void)fk_cancelDownloadTaskWithDownloadModel:(FKNetworkingDownloadModel )downloadModel;
/**
- check the resourece has been downloaded or not from the download model resourceURL.
* - @param downloadModel download model
* - @return YES or NO.
/
-(BOOL)fk_hasDownloadedFileWithDownloadModel:(FKNetworkingDownloadModel )downloadModel;
/**
- delete local file if exist.
* - @param downloadModel download model.
/
-(void)fk_deleteDownloadedFileWithDownloadModel:(FKNetworkingDownloadModel )downloadModel;
/**
- delete all downloaded files.
*/
-(void)fk_deleteAllDownloadedFiles;
/**
- get a download model, which is downloading with a URLString. if there is not exist a model, will return nil.
* - @param URLString URLString.
* - @return download model
/
-(nullable FKNetworkingDownloadModel )fk_getDownloadingModelWithURLString:(NSString *)URLString;
/**
- get download progress information. such as download speed, progress and so on.
* - @param downloadModel download model.
* - @return progress model.
/
-(nullable FKNetworkingProgressModel )fk_getDownloadProgressModelWithDownloadModel:(FKNetworkingDownloadModel *)downloadModel;
@end
NS_ASSUME_NONNULL_END
1 |
|
#import “FKNetworkingManager.h”
#import “NSObject+FKAdd.h”
NSString *const FKNetworkingManagerFileName = @”forkid.networking.manager1.0”;
@interface FKNetworkingManager ()
/**
- AFNetworking manager.
/
@property (nonatomic, strong) AFHTTPSessionManager AFManager;
/**
- download root directory.
/
@property (nonatomic, copy) NSString downloadDirectory;
/**
- fileManager to manage download files
/
@property (nonatomic, strong) NSFileManager fileManager;
/*
- the models for waiting for download, the elements should be FKDownloadModel and it’s subClasses
/
@property (nonatomic, strong) NSMutableArray <__kindof FKNetworkingDownloadModel > *waitingModels;
/*
- the models whose being downloaded, the elements should be FKDownloadModel and it’s subClasses
/
@property (nonatomic, strong) NSMutableArray <__kindof FKNetworkingDownloadModel > *downloadingModels;
/*
- key-values dictionary of the downloadModels, format as ‘
‘ to make constraints - used to find a downloadModel from this container,
- when the program will terminate, container will be clear
/
@property (nonatomic, strong) NSMutableDictionary <NSString , __kindof FKNetworkingDownloadModel > downloadModelsDict;
@end
NSInteger const fk_timeInterval = 5;
@implementation FKNetworkingManager
#pragma mark - download methods
-(void)fk_startDownloadWithDownloadModel:(FKNetworkingDownloadModel )downloadModel
progress:(void (^)(FKNetworkingDownloadModel _Nonnull))progress
completionHandler:(void (^)(FKNetworkingDownloadModel _Nonnull, NSError _Nullable))completionHandler{
NSString *fileName = [downloadModel.fileName componentsSeparatedByString:@"."].firstObject;
downloadModel.fileDirectory = [self.downloadDirectory stringByAppendingPathComponent:fileName];
downloadModel.filePath = [[self.downloadDirectory stringByAppendingPathComponent:fileName] stringByAppendingPathComponent:downloadModel.fileName];
downloadModel.plistFilePath = [downloadModel.fileDirectory stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.plist", fileName]];
if (![self canBeStartDownloadTaskWithDownloadModel:downloadModel]) return;
downloadModel.resumeData = [NSData dataWithContentsOfFile:downloadModel.plistFilePath];
if (downloadModel.resumeData.length == 0) {
NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:downloadModel.resourceURLString]];
downloadModel.downloadTask = [self.AFManager downloadTaskWithRequest:request progress:^(NSProgress * _Nonnull downloadProgress) {
[self setValuesForDownloadModel:downloadModel withProgress:downloadProgress.fractionCompleted];
progress(downloadModel);
} destination:^NSURL * _Nonnull(NSURL * _Nonnull targetPath, NSURLResponse * _Nonnull response) {
return [NSURL fileURLWithPath:downloadModel.filePath];
} completionHandler:^(NSURLResponse * _Nonnull response, NSURL * _Nullable filePath, NSError * _Nullable error) {
if (error) {
[self fk_cancelDownloadTaskWithDownloadModel:downloadModel];
completionHandler(downloadModel, error);
}else{
[self.downloadModelsDict removeObjectForKey:downloadModel.resourceURLString];
completionHandler(downloadModel, nil);
[self deletePlistFileWithDownloadModel:downloadModel];
}
}];
}else{
downloadModel.progressModel.totalBytesWritten = [self getResumeByteWithDownloadModel:downloadModel];
downloadModel.downloadTask = [self.AFManager downloadTaskWithResumeData:downloadModel.resumeData progress:^(NSProgress * _Nonnull downloadProgress) {
[self setValuesForDownloadModel:downloadModel withProgress:[self.AFManager downloadProgressForTask:downloadModel.downloadTask].fractionCompleted];
progress(downloadModel);
} destination:^NSURL * _Nonnull(NSURL * _Nonnull targetPath, NSURLResponse * _Nonnull response) {
return [NSURL fileURLWithPath:downloadModel.filePath];
} completionHandler:^(NSURLResponse * _Nonnull response, NSURL * _Nullable filePath, NSError * _Nullable error) {
if (error) {
[self fk_cancelDownloadTaskWithDownloadModel:downloadModel];
completionHandler(downloadModel, error);
}else{
[self.downloadModelsDict removeObjectForKey:downloadModel.resourceURLString];
completionHandler(downloadModel, nil);
[self deletePlistFileWithDownloadModel:downloadModel];
}
}];
}
if (![self.fileManager fileExistsAtPath:self.downloadDirectory]) {
[self.fileManager createDirectoryAtPath:self.downloadDirectory withIntermediateDirectories:YES attributes:nil error:nil];
}
[self createFolderAtPath:[self.downloadDirectory stringByAppendingPathComponent:fileName]];
[self fk_resumeDownloadWithDownloadModel:downloadModel];
}
-(void)fk_resumeDownloadWithDownloadModel:(FKNetworkingDownloadModel *)downloadModel{
if (downloadModel.downloadTask) {
downloadModel.downloadDate = [NSDate date];
[downloadModel.downloadTask resume];
self.downloadModelsDict[downloadModel.resourceURLString] = downloadModel;
[self.downloadingModels addObject:downloadModel];
}
}
-(void)fk_cancelDownloadTaskWithDownloadModel:(FKNetworkingDownloadModel )downloadModel{
if (!downloadModel) return;
NSURLSessionTaskState state = downloadModel.downloadTask.state;
if (state == NSURLSessionTaskStateRunning) {
[downloadModel.downloadTask cancelByProducingResumeData:^(NSData _Nullable resumeData) {
downloadModel.resumeData = resumeData;
@synchronized (self) {
BOOL isSuc = [downloadModel.resumeData writeToFile:downloadModel.plistFilePath atomically:YES];
[self saveTotalBytesExpectedToWriteWithDownloadModel:downloadModel];
if (isSuc) {
downloadModel.resumeData = nil;
[self.downloadModelsDict removeObjectForKey:downloadModel.resourceURLString];
[self.downloadingModels removeObject:downloadModel];
}
}
}];
}
}
-(void)fk_deleteDownloadedFileWithDownloadModel:(FKNetworkingDownloadModel *)downloadModel{
if ([self.fileManager fileExistsAtPath:downloadModel.fileDirectory]) {
[self.fileManager removeItemAtPath:downloadModel.fileDirectory error:nil];
}
}
-(void)fk_deleteAllDownloadedFiles{
if ([self.fileManager fileExistsAtPath:self.downloadDirectory]) {
[self.fileManager removeItemAtPath:self.downloadDirectory error:nil];
}
}
-(BOOL)fk_hasDownloadedFileWithDownloadModel:(FKNetworkingDownloadModel *)downloadModel{
if ([self.fileManager fileExistsAtPath:downloadModel.filePath]) {
NSLog(@”已下载的文件…”);
return YES;
}
return NO;
}
-(FKNetworkingDownloadModel )fk_getDownloadingModelWithURLString:(NSString )URLString{
return self.downloadModelsDict[URLString];
}
-(FKNetworkingProgressModel )fk_getDownloadProgressModelWithDownloadModel:(FKNetworkingDownloadModel )downloadModel{
FKNetworkingProgressModel *progressModel = downloadModel.progressModel;
progressModel.downloadProgress = [self.AFManager downloadProgressForTask:downloadModel.downloadTask].fractionCompleted;
return progressModel;
}
#pragma mark - private methods
-(BOOL)canBeStartDownloadTaskWithDownloadModel:(FKNetworkingDownloadModel *)downloadModel{
if (!downloadModel) return NO;
if (downloadModel.downloadTask && downloadModel.downloadTask.state == NSURLSessionTaskStateRunning) return NO;
if ([self fk_hasDownloadedFileWithDownloadModel:downloadModel]) return NO;
return YES;
}
-(void)setValuesForDownloadModel:(FKNetworkingDownloadModel )downloadModel withProgress:(double)progress{
NSTimeInterval interval = -1 [downloadModel.downloadDate timeIntervalSinceNow];
downloadModel.progressModel.totalBytesWritten = downloadModel.downloadTask.countOfBytesReceived;
downloadModel.progressModel.totalBytesExpectedToWrite = downloadModel.downloadTask.countOfBytesExpectedToReceive;
downloadModel.progressModel.downloadProgress = progress;
downloadModel.progressModel.downloadSpeed = (int64_t)((downloadModel.progressModel.totalBytesWritten - [self getResumeByteWithDownloadModel:downloadModel]) / interval);
if (downloadModel.progressModel.downloadSpeed != 0) {
int64_t remainingContentLength = downloadModel.progressModel.totalBytesExpectedToWrite - downloadModel.progressModel.totalBytesWritten;
int currentLeftTime = (int)(remainingContentLength / downloadModel.progressModel.downloadSpeed);
downloadModel.progressModel.downloadLeft = currentLeftTime;
}
}
-(int64_t)getResumeByteWithDownloadModel:(FKNetworkingDownloadModel )downloadModel{
int64_t resumeBytes = 0;
NSDictionary dict = [NSDictionary dictionaryWithContentsOfFile:downloadModel.plistFilePath];
if (dict) {
resumeBytes = [dict[@”NSURLSessionResumeBytesReceived”] longLongValue];
}
return resumeBytes;
}
-(NSString )getTmpFileNameWithDownloadModel:(FKNetworkingDownloadModel )downloadModel{
NSString fileName = nil;
NSDictionary dict = [NSDictionary dictionaryWithContentsOfFile:downloadModel.plistFilePath];
if (dict) {
fileName = dict[@”NSURLSessionResumeInfoTempFileName”];
}
return fileName;
}
-(void)createFolderAtPath:(NSString *)path{
if ([self.fileManager fileExistsAtPath:path]) return;
[self.fileManager createDirectoryAtPath:path withIntermediateDirectories:YES attributes:nil error:nil];
}
-(void)deletePlistFileWithDownloadModel:(FKNetworkingDownloadModel *)downloadModel{
if (downloadModel.downloadTask.countOfBytesReceived == downloadModel.downloadTask.countOfBytesExpectedToReceive) {
[self.fileManager removeItemAtPath:downloadModel.plistFilePath error:nil];
[self removeTotalBytesExpectedToWriteWhenDownloadFinishedWithDownloadModel:downloadModel];
}
}
-(NSString *)managerPlistFilePath{
return [self.downloadDirectory stringByAppendingPathComponent:@”ForKidManager.plist”];
}
-(nullable NSMutableDictionary
NSMutableDictionary dict = [NSMutableDictionary dictionaryWithContentsOfFile:[self managerPlistFilePath]];
return dict;
}
-(void)saveTotalBytesExpectedToWriteWithDownloadModel:(FKNetworkingDownloadModel )downloadModel{
NSMutableDictionary <NSString , NSString > dict = [self managerPlistDict];
[dict setValue:[NSString stringWithFormat:@”%lld”, downloadModel.downloadTask.countOfBytesExpectedToReceive] forKey:downloadModel.resourceURLString];
[dict writeToFile:[self managerPlistFilePath] atomically:YES];
}
-(void)removeTotalBytesExpectedToWriteWhenDownloadFinishedWithDownloadModel:(FKNetworkingDownloadModel )downloadModel{
NSMutableDictionary <NSString , NSString > dict = [self managerPlistDict];
[dict removeObjectForKey:downloadModel.resourceURLString];
[dict writeToFile:[self managerPlistFilePath] atomically:YES];
}
#pragma mark - share instance
+(FKNetworkingManager )shareManager{
static FKNetworkingManager manager = nil;
static dispatch_once_t sigletonOnceToken;
dispatch_once(&sigletonOnceToken, ^{
manager = [[self alloc] init];
});
return manager;
}
(instancetype)init{
self = [super init];
if (self) {_AFManager = [[AFHTTPSessionManager alloc]init]; _AFManager.requestSerializer.timeoutInterval = 5; _AFManager.requestSerializer.cachePolicy = NSURLRequestReloadIgnoringLocalCacheData;//NSURLRequestUseProtocolCachePolicy; NSSet *typeSet = [NSSet setWithObjects:@"application/json", @"text/plain", @"text/javascript", @"text/json", @"text/html", nil]; _AFManager.responseSerializer.acceptableContentTypes = typeSet; _AFManager.securityPolicy.allowInvalidCertificates = YES; _maxDownloadCount = 1; _resumeTaskFIFO = YES; _batchDownload = NO; _fileManager = [NSFileManager defaultManager]; _waitingModels = [[NSMutableArray alloc] initWithCapacity:1]; _downloadingModels = [[NSMutableArray alloc] initWithCapacity:1]; _downloadModelsDict = [[NSMutableDictionary alloc] initWithCapacity:1]; _downloadDirectory = [NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES).firstObject stringByAppendingPathComponent:FKNetworkingManagerFileName]; [_fileManager createDirectoryAtPath:_downloadDirectory withIntermediateDirectories:YES attributes:nil error:nil]; NSDictionary <NSString *, NSString *> *plistDict = [[NSDictionary alloc] init]; NSString *managerPlistFilePath = [_downloadDirectory stringByAppendingPathComponent:@"ForKidManager.plist"]; [plistDict writeToFile:managerPlistFilePath atomically:YES];
}
return self;
}
@end
`
如有问题或者BUG 欢迎指正
前段时间由于工作上的事情耽搁了,所以很多朋友问的Demo迟迟没有上传上去,这段时间我会完善当前的功能整理下传上去,简要的修复一点小bug和完善一下上传和POST方法以及网络请求管理。
未完待续…