downloadManager.m 13 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464
  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. KWeakSelf
  83. //查
  84. [self getDataInDatabaseFun:YES complete:^(NSMutableArray * _Nonnull Arr) {
  85. //排重
  86. NSMutableArray *newDownArr = [weakSelf checkDownloadTaskRepeatFun:indexPathsForSelectedItems];
  87. if(newDownArr.count>0){
  88. //存
  89. [couldPhoneFileModel bg_saveOrUpdateArrayAsync:newDownArr complete:^(BOOL isSuccess) {
  90. //再查
  91. [weakSelf getDataInDatabaseFun:YES complete:^(NSMutableArray * _Nonnull Arr) {
  92. complete(Arr);
  93. [weakSelf planToDownloadFileFun];
  94. }];
  95. }];
  96. }
  97. else{
  98. complete(Arr);
  99. [weakSelf planToDownloadFileFun];
  100. }
  101. }];
  102. }
  103. #pragma mark 检测是否有下载任务重复
  104. - (NSMutableArray*)checkDownloadTaskRepeatFun:(NSMutableArray*)indexPathsForSelectedItems
  105. {
  106. if(!_databaseArr || _databaseArr.count != 3){
  107. return indexPathsForSelectedItems;
  108. }
  109. NSMutableArray *downloadArr = _databaseArr[0];
  110. NSMutableArray *newTaskArr = [NSMutableArray new];
  111. for (couldPhoneFileModel*model in indexPathsForSelectedItems) {
  112. NSString *fileName = [model getFileNameFun];
  113. BOOL isNeedAddType = YES;
  114. for (couldPhoneFileModel* hadModel in downloadArr) {
  115. NSString *hadFileName = [hadModel getFileNameFun];
  116. if([fileName isEqualToString:hadFileName]){
  117. isNeedAddType = NO;
  118. break;
  119. }
  120. }
  121. if(isNeedAddType){
  122. [newTaskArr addObject:model];
  123. }
  124. }
  125. return newTaskArr;
  126. }
  127. - (void)planToDownloadFileFun
  128. {
  129. if(!_databaseArr || _databaseArr.count != 3){
  130. return;
  131. }
  132. _downLoadFileModelDataArr = _databaseArr[0];
  133. [self beginDownloadFileFun];
  134. }
  135. - (BOOL)checkFreeDiskSpaceInBytesIsOK:(long)fileByte{
  136. struct statfs buf;
  137. unsigned long long freeSpace = -1;
  138. if (statfs("/var", &buf) >= 0) {
  139. freeSpace = (unsigned long long)(buf.f_bsize * buf.f_bavail);
  140. }
  141. long space = freeSpace - fileByte;
  142. if( space/fileByte < 2.0 || space < 500*1024*1024){
  143. return NO;
  144. }
  145. return YES;
  146. }
  147. - (void)beginDownloadFileFun
  148. {
  149. if(_downLoadFileModelDataArr.count == 0)
  150. {
  151. _curDownloadFileModel = nil;
  152. return;
  153. }
  154. self.taskRenewTime = [iTools getNowTimeStamp];
  155. _isSuspendType = NO;
  156. _isDownloadIngType = YES;
  157. if(_reDownloadIngSelectIndex > 0
  158. && _reDownloadIngSelectIndex < _downLoadFileModelDataArr.count){
  159. _curDownloadFileModel = _downLoadFileModelDataArr[_reDownloadIngSelectIndex];
  160. }
  161. else
  162. {
  163. _curDownloadFileModel = _downLoadFileModelDataArr.firstObject;
  164. }
  165. if(!_curDownloadFileModel){
  166. return;
  167. }
  168. if(![self checkFreeDiskSpaceInBytesIsOK:_curDownloadFileModel.length]){
  169. mainBlock(^{
  170. [[iToast makeText:NSLocalizedString(@"phone_space_Insufficient",nil)] show];
  171. });
  172. return;
  173. }
  174. _curDownloadFileModel.curDownloadStateType = downloadStateUploading;
  175. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileBeginNotification object:_curDownloadFileModel];
  176. }
  177. - (void)DownloadFileDoneOneFileFun
  178. {
  179. self.taskRenewTime = [iTools getNowTimeStamp];
  180. _curDownloadFileModel.curDownloadStateType = downloadStateDone;
  181. // NSString *fileName = [_curDownloadFileModel getFileNameFun];
  182. // [cachesFileManager removeItemAtPath:fileName type:DownLoadFileType error:nil];
  183. [self DownloadFileChangeingOneFileFun];
  184. if(_isSuspendType){
  185. return;
  186. }
  187. [self beginDownloadFileFun];
  188. }
  189. - (void)DownloadFileFailOneFileFun
  190. {
  191. self.taskRenewTime = [iTools getNowTimeStamp];
  192. _curDownloadFileModel.curDownloadStateType = downloadStateFail;
  193. [self DownloadFileChangeingOneFileFun];
  194. if(_isSuspendType){
  195. return;
  196. }
  197. [self beginDownloadFileFun];
  198. }
  199. - (void)DownloadFileChangeingOneFileFun
  200. {
  201. self.taskRenewTime = [iTools getNowTimeStamp];
  202. if(_curDownloadFileModel.curDownloadStateType == downloadStateDone){
  203. [_downLoadFileModelDataArr removeObject:_curDownloadFileModel];
  204. [self handleDatabaseArrDeleteObjectInDownloading:_curDownloadFileModel];
  205. [self handleDatabaseArrAddObjectInDone:_curDownloadFileModel];
  206. }
  207. else if(_curDownloadFileModel.curDownloadStateType == uploadStateFail){
  208. [_downLoadFileModelDataArr removeObject:_curDownloadFileModel];
  209. [self handleDatabaseArrDeleteObjectInDownloading:_curDownloadFileModel];
  210. [self handleDatabaseArrAddObjectInFail:_curDownloadFileModel];
  211. }
  212. [_curDownloadFileModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  213. //HLog(@"%@ 写入 %@", model.filename, isSuccess ? @"成功":@"失败");
  214. }];
  215. if(!_isSuspendType || _curDownloadFileModel.curDownloadStateType != uploadStateSuspend){
  216. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileRefreshNotification object:_curDownloadFileModel];
  217. }
  218. }
  219. - (void)handleDatabaseArrDeleteObjectInDownloading:(couldPhoneFileModel*)model
  220. {
  221. if(_databaseArr && _databaseArr.count == 3 ){
  222. NSMutableArray *ingArr = _databaseArr[0];
  223. [ingArr removeObject:model];
  224. }
  225. }
  226. - (void)handleDatabaseArrAddObjectInDone:(couldPhoneFileModel*)model
  227. {
  228. if(_databaseArr && _databaseArr.count == 3 ){
  229. NSMutableArray *doneArr = _databaseArr[1];
  230. [doneArr insertObject:model atIndex:0];
  231. }
  232. }
  233. - (void)handleDatabaseArrAddObjectInFail:(couldPhoneFileModel*)model
  234. {
  235. if(_databaseArr && _databaseArr.count == 3 ){
  236. NSMutableArray *failArr = _databaseArr[2];
  237. [failArr insertObject:model atIndex:0];
  238. }
  239. }
  240. - (void)suspendDownloadFileFun:(BOOL)isSuspendAll
  241. {
  242. _isSuspendType = YES;
  243. if(isSuspendAll){
  244. for (couldPhoneFileModel *couldPhoneFileMod in _downLoadFileModelDataArr) {
  245. couldPhoneFileMod.curDownloadStateType = downloadStateSuspend;
  246. }
  247. }
  248. }
  249. - (void)reDownloadFileFunBy:(NSMutableArray*)Arr withAll:(BOOL)isAllType
  250. {
  251. if(!_downLoadFileModelDataArr){
  252. _downLoadFileModelDataArr = [NSMutableArray new];
  253. }
  254. if(isAllType)
  255. {
  256. _downLoadFileModelDataArr = Arr;
  257. _reDownloadIngSelectIndex = -1;
  258. [self beginDownloadFileFun];
  259. return;
  260. }
  261. _reDownloadIngSelectIndex = 0;
  262. for (uploadFileDataModel*addModel in Arr) {
  263. BOOL needAddModel = YES;
  264. for (int i=0;i< _downLoadFileModelDataArr.count;i++)
  265. {
  266. couldPhoneFileModel*preModel = _downLoadFileModelDataArr[i];
  267. if(addModel.bg_id.integerValue == preModel.bg_id.integerValue)
  268. {
  269. needAddModel = NO;
  270. if(_reDownloadIngSelectIndex == 0){
  271. _reDownloadIngSelectIndex = i;
  272. if(i==0){
  273. _reDownloadIngSelectIndex = -1;
  274. }
  275. }
  276. break;
  277. }
  278. }
  279. if(needAddModel){
  280. [_downLoadFileModelDataArr addObject:addModel];
  281. _reDownloadIngSelectIndex = _downLoadFileModelDataArr.count -1;
  282. }
  283. }
  284. [self beginDownloadFileFun];
  285. }
  286. - (void)handleDatabaseArrByDelete:(couldPhoneFileModel*)delModel
  287. {
  288. if(_databaseArr && _databaseArr.count == 3 ){
  289. if(delModel.curDownloadStateType == downloadStateDone)
  290. {
  291. NSMutableArray *arr = _databaseArr[1];
  292. [arr removeObject:delModel];
  293. }
  294. else if(delModel.curDownloadStateType == downloadStateFail){
  295. NSMutableArray *arr = _databaseArr[2];
  296. [arr removeObject:delModel];
  297. }
  298. else{
  299. NSMutableArray *arr = _databaseArr[0];
  300. [arr removeObject:delModel];
  301. }
  302. }
  303. }
  304. - (void)deleteDownloadFileRecordBy:(NSMutableArray *)delArr withDelCache:(BOOL)isDelCache
  305. {
  306. //逻辑待优化
  307. BOOL isSuc = false;
  308. BOOL isDelUploadingModel = false;
  309. NSMutableArray *curDelArr = [NSMutableArray arrayWithArray:delArr];
  310. for (couldPhoneFileModel *couldPhoneFileMod in curDelArr) {
  311. NSMutableString* where = [[NSMutableString alloc] initWithString:@"where "];
  312. NSString *curStr = [NSString stringWithFormat:@"%@=%@ ",bg_sqlKey(@"bg_id"),bg_sqlValue(couldPhoneFileMod.bg_id)];
  313. [where appendString:curStr];
  314. [couldPhoneFileModel bg_deleteAsync:download_tableName where:where complete:^(BOOL isSuccess) {
  315. }];
  316. //isSuc = [couldPhoneFileModel bg_delete:download_tableName where:where];
  317. [self handleDatabaseArrByDelete:couldPhoneFileMod];
  318. if(couldPhoneFileMod.bg_id.integerValue == _curDownloadFileModel.bg_id.integerValue){
  319. isDelUploadingModel = YES;
  320. [self suspendDownloadFileFun:NO];
  321. }
  322. }
  323. if(isDelUploadingModel){
  324. _isDownloadIngType = NO;
  325. if(_downLoadFileModelDataArr > 0){
  326. [self beginDownloadFileFun];
  327. }
  328. }
  329. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileRefreshNotification object:_curDownloadFileModel];
  330. }
  331. //检查是否需要重新下载
  332. - (void)checkReDownloadFileFun
  333. {
  334. if(!_curDownloadFileModel){
  335. return;
  336. }
  337. if(_isSuspendType){
  338. return;
  339. }
  340. NSInteger curTime = [iTools getNowTimeStamp];
  341. if(self.taskRenewTime == 0){
  342. return;
  343. }
  344. if(curTime - self.taskRenewTime < 10){
  345. return;
  346. }
  347. if(![self checkFreeDiskSpaceInBytesIsOK:_curDownloadFileModel.length]){
  348. mainBlock(^{
  349. [[iToast makeText:NSLocalizedString(@"phone_space_Insufficient",nil)] show];
  350. });
  351. _curDownloadFileModel = nil;
  352. return;
  353. }
  354. [[NSNotificationCenter defaultCenter] postNotificationName:downloadFileBeginNotification object:_curDownloadFileModel];
  355. }
  356. @end