TYDownloadSessionManager.m 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730
  1. //
  2. // TYDownloadSessionManager.m
  3. // TYDownloadManagerDemo
  4. //
  5. // Created by tany on 16/6/12.
  6. // Copyright © 2016年 tany. All rights reserved.
  7. //
  8. #import "TYDownloadSessionManager.h"
  9. #import <CommonCrypto/CommonDigest.h>
  10. #import <UIKit/UIKit.h>
  11. #import "NSURLSession+TYCorrectedResumeData.h"
  12. /**
  13. * 下载模型
  14. */
  15. @interface TYDownloadModel ()
  16. // >>>>>>>>>>>>>>>>>>>>>>>>>> task info
  17. // 下载状态
  18. @property (nonatomic, assign) TYDownloadState state;
  19. // 下载任务
  20. @property (nonatomic, strong) NSURLSessionDownloadTask *task;
  21. // 文件流
  22. @property (nonatomic, strong) NSOutputStream *stream;
  23. // 下载文件路径,下载完成后有值,把它移动到你的目录
  24. @property (nonatomic, strong) NSString *filePath;
  25. // 下载时间
  26. @property (nonatomic, strong) NSDate *downloadDate;
  27. // 断点续传需要设置这个数据
  28. @property (nonatomic, strong) NSData *resumeData;
  29. // 手动取消当做暂停
  30. @property (nonatomic, assign) BOOL manualCancle;
  31. @end
  32. /**
  33. * 下载进度
  34. */
  35. @interface TYDownloadProgress ()
  36. // 续传大小
  37. @property (nonatomic, assign) int64_t resumeBytesWritten;
  38. // 这次写入的数量
  39. @property (nonatomic, assign) int64_t bytesWritten;
  40. // 已下载的数量
  41. @property (nonatomic, assign) int64_t totalBytesWritten;
  42. // 文件的总大小
  43. @property (nonatomic, assign) int64_t totalBytesExpectedToWrite;
  44. // 下载进度
  45. @property (nonatomic, assign) float progress;
  46. // 下载速度
  47. @property (nonatomic, assign) float speed;
  48. // 下载剩余时间
  49. @property (nonatomic, assign) int remainingTime;
  50. @end
  51. @interface TYDownloadSessionManager ()
  52. // >>>>>>>>>>>>>>>>>>>>>>>>>> file info
  53. // 文件管理
  54. @property (nonatomic, strong) NSFileManager *fileManager;
  55. // 缓存文件目录
  56. @property (nonatomic, strong) NSString *downloadDirectory;
  57. // >>>>>>>>>>>>>>>>>>>>>>>>>> session info
  58. // 下载seesion会话
  59. @property (nonatomic, strong) NSURLSession *session;
  60. // 下载模型字典 key = url, value = model
  61. @property (nonatomic, strong) NSMutableDictionary *downloadingModelDic;
  62. // 下载中的模型
  63. @property (nonatomic, strong) NSMutableArray *waitingDownloadModels;
  64. // 等待中的模型
  65. @property (nonatomic, strong) NSMutableArray *downloadingModels;
  66. // 回调代理的队列
  67. @property (strong, nonatomic) NSOperationQueue *queue;
  68. @end
  69. #define IS_IOS8ORLATER ([[[UIDevice currentDevice] systemVersion] floatValue] >= 8)
  70. #define IS_IOS10ORLATER ([[[UIDevice currentDevice] systemVersion] floatValue] >= 10)
  71. #define IS_IOS12ORLATER ([[[UIDevice currentDevice] systemVersion] floatValue] >= 12)
  72. @implementation TYDownloadSessionManager
  73. + (TYDownloadSessionManager *)manager
  74. {
  75. static id sharedInstance = nil;
  76. static dispatch_once_t onceToken;
  77. dispatch_once(&onceToken, ^{
  78. sharedInstance = [[self alloc] init];
  79. });
  80. return sharedInstance;
  81. }
  82. - (instancetype)init
  83. {
  84. if (self = [super init]) {
  85. _backgroundConfigure = @"TYDownloadSessionManager.backgroundConfigure";
  86. _maxDownloadCount = 1;
  87. _resumeDownloadFIFO = YES;
  88. _isBatchDownload = NO;
  89. }
  90. return self;
  91. }
  92. - (void)configureBackroundSession
  93. {
  94. if (!_backgroundConfigure) {
  95. return;
  96. }
  97. [self session];
  98. }
  99. #pragma mark - getter
  100. - (NSFileManager *)fileManager
  101. {
  102. if (!_fileManager) {
  103. _fileManager = [[NSFileManager alloc]init];
  104. }
  105. return _fileManager;
  106. }
  107. - (NSURLSession *)session
  108. {
  109. if (!_session) {
  110. if (_backgroundConfigure) {
  111. if (IS_IOS8ORLATER) {
  112. NSURLSessionConfiguration *configure = [NSURLSessionConfiguration backgroundSessionConfigurationWithIdentifier:_backgroundConfigure];
  113. _session = [NSURLSession sessionWithConfiguration:configure delegate:self delegateQueue:self.queue];
  114. }else{
  115. _session = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration backgroundSessionConfiguration:_backgroundConfigure]delegate:self delegateQueue:self.queue];
  116. }
  117. }else {
  118. _session = [NSURLSession sessionWithConfiguration:[NSURLSessionConfiguration defaultSessionConfiguration] delegate:self delegateQueue:self.queue];
  119. }
  120. }
  121. return _session;
  122. }
  123. - (NSOperationQueue *)queue
  124. {
  125. if (!_queue) {
  126. _queue = [[NSOperationQueue alloc]init];
  127. _queue.maxConcurrentOperationCount = 1;
  128. }
  129. return _queue;
  130. }
  131. - (NSString *)downloadDirectory
  132. {
  133. if (!_downloadDirectory) {
  134. _downloadDirectory = [[NSSearchPathForDirectoriesInDomains(NSCachesDirectory, NSUserDomainMask, YES) lastObject] stringByAppendingPathComponent:@"TYDownloadCache"];
  135. [self createDirectory:_downloadDirectory];
  136. }
  137. return _downloadDirectory;
  138. }
  139. - (NSMutableDictionary *)downloadingModelDic
  140. {
  141. if (!_downloadingModelDic) {
  142. _downloadingModelDic = [NSMutableDictionary dictionary];
  143. }
  144. return _downloadingModelDic;
  145. }
  146. - (NSMutableArray *)waitingDownloadModels
  147. {
  148. if (!_waitingDownloadModels) {
  149. _waitingDownloadModels = [NSMutableArray array];
  150. }
  151. return _waitingDownloadModels;
  152. }
  153. - (NSMutableArray *)downloadingModels
  154. {
  155. if (!_downloadingModels) {
  156. _downloadingModels = [NSMutableArray array];
  157. }
  158. return _downloadingModels;
  159. }
  160. #pragma mark - downlaod
  161. // 开始下载
  162. - (TYDownloadModel *)startDownloadURLString:(NSString *)URLString toDestinationPath:(NSString *)destinationPath progress:(TYDownloadProgressBlock)progress state:(TYDownloadStateBlock)state
  163. {
  164. // 验证下载地址
  165. if (!URLString) {
  166. NSLog(@"dwonloadURL can't nil");
  167. return nil;
  168. }
  169. TYDownloadModel *downloadModel = [self downLoadingModelForURLString:URLString];
  170. if (!downloadModel || ![downloadModel.filePath isEqualToString:destinationPath]) {
  171. downloadModel = [[TYDownloadModel alloc]initWithURLString:URLString filePath:destinationPath];
  172. }
  173. [self startWithDownloadModel:downloadModel progress:progress state:state];
  174. return downloadModel;
  175. }
  176. - (void)startWithDownloadModel:(TYDownloadModel *)downloadModel progress:(TYDownloadProgressBlock)progress state:(TYDownloadStateBlock)state
  177. {
  178. downloadModel.progressBlock = progress;
  179. downloadModel.stateBlock = state;
  180. [self startWithDownloadModel:downloadModel];
  181. }
  182. - (void)startWithDownloadModel:(TYDownloadModel *)downloadModel
  183. {
  184. if (!downloadModel) {
  185. return;
  186. }
  187. if (downloadModel.state == TYDownloadStateReadying) {
  188. [self downloadModel:downloadModel didChangeState:TYDownloadStateReadying filePath:nil error:nil];
  189. return;
  190. }
  191. // 验证是否存在
  192. if (downloadModel.task && downloadModel.task.state == NSURLSessionTaskStateRunning) {
  193. downloadModel.state = TYDownloadStateRunning;
  194. [self downloadModel:downloadModel didChangeState:TYDownloadStateRunning filePath:nil error:nil];
  195. return;
  196. }
  197. // 后台下载设置
  198. [self configirebackgroundSessionTasksWithDownloadModel:downloadModel];
  199. [self resumeWithDownloadModel:downloadModel];
  200. }
  201. // 恢复下载
  202. - (void)resumeWithDownloadModel:(TYDownloadModel *)downloadModel
  203. {
  204. if (!downloadModel) {
  205. return;
  206. }
  207. if (![self canResumeDownlaodModel:downloadModel]) {
  208. return;
  209. }
  210. // 如果task 不存在 或者 取消了
  211. if (!downloadModel.task || downloadModel.task.state == NSURLSessionTaskStateCanceling) {
  212. NSData *resumeData = [self resumeDataFromFileWithDownloadModel:downloadModel];
  213. if ([self isValideResumeData:resumeData]) {
  214. if (IS_IOS10ORLATER && !IS_IOS12ORLATER) {
  215. downloadModel.task = [self.session downloadTaskWithCorrectResumeData:resumeData];
  216. }else {
  217. downloadModel.task = [self.session downloadTaskWithResumeData:resumeData];
  218. }
  219. }else {
  220. NSURLRequest *request = [NSURLRequest requestWithURL:[NSURL URLWithString:downloadModel.downloadURL]];
  221. downloadModel.task = [self.session downloadTaskWithRequest:request];
  222. }
  223. downloadModel.task.taskDescription = downloadModel.downloadURL;
  224. downloadModel.downloadDate = [NSDate date];
  225. }
  226. if (!downloadModel.downloadDate) {
  227. downloadModel.downloadDate = [NSDate date];
  228. }
  229. if (![self.downloadingModelDic objectForKey:downloadModel.downloadURL]) {
  230. self.downloadingModelDic[downloadModel.downloadURL] = downloadModel;
  231. }
  232. [downloadModel.task resume];
  233. downloadModel.state = TYDownloadStateRunning;
  234. [self downloadModel:downloadModel didChangeState:TYDownloadStateRunning filePath:nil error:nil];
  235. }
  236. - (BOOL)isValideResumeData:(NSData *)resumeData
  237. {
  238. if (!resumeData || resumeData.length == 0) {
  239. return NO;
  240. }
  241. return YES;
  242. }
  243. // 暂停下载
  244. - (void)suspendWithDownloadModel:(TYDownloadModel *)downloadModel
  245. {
  246. if (!downloadModel.manualCancle) {
  247. downloadModel.manualCancle = YES;
  248. [self cancleWithDownloadModel:downloadModel clearResumeData:NO];
  249. }
  250. }
  251. // 取消下载
  252. - (void)cancleWithDownloadModel:(TYDownloadModel *)downloadModel
  253. {
  254. if (downloadModel.state != TYDownloadStateCompleted && downloadModel.state != TYDownloadStateFailed){
  255. [self cancleWithDownloadModel:downloadModel clearResumeData:NO];
  256. }
  257. }
  258. // 删除下载
  259. - (void)deleteFileWithDownloadModel:(TYDownloadModel *)downloadModel
  260. {
  261. if (!downloadModel || !downloadModel.filePath) {
  262. return;
  263. }
  264. [self cancleWithDownloadModel:downloadModel clearResumeData:YES];
  265. [self deleteFileIfExist:downloadModel.filePath];
  266. }
  267. - (void)deleteAllFileWithDownloadDirectory:(NSString *)downloadDirectory
  268. {
  269. if (!downloadDirectory) {
  270. downloadDirectory = self.downloadDirectory;
  271. }
  272. for (TYDownloadModel *downloadModel in [self.downloadingModelDic allValues]) {
  273. if ([downloadModel.downloadDirectory isEqualToString:downloadDirectory]) {
  274. [self cancleWithDownloadModel:downloadModel clearResumeData:YES];
  275. }
  276. }
  277. // 删除沙盒中所有资源
  278. [self.fileManager removeItemAtPath:downloadDirectory error:nil];
  279. }
  280. // 取消下载 是否删除resumeData
  281. - (void)cancleWithDownloadModel:(TYDownloadModel *)downloadModel clearResumeData:(BOOL)clearResumeData
  282. {
  283. if (!downloadModel.task && downloadModel.state == TYDownloadStateReadying) {
  284. [self removeDownLoadingModelForURLString:downloadModel.downloadURL];
  285. @synchronized (self) {
  286. [self.waitingDownloadModels removeObject:downloadModel];
  287. }
  288. downloadModel.state = TYDownloadStateNone;
  289. [self downloadModel:downloadModel didChangeState:TYDownloadStateNone filePath:nil error:nil];
  290. return;
  291. }
  292. if (clearResumeData) {
  293. downloadModel.state = TYDownloadStateNone;
  294. downloadModel.resumeData = nil;
  295. [self deleteFileIfExist:[self resumeDataPathWithDownloadURL:downloadModel.downloadURL]];
  296. [downloadModel.task cancel];
  297. }else {
  298. [(NSURLSessionDownloadTask *)downloadModel.task cancelByProducingResumeData:^(NSData *resumeData){
  299. }];
  300. }
  301. }
  302. - (void)willResumeNextWithDowloadModel:(TYDownloadModel *)downloadModel
  303. {
  304. if (_isBatchDownload) {
  305. return;
  306. }
  307. @synchronized (self) {
  308. [self.downloadingModels removeObject:downloadModel];
  309. // 还有未下载的
  310. if (self.waitingDownloadModels.count > 0) {
  311. [self resumeWithDownloadModel:_resumeDownloadFIFO ? self.waitingDownloadModels.firstObject:self.waitingDownloadModels.lastObject];
  312. }
  313. }
  314. }
  315. - (BOOL)canResumeDownlaodModel:(TYDownloadModel *)downloadModel
  316. {
  317. if (_isBatchDownload) {
  318. return YES;
  319. }
  320. @synchronized (self) {
  321. if (self.downloadingModels.count >= _maxDownloadCount ) {
  322. if ([self.waitingDownloadModels indexOfObject:downloadModel] == NSNotFound) {
  323. [self.waitingDownloadModels addObject:downloadModel];
  324. self.downloadingModelDic[downloadModel.downloadURL] = downloadModel;
  325. }
  326. downloadModel.state = TYDownloadStateReadying;
  327. [self downloadModel:downloadModel didChangeState:TYDownloadStateReadying filePath:nil error:nil];
  328. return NO;
  329. }
  330. if ([self.waitingDownloadModels indexOfObject:downloadModel] != NSNotFound) {
  331. [self.waitingDownloadModels removeObject:downloadModel];
  332. }
  333. if ([self.downloadingModels indexOfObject:downloadModel] == NSNotFound) {
  334. [self.downloadingModels addObject:downloadModel];
  335. }
  336. return YES;
  337. }
  338. }
  339. #pragma mark - configire background task
  340. // 配置后台后台下载session
  341. - (void)configirebackgroundSessionTasksWithDownloadModel:(TYDownloadModel *)downloadModel
  342. {
  343. if (!_backgroundConfigure) {
  344. return ;
  345. }
  346. NSURLSessionDownloadTask *task = [self backgroundSessionTasksWithDownloadModel:downloadModel];
  347. if (!task) {
  348. return;
  349. }
  350. downloadModel.task = task;
  351. if (task.state == NSURLSessionTaskStateRunning) {
  352. [task suspend];
  353. }
  354. }
  355. - (NSURLSessionDownloadTask *)backgroundSessionTasksWithDownloadModel:(TYDownloadModel *)downloadModel
  356. {
  357. NSArray *tasks = [self sessionDownloadTasks];
  358. for (NSURLSessionDownloadTask *task in tasks) {
  359. if (task.state == NSURLSessionTaskStateRunning || task.state == NSURLSessionTaskStateSuspended) {
  360. if ([downloadModel.downloadURL isEqualToString:task.taskDescription]) {
  361. return task;
  362. }
  363. }
  364. }
  365. return nil;
  366. }
  367. // 获取所以的后台下载session
  368. - (NSArray *)sessionDownloadTasks
  369. {
  370. __block NSArray *tasks = nil;
  371. dispatch_semaphore_t semaphore = dispatch_semaphore_create(0);
  372. [self.session getTasksWithCompletionHandler:^(NSArray *dataTasks, NSArray *uploadTasks, NSArray *downloadTasks) {
  373. tasks = downloadTasks;
  374. dispatch_semaphore_signal(semaphore);
  375. }];
  376. dispatch_semaphore_wait(semaphore, DISPATCH_TIME_FOREVER);
  377. return tasks;
  378. }
  379. #pragma mark - public
  380. // 获取下载模型
  381. - (TYDownloadModel *)downLoadingModelForURLString:(NSString *)URLString
  382. {
  383. return [self.downloadingModelDic objectForKey:URLString];
  384. }
  385. // 是否已经下载
  386. - (BOOL)isDownloadCompletedWithDownloadModel:(TYDownloadModel *)downloadModel
  387. {
  388. return [self.fileManager fileExistsAtPath:downloadModel.filePath];
  389. }
  390. // 取消所有后台
  391. - (void)cancleAllBackgroundSessionTasks
  392. {
  393. if (!_backgroundConfigure) {
  394. return;
  395. }
  396. for (NSURLSessionDownloadTask *task in [self sessionDownloadTasks]) {
  397. [task cancelByProducingResumeData:^(NSData * resumeData) {
  398. }];
  399. }
  400. }
  401. #pragma mark - private
  402. - (void)downloadModel:(TYDownloadModel *)downloadModel didChangeState:(TYDownloadState)state filePath:(NSString *)filePath error:(NSError *)error
  403. {
  404. if (_delegate && [_delegate respondsToSelector:@selector(downloadModel:didChangeState:filePath:error:)]) {
  405. [_delegate downloadModel:downloadModel didChangeState:state filePath:filePath error:error];
  406. }
  407. if (downloadModel.stateBlock) {
  408. downloadModel.stateBlock(state,filePath,error);
  409. }
  410. }
  411. - (void)downloadModel:(TYDownloadModel *)downloadModel updateProgress:(TYDownloadProgress *)progress
  412. {
  413. if (_delegate && [_delegate respondsToSelector:@selector(downloadModel:didUpdateProgress:)]) {
  414. [_delegate downloadModel:downloadModel didUpdateProgress:progress];
  415. }
  416. if (downloadModel.progressBlock) {
  417. downloadModel.progressBlock(progress);
  418. }
  419. }
  420. - (void)removeDownLoadingModelForURLString:(NSString *)URLString
  421. {
  422. [self.downloadingModelDic removeObjectForKey:URLString];
  423. }
  424. // 获取resumeData
  425. - (NSData *)resumeDataFromFileWithDownloadModel:(TYDownloadModel *)downloadModel
  426. {
  427. if (downloadModel.resumeData) {
  428. return downloadModel.resumeData;
  429. }
  430. NSString *resumeDataPath = [self resumeDataPathWithDownloadURL:downloadModel.downloadURL];
  431. if ([_fileManager fileExistsAtPath:resumeDataPath]) {
  432. NSData *resumeData = [NSData dataWithContentsOfFile:resumeDataPath];
  433. return resumeData;
  434. }
  435. return nil;
  436. }
  437. // resumeData 路径
  438. - (NSString *)resumeDataPathWithDownloadURL:(NSString *)downloadURL
  439. {
  440. NSString *resumeFileName = [[self class] md5:downloadURL];
  441. return [self.downloadDirectory stringByAppendingPathComponent:resumeFileName];
  442. }
  443. + (NSString *)md5:(NSString *)str
  444. {
  445. const char *cStr = [str UTF8String];
  446. if (cStr == NULL) {
  447. cStr = "";
  448. }
  449. unsigned char result[CC_MD5_DIGEST_LENGTH];
  450. CC_MD5( cStr, (CC_LONG)strlen(cStr), result );
  451. return [NSString stringWithFormat:
  452. @"%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X%02X",
  453. result[0], result[1], result[2], result[3],
  454. result[4], result[5], result[6], result[7],
  455. result[8], result[9], result[10], result[11],
  456. result[12], result[13], result[14], result[15]
  457. ];
  458. }
  459. // 创建缓存目录文件
  460. - (void)createDirectory:(NSString *)directory
  461. {
  462. if (![self.fileManager fileExistsAtPath:directory]) {
  463. [self.fileManager createDirectoryAtPath:directory withIntermediateDirectories:YES attributes:nil error:NULL];
  464. }
  465. }
  466. - (void)moveFileAtURL:(NSURL *)srcURL toPath:(NSString *)dstPath
  467. {
  468. if (!dstPath) {
  469. NSLog(@"error filePath is nil!");
  470. return;
  471. }
  472. NSError *error = nil;
  473. if ([self.fileManager fileExistsAtPath:dstPath] ) {
  474. [self.fileManager removeItemAtPath:dstPath error:&error];
  475. if (error) {
  476. NSLog(@"removeItem error %@",error);
  477. }
  478. }
  479. NSURL *dstURL = [NSURL fileURLWithPath:dstPath];
  480. [self.fileManager moveItemAtURL:srcURL toURL:dstURL error:&error];
  481. if (error){
  482. NSLog(@"moveItem error:%@",error);
  483. }
  484. }
  485. - (void)deleteFileIfExist:(NSString *)filePath
  486. {
  487. if ([self.fileManager fileExistsAtPath:filePath] ) {
  488. NSError *error = nil;
  489. [self.fileManager removeItemAtPath:filePath error:&error];
  490. if (error) {
  491. NSLog(@"emoveItem error %@",error);
  492. }
  493. }
  494. }
  495. #pragma mark - NSURLSessionDownloadDelegate
  496. // 恢复下载
  497. - (void)URLSession:(NSURLSession *)session downloadTask:(NSURLSessionDownloadTask *)downloadTask didResumeAtOffset:(int64_t)fileOffset expectedTotalBytes:(int64_t)expectedTotalBytes
  498. {
  499. TYDownloadModel *downloadModel = [self downLoadingModelForURLString:downloadTask.taskDescription];
  500. if (!downloadModel || downloadModel.state == TYDownloadStateSuspended) {
  501. return;
  502. }
  503. downloadModel.progress.resumeBytesWritten = fileOffset;
  504. }
  505. // 监听文件下载进度
  506. - (void)URLSession:(NSURLSession *)session downloadTask:(NSURLSessionDownloadTask *)downloadTask
  507. didWriteData:(int64_t)bytesWritten
  508. totalBytesWritten:(int64_t)totalBytesWritten
  509. totalBytesExpectedToWrite:(int64_t)totalBytesExpectedToWrite
  510. {
  511. TYDownloadModel *downloadModel = [self downLoadingModelForURLString:downloadTask.taskDescription];
  512. if (!downloadModel || downloadModel.state == TYDownloadStateSuspended) {
  513. return;
  514. }
  515. float progress = (double)totalBytesWritten/totalBytesExpectedToWrite;
  516. int64_t resumeBytesWritten = downloadModel.progress.resumeBytesWritten;
  517. NSTimeInterval downloadTime = -1 * [downloadModel.downloadDate timeIntervalSinceNow];
  518. float speed = (totalBytesWritten - resumeBytesWritten) / downloadTime;
  519. int64_t remainingContentLength = totalBytesExpectedToWrite - totalBytesWritten;
  520. int remainingTime = ceilf(remainingContentLength / speed);
  521. downloadModel.progress.bytesWritten = bytesWritten;
  522. downloadModel.progress.totalBytesWritten = totalBytesWritten;
  523. downloadModel.progress.totalBytesExpectedToWrite = totalBytesExpectedToWrite;
  524. downloadModel.progress.progress = progress;
  525. downloadModel.progress.speed = speed;
  526. downloadModel.progress.remainingTime = remainingTime;
  527. dispatch_async(dispatch_get_main_queue(), ^(){
  528. [self downloadModel:downloadModel updateProgress:downloadModel.progress];
  529. });
  530. }
  531. // 下载成功
  532. - (void)URLSession:(NSURLSession *)session downloadTask:(NSURLSessionDownloadTask *)downloadTask
  533. didFinishDownloadingToURL:(NSURL *)location
  534. {
  535. TYDownloadModel *downloadModel = [self downLoadingModelForURLString:downloadTask.taskDescription];
  536. if (!downloadModel && _backgroundSessionDownloadCompleteBlock) {
  537. NSString *filePath = _backgroundSessionDownloadCompleteBlock(downloadTask.taskDescription);
  538. // 移动文件到下载目录
  539. [self createDirectory:filePath.stringByDeletingLastPathComponent];
  540. [self moveFileAtURL:location toPath:filePath];
  541. return;
  542. }
  543. if (location) {
  544. // 移动文件到下载目录
  545. [self createDirectory:downloadModel.downloadDirectory];
  546. [self moveFileAtURL:location toPath:downloadModel.filePath];
  547. }
  548. }
  549. // 下载完成
  550. - (void)URLSession:(NSURLSession *)session task:(NSURLSessionTask *)task didCompleteWithError:(NSError *)error
  551. {
  552. TYDownloadModel *downloadModel = [self downLoadingModelForURLString:task.taskDescription];
  553. if (!downloadModel) {
  554. NSData *resumeData = error ? [error.userInfo objectForKey:NSURLSessionDownloadTaskResumeData]:nil;
  555. if (resumeData) {
  556. [self createDirectory:_downloadDirectory];
  557. [resumeData writeToFile:[self resumeDataPathWithDownloadURL:task.taskDescription] atomically:YES];
  558. }else {
  559. [self deleteFileIfExist:[self resumeDataPathWithDownloadURL:task.taskDescription]];
  560. }
  561. return;
  562. }
  563. NSData *resumeData = nil;
  564. if (error) {
  565. resumeData = [error.userInfo objectForKey:NSURLSessionDownloadTaskResumeData];
  566. }
  567. // 缓存resumeData
  568. if (resumeData) {
  569. downloadModel.resumeData = resumeData;
  570. [self createDirectory:_downloadDirectory];
  571. [downloadModel.resumeData writeToFile:[self resumeDataPathWithDownloadURL:downloadModel.downloadURL] atomically:YES];
  572. }else {
  573. downloadModel.resumeData = nil;
  574. [self deleteFileIfExist:[self resumeDataPathWithDownloadURL:downloadModel.downloadURL]];
  575. }
  576. downloadModel.progress.resumeBytesWritten = 0;
  577. downloadModel.task = nil;
  578. [self removeDownLoadingModelForURLString:downloadModel.downloadURL];
  579. if (downloadModel.manualCancle) {
  580. // 手动取消,当做暂停
  581. dispatch_async(dispatch_get_main_queue(), ^(){
  582. downloadModel.manualCancle = NO;
  583. downloadModel.state = TYDownloadStateSuspended;
  584. [self downloadModel:downloadModel didChangeState:TYDownloadStateSuspended filePath:nil error:nil];
  585. [self willResumeNextWithDowloadModel:downloadModel];
  586. });
  587. }else if (error){
  588. if (downloadModel.state == TYDownloadStateNone) {
  589. // 删除下载
  590. dispatch_async(dispatch_get_main_queue(), ^(){
  591. downloadModel.state = TYDownloadStateNone;
  592. [self downloadModel:downloadModel didChangeState:TYDownloadStateNone filePath:nil error:error];
  593. [self willResumeNextWithDowloadModel:downloadModel];
  594. });
  595. }else {
  596. // 下载失败
  597. dispatch_async(dispatch_get_main_queue(), ^(){
  598. downloadModel.state = TYDownloadStateFailed;
  599. [self downloadModel:downloadModel didChangeState:TYDownloadStateFailed filePath:nil error:error];
  600. [self willResumeNextWithDowloadModel:downloadModel];
  601. });
  602. }
  603. }else {
  604. // 下载完成
  605. dispatch_async(dispatch_get_main_queue(), ^(){
  606. downloadModel.state = TYDownloadStateCompleted;
  607. [self downloadModel:downloadModel didChangeState:TYDownloadStateCompleted filePath:downloadModel.filePath error:nil];
  608. [self willResumeNextWithDowloadModel:downloadModel];
  609. });
  610. }
  611. }
  612. // 后台session下载完成
  613. - (void)URLSessionDidFinishEventsForBackgroundURLSession:(NSURLSession *)session
  614. {
  615. if (self.backgroundSessionCompletionHandler) {
  616. self.backgroundSessionCompletionHandler();
  617. }
  618. }
  619. @end