downloadManager.m 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500
  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. return;
  154. }
  155. self.taskRenewTime = [iTools getNowTimeStamp];
  156. _isSuspendType = NO;
  157. _isDownloadIngType = YES;
  158. if(_reDownloadIngSelectIndex > 0
  159. && _reDownloadIngSelectIndex < _downLoadFileModelDataArr.count){
  160. _curDownloadFileModel = _downLoadFileModelDataArr[_reDownloadIngSelectIndex];
  161. }
  162. else
  163. {
  164. _curDownloadFileModel = _downLoadFileModelDataArr.firstObject;
  165. }
  166. if(!_curDownloadFileModel){
  167. return;
  168. }
  169. //做个异常处理
  170. if(_databaseArr.count == 3 && _isNewAddTaskType){
  171. NSMutableArray *downloadIngArr = _databaseArr[0];
  172. if(_downLoadFileModelDataArr.count != downloadIngArr.count){
  173. _databaseArr[0] = _downLoadFileModelDataArr;
  174. }
  175. }
  176. if(![self checkFreeDiskSpaceInBytesIsOK:_curDownloadFileModel.length]){
  177. mainBlock(^{
  178. [[iToast makeText:NSLocalizedString(@"phone_space_Insufficient",nil)] show];
  179. });
  180. return;
  181. }
  182. _curDownloadFileModel.curDownloadStateType = downloadStateUploading;
  183. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileBeginNotification object:_curDownloadFileModel];
  184. }
  185. - (void)DownloadFileDoneOneFileFun
  186. {
  187. self.taskRenewTime = [iTools getNowTimeStamp];
  188. _curDownloadFileModel.curDownloadStateType = downloadStateDone;
  189. // NSString *fileName = [_curDownloadFileModel getFileNameFun];
  190. // [cachesFileManager removeItemAtPath:fileName type:DownLoadFileType error:nil];
  191. [self DownloadFileChangeingOneFileFun];
  192. if(_isSuspendType){
  193. return;
  194. }
  195. [self beginDownloadFileFun];
  196. }
  197. - (void)DownloadFileFailOneFileFun
  198. {
  199. self.taskRenewTime = [iTools getNowTimeStamp];
  200. _curDownloadFileModel.curDownloadStateType = downloadStateFail;
  201. [self DownloadFileChangeingOneFileFun];
  202. if(_isSuspendType){
  203. return;
  204. }
  205. [self beginDownloadFileFun];
  206. }
  207. - (void)DownloadFileChangeingOneFileFun
  208. {
  209. self.taskRenewTime = [iTools getNowTimeStamp];
  210. if(_curDownloadFileModel.curDownloadStateType == downloadStateDone){
  211. [_downLoadFileModelDataArr removeObject:_curDownloadFileModel];
  212. [self handleDatabaseArrDeleteObjectInDownloading:_curDownloadFileModel];
  213. [self handleDatabaseArrAddObjectInDone:_curDownloadFileModel];
  214. }
  215. else if(_curDownloadFileModel.curDownloadStateType == uploadStateFail){
  216. [_downLoadFileModelDataArr removeObject:_curDownloadFileModel];
  217. [self handleDatabaseArrDeleteObjectInDownloading:_curDownloadFileModel];
  218. [self handleDatabaseArrAddObjectInFail:_curDownloadFileModel];
  219. }
  220. [_curDownloadFileModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  221. //HLog(@"%@ 写入 %@", model.filename, isSuccess ? @"成功":@"失败");
  222. }];
  223. if(!_isSuspendType || _curDownloadFileModel.curDownloadStateType != uploadStateSuspend){
  224. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileRefreshNotification object:_curDownloadFileModel];
  225. }
  226. }
  227. - (void)handleDatabaseArrDeleteObjectInDownloading:(couldPhoneFileModel*)model
  228. {
  229. if(_databaseArr && _databaseArr.count == 3 ){
  230. NSMutableArray *ingArr = _databaseArr[0];
  231. NSInteger count = ingArr.count;
  232. [ingArr removeObject:model];
  233. NSInteger atferDelCount = ingArr.count;
  234. if(count == atferDelCount){
  235. for (couldPhoneFileModel*preModel in ingArr) {
  236. if(preModel.bg_id.integerValue == model.bg_id.integerValue){
  237. [ingArr removeObject:preModel];
  238. break;
  239. }
  240. }
  241. }
  242. NSInteger atferDelCount2 = ingArr.count;
  243. if(count == atferDelCount2){
  244. // HLog(@"\n\n\nerror:下载中的任务没有删除掉database bgid:%@\n\n\n\n",model.bg_id);
  245. }
  246. }
  247. }
  248. - (void)handleDatabaseArrAddObjectInDone:(couldPhoneFileModel*)model
  249. {
  250. if(_databaseArr && _databaseArr.count == 3 ){
  251. NSMutableArray *doneArr = _databaseArr[1];
  252. [doneArr insertObject:model atIndex:0];
  253. }
  254. }
  255. - (void)handleDatabaseArrAddObjectInFail:(couldPhoneFileModel*)model
  256. {
  257. if(_databaseArr && _databaseArr.count == 3 ){
  258. NSMutableArray *failArr = _databaseArr[2];
  259. [failArr insertObject:model atIndex:0];
  260. }
  261. }
  262. - (void)suspendDownloadFileFun:(BOOL)isSuspendAll
  263. {
  264. _isSuspendType = YES;
  265. if(isSuspendAll){
  266. for (couldPhoneFileModel *couldPhoneFileMod in _downLoadFileModelDataArr) {
  267. couldPhoneFileMod.curDownloadStateType = downloadStateSuspend;
  268. }
  269. }
  270. }
  271. - (void)reDownloadFileFunBy:(NSMutableArray*)Arr withAll:(BOOL)isAllType
  272. {
  273. if(!_downLoadFileModelDataArr){
  274. _downLoadFileModelDataArr = [NSMutableArray new];
  275. }
  276. if(isAllType)
  277. {
  278. _isNewAddTaskType = YES;
  279. _downLoadFileModelDataArr = Arr;
  280. _reDownloadIngSelectIndex = -1;
  281. [self beginDownloadFileFun];
  282. return;
  283. }
  284. _isNewAddTaskType = NO;
  285. _reDownloadIngSelectIndex = 0;
  286. NSMutableArray *curArr = [NSMutableArray arrayWithArray:Arr];
  287. _downLoadFileModelDataArr = curArr;
  288. // for (uploadFileDataModel*addModel in Arr) {
  289. //
  290. // BOOL needAddModel = YES;
  291. //
  292. // for (int i=0;i< _downLoadFileModelDataArr.count;i++)
  293. // {
  294. // couldPhoneFileModel*preModel = _downLoadFileModelDataArr[i];
  295. //
  296. // if(addModel.bg_id.integerValue == preModel.bg_id.integerValue)
  297. // {
  298. // needAddModel = NO;
  299. //
  300. // if(_reDownloadIngSelectIndex == 0){
  301. // _reDownloadIngSelectIndex = i;
  302. //
  303. // if(i==0){
  304. // _reDownloadIngSelectIndex = -1;
  305. // }
  306. // }
  307. //
  308. // break;
  309. // }
  310. // }
  311. //
  312. // if(needAddModel){
  313. // [_downLoadFileModelDataArr addObject:addModel];
  314. // _reDownloadIngSelectIndex = _downLoadFileModelDataArr.count -1;
  315. // }
  316. // }
  317. [self beginDownloadFileFun];
  318. }
  319. - (void)handleDatabaseArrByDelete:(couldPhoneFileModel*)delModel
  320. {
  321. if(_databaseArr && _databaseArr.count == 3 ){
  322. if(delModel.curDownloadStateType == downloadStateDone)
  323. {
  324. NSMutableArray *arr = _databaseArr[1];
  325. [arr removeObject:delModel];
  326. }
  327. else if(delModel.curDownloadStateType == downloadStateFail){
  328. NSMutableArray *arr = _databaseArr[2];
  329. [arr removeObject:delModel];
  330. }
  331. else{
  332. NSMutableArray *arr = _databaseArr[0];
  333. [arr removeObject:delModel];
  334. }
  335. }
  336. }
  337. - (void)deleteDownloadFileRecordBy:(NSMutableArray *)delArr withDelCache:(BOOL)isDelCache
  338. {
  339. //逻辑待优化
  340. BOOL isSuc = false;
  341. BOOL isDelUploadingModel = false;
  342. NSMutableArray *curDelArr = [NSMutableArray arrayWithArray:delArr];
  343. for (couldPhoneFileModel *couldPhoneFileMod in curDelArr) {
  344. NSMutableString* where = [[NSMutableString alloc] initWithString:@"where "];
  345. NSString *curStr = [NSString stringWithFormat:@"%@=%@ ",bg_sqlKey(@"bg_id"),bg_sqlValue(couldPhoneFileMod.bg_id)];
  346. [where appendString:curStr];
  347. [couldPhoneFileModel bg_deleteAsync:download_tableName where:where complete:^(BOOL isSuccess) {
  348. }];
  349. //isSuc = [couldPhoneFileModel bg_delete:download_tableName where:where];
  350. [self handleDatabaseArrByDelete:couldPhoneFileMod];
  351. if(couldPhoneFileMod.bg_id.integerValue == _curDownloadFileModel.bg_id.integerValue){
  352. isDelUploadingModel = YES;
  353. [self suspendDownloadFileFun:NO];
  354. }
  355. }
  356. if(isDelUploadingModel){
  357. _isDownloadIngType = NO;
  358. if(_downLoadFileModelDataArr > 0){
  359. [self beginDownloadFileFun];
  360. }
  361. }
  362. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileRefreshNotification object:_curDownloadFileModel];
  363. }
  364. //检查是否需要重新下载
  365. - (void)checkReDownloadFileFun
  366. {
  367. if(!_curDownloadFileModel){
  368. return;
  369. }
  370. if(_isSuspendType){
  371. return;
  372. }
  373. NSInteger curTime = [iTools getNowTimeStamp];
  374. if(self.taskRenewTime == 0){
  375. return;
  376. }
  377. if(curTime - self.taskRenewTime < 10){
  378. return;
  379. }
  380. if(![self checkFreeDiskSpaceInBytesIsOK:_curDownloadFileModel.length]){
  381. mainBlock(^{
  382. [[iToast makeText:NSLocalizedString(@"phone_space_Insufficient",nil)] show];
  383. });
  384. _curDownloadFileModel = nil;
  385. return;
  386. }
  387. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileBeginNotification object:_curDownloadFileModel];
  388. }
  389. @end