downloadManager.m 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537
  1. //
  2. // downloadManager.m
  3. // 隐私保护
  4. //
  5. // Created by xd h on 2024/1/9.
  6. //
  7. #import "downloadManager.h"
  8. #import "cachesFileManager.h"
  9. #import <sys/mount.h>
  10. #include <sys/param.h>
  11. @interface downloadManager ()
  12. /**检测任务10s后 无反馈重新开始 */
  13. @property (nonatomic, assign) NSInteger taskRenewTime; //
  14. @end
  15. @implementation downloadManager
  16. static downloadManager * cur_downloadManager = nil;
  17. +(downloadManager *)shareInstance;
  18. {
  19. static dispatch_once_t onceToken;
  20. dispatch_once(&onceToken, ^{
  21. cur_downloadManager = [[downloadManager alloc] init];
  22. });
  23. return cur_downloadManager;
  24. }
  25. - (id)init
  26. {
  27. self = [super init];
  28. if (self) {
  29. //[self initManager];
  30. }
  31. return self;
  32. }
  33. #pragma mark 读取数据库数据
  34. - (void)getDataInDatabaseFun:(BOOL)isReGet complete:(custom_complete_Arr)complete
  35. {
  36. if(_databaseArr && _databaseArr.count == 3 && !isReGet){
  37. complete(_databaseArr);
  38. return;
  39. }
  40. if(!_databaseArr)
  41. {
  42. _databaseArr = [NSMutableArray new];
  43. }
  44. //KWeakSelf
  45. //dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT,0), ^{
  46. [couldPhoneFileModel bg_findAsync:download_tableName limit:0 orderBy:nil desc:NO complete:^(NSArray * _Nullable array) {
  47. NSMutableArray *failArr = [NSMutableArray new];
  48. NSMutableArray *doneArr = [NSMutableArray new];
  49. NSMutableArray *otherArr = [NSMutableArray new];
  50. for (couldPhoneFileModel * curModel in array) {
  51. if(curModel.curDownloadStateType == downloadStateFail){
  52. [failArr addObject:curModel];
  53. }
  54. else if(curModel.curDownloadStateType == downloadStateDone){
  55. [doneArr addObject:curModel];
  56. }
  57. else{
  58. [otherArr addObject:curModel];
  59. }
  60. }
  61. NSLock *lock = [NSLock new];
  62. [lock lock];
  63. NSMutableArray *newArr = [NSMutableArray new];
  64. [newArr addObject:otherArr];
  65. [newArr addObject:doneArr];
  66. [newArr addObject:failArr];
  67. self->_databaseArr = newArr;
  68. [lock unlock];
  69. complete(self->_databaseArr);
  70. }];
  71. //});
  72. }
  73. -(void)handleCouldPhoneFileModelToDownloadFileDataFunBy:(NSMutableArray*)indexPathsForSelectedItems complete:(custom_complete_Arr)complete
  74. {
  75. if(!indexPathsForSelectedItems && indexPathsForSelectedItems.count == 0){
  76. complete(nil);
  77. return;
  78. }
  79. for (couldPhoneFileModel*model in indexPathsForSelectedItems) {
  80. model.bg_tableName = download_tableName;
  81. }
  82. _isNewAddTaskType = YES;
  83. KWeakSelf
  84. //查
  85. [self getDataInDatabaseFun:YES complete:^(NSMutableArray * _Nonnull Arr) {
  86. //排重
  87. NSMutableArray *newDownArr = [weakSelf checkDownloadTaskRepeatFun:indexPathsForSelectedItems];
  88. if(newDownArr.count>0){
  89. //存
  90. [couldPhoneFileModel bg_saveOrUpdateArrayAsync:newDownArr complete:^(BOOL isSuccess) {
  91. //再查
  92. [weakSelf getDataInDatabaseFun:YES complete:^(NSMutableArray * _Nonnull Arr) {
  93. complete(Arr);
  94. [weakSelf planToDownloadFileFun];
  95. }];
  96. }];
  97. }
  98. else{
  99. complete(Arr);
  100. [weakSelf planToDownloadFileFun];
  101. }
  102. }];
  103. }
  104. #pragma mark 检测是否有下载任务重复
  105. - (NSMutableArray*)checkDownloadTaskRepeatFun:(NSMutableArray*)indexPathsForSelectedItems
  106. {
  107. if(!_databaseArr || _databaseArr.count != 3){
  108. return indexPathsForSelectedItems;
  109. }
  110. NSMutableArray *downloadArr = _databaseArr[0];
  111. NSMutableArray *newTaskArr = [NSMutableArray new];
  112. for (couldPhoneFileModel*model in indexPathsForSelectedItems) {
  113. NSString *fileName = [model getFileNameFun];
  114. BOOL isNeedAddType = YES;
  115. for (couldPhoneFileModel* hadModel in downloadArr) {
  116. NSString *hadFileName = [hadModel getFileNameFun];
  117. if([fileName isEqualToString:hadFileName]){
  118. isNeedAddType = NO;
  119. break;
  120. }
  121. }
  122. if(isNeedAddType){
  123. [newTaskArr addObject:model];
  124. }
  125. }
  126. return newTaskArr;
  127. }
  128. - (void)planToDownloadFileFun
  129. {
  130. if(!_databaseArr || _databaseArr.count != 3){
  131. return;
  132. }
  133. _downLoadFileModelDataArr = _databaseArr[0];
  134. [self beginDownloadFileFun];
  135. }
  136. - (BOOL)checkFreeDiskSpaceInBytesIsOK:(long)fileByte{
  137. struct statfs buf;
  138. unsigned long long freeSpace = -1;
  139. if (statfs("/var", &buf) >= 0) {
  140. freeSpace = (unsigned long long)(buf.f_bsize * buf.f_bavail);
  141. }
  142. long space = freeSpace - fileByte;
  143. if( space/fileByte < 2.0 || space < 500*1024*1024){
  144. return NO;
  145. }
  146. return YES;
  147. }
  148. - (void)beginDownloadFileFun
  149. {
  150. if(_downLoadFileModelDataArr.count == 0)
  151. {
  152. _curDownloadFileModel = nil;
  153. [self didDownloadAllTaskDoneFun];
  154. return;
  155. }
  156. self.taskRenewTime = [iTools getNowTimeStamp];
  157. _isSuspendType = NO;
  158. _isDownloadIngType = YES;
  159. if(_reDownloadIngSelectIndex > 0
  160. && _reDownloadIngSelectIndex < _downLoadFileModelDataArr.count){
  161. _curDownloadFileModel = _downLoadFileModelDataArr[_reDownloadIngSelectIndex];
  162. }
  163. else
  164. {
  165. _curDownloadFileModel = _downLoadFileModelDataArr.firstObject;
  166. }
  167. if(!_curDownloadFileModel){
  168. return;
  169. }
  170. //做个异常处理
  171. if(_databaseArr.count == 3 && _isNewAddTaskType){
  172. NSMutableArray *downloadIngArr = _databaseArr[0];
  173. if(_downLoadFileModelDataArr.count != downloadIngArr.count){
  174. _databaseArr[0] = _downLoadFileModelDataArr;
  175. }
  176. }
  177. if(![self checkFreeDiskSpaceInBytesIsOK:_curDownloadFileModel.length]){
  178. mainBlock(^{
  179. [[iToast makeText:NSLocalizedString(@"phone_space_Insufficient",nil)] show];
  180. });
  181. return;
  182. }
  183. _curDownloadFileModel.curDownloadStateType = downloadStateUploading;
  184. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileBeginNotification object:_curDownloadFileModel];
  185. }
  186. - (void)DownloadFileDoneOneFileFun
  187. {
  188. self.taskRenewTime = [iTools getNowTimeStamp];
  189. _curDownloadFileModel.curDownloadStateType = downloadStateDone;
  190. // NSString *fileName = [_curDownloadFileModel getFileNameFun];
  191. // [cachesFileManager removeItemAtPath:fileName type:DownLoadFileType error:nil];
  192. [self DownloadFileChangeingOneFileFun];
  193. if(_isSuspendType){
  194. return;
  195. }
  196. [self beginDownloadFileFun];
  197. }
  198. - (void)DownloadFileFailOneFileFun
  199. {
  200. self.taskRenewTime = [iTools getNowTimeStamp];
  201. _curDownloadFileModel.curDownloadStateType = downloadStateFail;
  202. [self DownloadFileChangeingOneFileFun];
  203. if(_isSuspendType){
  204. return;
  205. }
  206. [self beginDownloadFileFun];
  207. }
  208. - (void)DownloadFileChangeingOneFileFun
  209. {
  210. self.taskRenewTime = [iTools getNowTimeStamp];
  211. if(_curDownloadFileModel.curDownloadStateType == downloadStateDone){
  212. [_downLoadFileModelDataArr removeObject:_curDownloadFileModel];
  213. [self handleDatabaseArrDeleteObjectInDownloading:_curDownloadFileModel];
  214. [self handleDatabaseArrAddObjectInDone:_curDownloadFileModel];
  215. }
  216. else if(_curDownloadFileModel.curDownloadStateType == uploadStateFail){
  217. [_downLoadFileModelDataArr removeObject:_curDownloadFileModel];
  218. [self handleDatabaseArrDeleteObjectInDownloading:_curDownloadFileModel];
  219. [self handleDatabaseArrAddObjectInFail:_curDownloadFileModel];
  220. }
  221. [_curDownloadFileModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  222. //HLog(@"%@ 写入 %@", model.filename, isSuccess ? @"成功":@"失败");
  223. }];
  224. if(!_isSuspendType || _curDownloadFileModel.curDownloadStateType != uploadStateSuspend){
  225. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileRefreshNotification object:_curDownloadFileModel];
  226. }
  227. }
  228. - (void)handleDatabaseArrDeleteObjectInDownloading:(couldPhoneFileModel*)model
  229. {
  230. NSLock *lock = [NSLock new];
  231. [lock lock];
  232. if(_databaseArr && _databaseArr.count == 3 ){
  233. NSMutableArray *ingArr = _databaseArr[0];
  234. NSInteger count = ingArr.count;
  235. [ingArr removeObject:model];
  236. NSInteger atferDelCount = ingArr.count;
  237. if(count == atferDelCount){
  238. for (couldPhoneFileModel*preModel in ingArr) {
  239. if(preModel.bg_id.integerValue == model.bg_id.integerValue){
  240. [ingArr removeObject:preModel];
  241. break;
  242. }
  243. }
  244. }
  245. NSInteger atferDelCount2 = ingArr.count;
  246. if(count == atferDelCount2){
  247. // HLog(@"\n\n\nerror:下载中的任务没有删除掉database bgid:%@\n\n\n\n",model.bg_id);
  248. }
  249. }
  250. [lock unlock];
  251. }
  252. - (void)handleDatabaseArrAddObjectInDone:(couldPhoneFileModel*)model
  253. {
  254. NSLock *lock = [NSLock new];
  255. [lock lock];
  256. if(_databaseArr && _databaseArr.count == 3 ){
  257. NSMutableArray *doneArr = _databaseArr[1];
  258. [doneArr insertObject:model atIndex:0];
  259. }
  260. [lock unlock];
  261. }
  262. - (void)handleDatabaseArrAddObjectInFail:(couldPhoneFileModel*)model
  263. {
  264. NSLock *lock = [NSLock new];
  265. [lock lock];
  266. if(_databaseArr && _databaseArr.count == 3 ){
  267. NSMutableArray *failArr = _databaseArr[2];
  268. [failArr insertObject:model atIndex:0];
  269. }
  270. [lock unlock];
  271. }
  272. - (void)suspendDownloadFileFun:(BOOL)isSuspendAll
  273. {
  274. _isSuspendType = YES;
  275. if(isSuspendAll){
  276. for (couldPhoneFileModel *couldPhoneFileMod in _downLoadFileModelDataArr) {
  277. couldPhoneFileMod.curDownloadStateType = downloadStateSuspend;
  278. }
  279. }
  280. }
  281. - (void)reDownloadFileFunBy:(NSMutableArray*)Arr withAll:(BOOL)isAllType
  282. {
  283. if(!_downLoadFileModelDataArr){
  284. _downLoadFileModelDataArr = [NSMutableArray new];
  285. }
  286. if(isAllType)
  287. {
  288. _isNewAddTaskType = YES;
  289. _downLoadFileModelDataArr = Arr;
  290. _reDownloadIngSelectIndex = -1;
  291. [self beginDownloadFileFun];
  292. return;
  293. }
  294. _isNewAddTaskType = NO;
  295. _reDownloadIngSelectIndex = 0;
  296. NSMutableArray *curArr = [NSMutableArray arrayWithArray:Arr];
  297. _downLoadFileModelDataArr = curArr;
  298. // for (uploadFileDataModel*addModel in Arr) {
  299. //
  300. // BOOL needAddModel = YES;
  301. //
  302. // for (int i=0;i< _downLoadFileModelDataArr.count;i++)
  303. // {
  304. // couldPhoneFileModel*preModel = _downLoadFileModelDataArr[i];
  305. //
  306. // if(addModel.bg_id.integerValue == preModel.bg_id.integerValue)
  307. // {
  308. // needAddModel = NO;
  309. //
  310. // if(_reDownloadIngSelectIndex == 0){
  311. // _reDownloadIngSelectIndex = i;
  312. //
  313. // if(i==0){
  314. // _reDownloadIngSelectIndex = -1;
  315. // }
  316. // }
  317. //
  318. // break;
  319. // }
  320. // }
  321. //
  322. // if(needAddModel){
  323. // [_downLoadFileModelDataArr addObject:addModel];
  324. // _reDownloadIngSelectIndex = _downLoadFileModelDataArr.count -1;
  325. // }
  326. // }
  327. [self beginDownloadFileFun];
  328. }
  329. - (void)handleDatabaseArrByDelete:(couldPhoneFileModel*)delModel
  330. {
  331. if(_databaseArr && _databaseArr.count == 3 ){
  332. if(delModel.curDownloadStateType == downloadStateDone)
  333. {
  334. NSMutableArray *arr = _databaseArr[1];
  335. [arr removeObject:delModel];
  336. }
  337. else if(delModel.curDownloadStateType == downloadStateFail){
  338. NSMutableArray *arr = _databaseArr[2];
  339. [arr removeObject:delModel];
  340. }
  341. else{
  342. NSMutableArray *arr = _databaseArr[0];
  343. [arr removeObject:delModel];
  344. }
  345. }
  346. }
  347. - (void)deleteDownloadFileRecordBy:(NSMutableArray *)delArr withDelCache:(BOOL)isDelCache
  348. {
  349. //逻辑待优化
  350. BOOL isSuc = false;
  351. BOOL isDelUploadingModel = false;
  352. NSMutableArray *curDelArr = [NSMutableArray arrayWithArray:delArr];
  353. for (couldPhoneFileModel *couldPhoneFileMod in curDelArr) {
  354. NSMutableString* where = [[NSMutableString alloc] initWithString:@"where "];
  355. NSString *curStr = [NSString stringWithFormat:@"%@=%@ ",bg_sqlKey(@"bg_id"),bg_sqlValue(couldPhoneFileMod.bg_id)];
  356. [where appendString:curStr];
  357. [couldPhoneFileModel bg_deleteAsync:download_tableName where:where complete:^(BOOL isSuccess) {
  358. }];
  359. //isSuc = [couldPhoneFileModel bg_delete:download_tableName where:where];
  360. [self handleDatabaseArrByDelete:couldPhoneFileMod];
  361. if(couldPhoneFileMod.bg_id.integerValue == _curDownloadFileModel.bg_id.integerValue){
  362. isDelUploadingModel = YES;
  363. [self suspendDownloadFileFun:NO];
  364. }
  365. }
  366. if(isDelUploadingModel){
  367. _isDownloadIngType = NO;
  368. if(_downLoadFileModelDataArr > 0){
  369. [self beginDownloadFileFun];
  370. }
  371. }
  372. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileRefreshNotification object:_curDownloadFileModel];
  373. }
  374. //检查是否需要重新下载
  375. - (void)checkReDownloadFileFun
  376. {
  377. if(!_curDownloadFileModel){
  378. return;
  379. }
  380. if(_isSuspendType){
  381. return;
  382. }
  383. NSInteger curTime = [iTools getNowTimeStamp];
  384. if(self.taskRenewTime == 0){
  385. return;
  386. }
  387. if(curTime - self.taskRenewTime < 10){
  388. return;
  389. }
  390. if(![self checkFreeDiskSpaceInBytesIsOK:_curDownloadFileModel.length]){
  391. mainBlock(^{
  392. [[iToast makeText:NSLocalizedString(@"phone_space_Insufficient",nil)] show];
  393. });
  394. _curDownloadFileModel = nil;
  395. return;
  396. }
  397. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileBeginNotification object:_curDownloadFileModel];
  398. }
  399. - (void)didDownloadAllTaskDoneFun
  400. {
  401. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileAllTaskDoneNotification object:nil];
  402. }
  403. - (void)checkHadDownloadTaskWithComplete:(custom_complete_B)complete
  404. {
  405. [self getDataInDatabaseFun:NO complete:^(NSMutableArray * _Nonnull Arr) {
  406. mainBlock(^{
  407. if(!Arr || Arr.count != 3){
  408. complete(NO);
  409. }
  410. else{
  411. NSArray *firstArr = Arr[0];
  412. if(firstArr.count >0){
  413. complete(YES);
  414. }
  415. else{
  416. complete(NO);
  417. }
  418. }
  419. });
  420. }];
  421. }
  422. @end