boxDownloadFileManager.m 12 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364
  1. //
  2. // boxDownloadFileManager.m
  3. // 双子星云手机
  4. //
  5. // Created by xd h on 2024/5/24.
  6. //
  7. #import "boxDownloadFileManager.h"
  8. @implementation boxDownloadFileManager
  9. static boxDownloadFileManager * cur_boxDownloadFileShareInstance = nil;
  10. +(boxDownloadFileManager *)shareInstance;
  11. {
  12. static dispatch_once_t onceToken;
  13. dispatch_once(&onceToken, ^{
  14. cur_boxDownloadFileShareInstance = [[boxDownloadFileManager alloc] init];
  15. });
  16. return cur_boxDownloadFileShareInstance;
  17. }
  18. - (id)init
  19. {
  20. self = [super init];
  21. if (self) {
  22. //[self initManager];
  23. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(downloadTaskExeEnd:) name:SGDownloadTaskExeEnd object:nil];
  24. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(downloadTaskExeError:) name:SGDownloadTaskExeError object:nil];
  25. }
  26. return self;
  27. }
  28. //添加保存记录
  29. - (void)addBoxDownloadRecordFunBy:(NSArray*)arr complete:(custom_complete_B)complete{
  30. if(arr && arr.count >0 ){
  31. //1.排重处理
  32. [self getDataInDatabaseFun:NO complete:^(NSMutableArray * _Nonnull Arr) {
  33. [self filterDownloadingTaskFunWithNewTaskArr:arr complete:complete];
  34. }];
  35. }
  36. }
  37. - (void)filterDownloadingTaskFunWithNewTaskArr:(NSArray*)taskArr complete:(custom_complete_B)complete
  38. {
  39. NSMutableArray *addTaskArr = [NSMutableArray new];
  40. NSMutableArray * downloadingArr = [NSMutableArray new];
  41. if(_databaseArr && _databaseArr.count == 3){
  42. downloadingArr = _databaseArr[0];
  43. }
  44. for (ShareFileDataModel*model in taskArr) {
  45. BOOL canAdd = YES;
  46. if(downloadingArr.count > 0){
  47. for (ShareFileDataModel*downloadingModel in downloadingArr) {
  48. if([downloadingModel.fileUrl isEqualToString:model.fileUrl]){
  49. canAdd = NO;
  50. break;
  51. }
  52. }
  53. }
  54. if(canAdd){
  55. model.bg_tableName = share_box_download_tableName;
  56. [addTaskArr addObject:model];
  57. }
  58. }
  59. KWeakSelf
  60. //[ShareFileDataModel bg_saveOrUpdateArray:arr];
  61. [ShareFileDataModel bg_saveOrUpdateArrayAsync:addTaskArr complete:^(BOOL isSuccess) {
  62. [weakSelf beginDownloadFilesByUrls:addTaskArr];
  63. complete(isSuccess);
  64. }];
  65. }
  66. #pragma mark 读取数据库数据
  67. - (void)getDataInDatabaseFun:(BOOL)isReGet complete:(custom_complete_Arr)complete
  68. {
  69. if(_databaseArr && _databaseArr.count == 3 && !isReGet){
  70. complete(_databaseArr);
  71. return;
  72. }
  73. if(!_databaseArr)
  74. {
  75. _databaseArr = [NSMutableArray new];
  76. }
  77. //KWeakSelf
  78. //dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT,0), ^{
  79. // NSLock *lock = [NSLock new];
  80. // [lock lock];
  81. [ShareFileDataModel bg_findAsync:share_box_download_tableName limit:0 orderBy:nil desc:YES complete:^(NSArray * _Nullable array) {
  82. NSMutableArray *failArr = [NSMutableArray new];
  83. NSMutableArray *doneArr = [NSMutableArray new];
  84. NSMutableArray *otherArr = [NSMutableArray new];
  85. if(array && array.count >0){
  86. for (ShareFileDataModel * curModel in array) {
  87. if(curModel.downloadBoxStateType == downloadBoxStateFail){
  88. [failArr addObject:curModel];
  89. }
  90. else if(curModel.downloadBoxStateType == downloadBoxStateDone){
  91. [doneArr addObject:curModel];
  92. }
  93. else{
  94. [otherArr addObject:curModel];
  95. }
  96. }
  97. }
  98. NSLock *lock = [NSLock new];
  99. [lock lock];
  100. NSMutableArray *newArr = [NSMutableArray new];
  101. [newArr addObject:otherArr];
  102. [newArr addObject:doneArr];
  103. [newArr addObject:failArr];
  104. self->_databaseArr = newArr;
  105. [lock unlock];
  106. complete(self->_databaseArr);
  107. }];
  108. // [lock unlock];
  109. //});
  110. }
  111. - (void)handleDatabaseArrByDelete:(ShareFileDataModel*)delModel
  112. {
  113. if(_databaseArr && _databaseArr.count == 3 ){
  114. if(delModel.downloadBoxStateType == downloadBoxStateDone)
  115. {
  116. NSMutableArray *arr = _databaseArr[1];
  117. [self deteleModel:delModel inArr:arr];
  118. }
  119. else if(delModel.downloadBoxStateType == downloadBoxStateFail){
  120. NSMutableArray *arr = _databaseArr[2];
  121. [self deteleModel:delModel inArr:arr];
  122. }
  123. else{
  124. NSMutableArray *arr = _databaseArr[0];
  125. [self deteleModel:delModel inArr:arr];
  126. }
  127. }
  128. }
  129. #pragma mark 处理删除内存数据
  130. - (void)deteleModel:(ShareFileDataModel*)delModel inArr:(NSMutableArray*)delArr{
  131. for (ShareFileDataModel *dataModel in delArr) {
  132. if(delModel.bg_id.integerValue == dataModel.bg_id.integerValue){
  133. [delArr removeObject:dataModel];
  134. break;;
  135. }
  136. }
  137. }
  138. #pragma mark 删除本地数据库记录
  139. - (void)deleteBoxSaveFileRecordBy:(NSMutableArray *)delArr{
  140. NSMutableArray *curDelArr = [NSMutableArray arrayWithArray:delArr];
  141. for (ShareFileDataModel *dataModel in curDelArr) {
  142. //1.内存数据删除
  143. [self handleDatabaseArrByDelete:dataModel];
  144. //2.删除数据量
  145. NSMutableString* where = [[NSMutableString alloc] initWithString:@"where "];
  146. NSString *curStr = [NSString stringWithFormat:@"%@=%@ ",bg_sqlKey(@"bg_id"),bg_sqlValue(dataModel.bg_id)];
  147. [where appendString:curStr];
  148. [ShareFileDataModel bg_deleteAsync:share_box_download_tableName where:where complete:^(BOOL isSuccess) {
  149. if(isSuccess){
  150. }
  151. }];
  152. //3.取消下载
  153. NSString *urlString = dataModel.fileUrl;
  154. NSString *requestURLEncodedString = [urlString stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  155. [[SGDownloadManager shareManager] cancelDownloadWithUrl:requestURLEncodedString];
  156. }
  157. }
  158. #pragma mark 更新本地数据库记录
  159. - (void)updataBoxSaveDataInFailBy:(ShareFileDataModel*)dataModel
  160. {
  161. //1.修改数据库
  162. if(_databaseArr.count == 3){
  163. NSMutableArray *failArr = _databaseArr[2];
  164. NSMutableArray *ingArr = _databaseArr[0];
  165. for (ShareFileDataModel* model in failArr) {
  166. if (dataModel.bg_id.integerValue == model.bg_id.integerValue) {
  167. [failArr removeObject:model];
  168. [ingArr addObject:dataModel];
  169. break;
  170. }
  171. }
  172. }
  173. [dataModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  174. }];
  175. }
  176. #pragma mark 开始下载文件
  177. - (void)beginDownloadFilesByUrls:(NSArray *)urlArr{
  178. // NSMutableArray*arr = [[SGDownloadManager shareManager] getAllOperation];
  179. // if(arr && arr.count > 0){
  180. // HLog(@"有任务正在下载");
  181. // [[SGDownloadManager shareManager] startAllDownloadTask];
  182. // return;
  183. // }
  184. for (ShareFileDataModel *fileModel in urlArr) {
  185. NSString *urlString = fileModel.fileUrl;
  186. NSString *requestURLEncodedString = [urlString stringByAddingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  187. NSURL *curUrl = [NSURL URLWithString:requestURLEncodedString];
  188. // 创建下载任务
  189. // [[SGDownloadManager shareManager] downloadWithURL:curUrl
  190. // complete:^(NSDictionary *respose, NSError *error) {
  191. // HLog(@"%@---%@",respose, error);
  192. // }];
  193. //[SGDownloadManager shareManager].
  194. [[SGDownloadManager shareManager] downloadWithURL:curUrl fileType:1 fileSize:0 complete:^(NSDictionary *respose, NSError *error) {
  195. HLog(@"%@---%@",respose, error);
  196. }];
  197. //测试 每次只跑一个
  198. //break;
  199. }
  200. }
  201. //- (void)downloadTaskExeIng:(NSNotification *)notification
  202. //{
  203. // SGDownloadOperation *model = notification.userInfo.allValues.firstObject;
  204. // [self handldDownloadingDataBy:model];
  205. //}
  206. - (void)downloadTaskExeEnd:(NSNotification *)notification
  207. {
  208. SGDownloadOperation *model = notification.userInfo.allValues.firstObject;
  209. [self handldDownloadDoneDataBy:model];
  210. //HLog(@"下载成功");
  211. }
  212. - (void)downloadTaskExeError:(NSNotification *)notification
  213. {
  214. SGDownloadOperation *model = notification.userInfo.allValues.firstObject;
  215. [self handldDownloadFailDataBy:model];
  216. //HLog(@"下载失败");
  217. }
  218. #pragma mark 下载中数据回调处理
  219. - (void)handldDownloadingDataBy:(SGDownloadOperation*)model
  220. {
  221. //解码
  222. NSString * urlString = [model.url stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  223. if(_databaseArr && _databaseArr.count == 3 ){
  224. NSMutableArray *ingArr = _databaseArr[0];
  225. for (ShareFileDataModel*dataModel in ingArr) {
  226. //HLog(@"22222222 %@\n%@",urlString,dataModel.fileUrl);
  227. if([urlString isEqualToString:dataModel.fileUrl]){
  228. if(dataModel.downloadBoxStateType != downloadBoxStateDone){
  229. dataModel.downloadBoxStateType = downloadBoxStateDownloadloading;
  230. dataModel.totalSize = model.totalSize;
  231. dataModel.currentSize = model.currentSize;
  232. //更新数据库
  233. [dataModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  234. }];
  235. }
  236. break;
  237. }
  238. }
  239. }
  240. }
  241. #pragma mark 下载完成数据回调处理
  242. - (void)handldDownloadDoneDataBy:(SGDownloadOperation*)model
  243. {
  244. //解码
  245. NSString * urlString = [model.url stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  246. if(_databaseArr && _databaseArr.count == 3 ){
  247. NSMutableArray *ingArr = _databaseArr[0];
  248. for (ShareFileDataModel*dataModel in ingArr) {
  249. //HLog(@"22222222 %@\n%@",urlString,dataModel.fileUrl);
  250. if([urlString isEqualToString:dataModel.fileUrl]){
  251. [self handleDatabaseArrByDelete:dataModel];
  252. dataModel.downloadBoxStateType = downloadBoxStateDone;
  253. NSMutableArray *doneArr = _databaseArr[1];
  254. [doneArr insertObject:dataModel atIndex:0];
  255. //更新数据库
  256. [dataModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  257. }];
  258. break;
  259. }
  260. }
  261. }
  262. }
  263. #pragma mark 下载失败数据回调处理
  264. - (void)handldDownloadFailDataBy:(SGDownloadOperation*)model
  265. {
  266. //解码
  267. NSString * urlString = [model.url stringByReplacingPercentEscapesUsingEncoding:NSUTF8StringEncoding];
  268. if(_databaseArr && _databaseArr.count == 3 ){
  269. NSMutableArray *ingArr = _databaseArr[0];
  270. for (ShareFileDataModel*dataModel in ingArr) {
  271. //HLog(@"22222222 %@\n%@",urlString,dataModel.fileUrl);
  272. if([urlString isEqualToString:dataModel.fileUrl]){
  273. [self handleDatabaseArrByDelete:dataModel];
  274. dataModel.downloadBoxStateType = downloadBoxStateFail;
  275. NSMutableArray *failArr = _databaseArr[2];
  276. [failArr insertObject:dataModel atIndex:0];
  277. //更新数据库
  278. [dataModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  279. }];
  280. break;
  281. }
  282. }
  283. }
  284. }
  285. @end