uploadFileManager.m 35 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016
  1. //
  2. // uploadFileManager.m
  3. // 隐私保护
  4. //
  5. // Created by xd h on 2023/11/15.
  6. //
  7. #import "uploadFileManager.h"
  8. #import <AssetsLibrary/AssetsLibrary.h>
  9. #import "photosBackupsTaskModel.h"
  10. #import "AFNetworkReachabilityManager.h"
  11. @interface uploadFileManager ()
  12. /**检测任务10s后 无反馈重新开始 */
  13. @property (nonatomic, assign) NSInteger taskRenewTime; //
  14. @end
  15. @implementation uploadFileManager
  16. static uploadFileManager * cur_uploadFileShareInstance = nil;
  17. +(uploadFileManager *)shareInstance
  18. {
  19. static dispatch_once_t onceToken;
  20. dispatch_once(&onceToken, ^{
  21. cur_uploadFileShareInstance = [[uploadFileManager alloc] init];
  22. });
  23. return cur_uploadFileShareInstance;
  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. // NSLock *lock = [NSLock new];
  47. // [lock lock];
  48. [uploadFileDataModel bg_findAsync:upLoadFile_image_tableName limit:0 orderBy:nil desc:YES complete:^(NSArray * _Nullable array) {
  49. NSMutableArray *failArr = [NSMutableArray new];
  50. NSMutableArray *doneArr = [NSMutableArray new];
  51. NSMutableArray *otherArr = [NSMutableArray new];
  52. for (uploadFileDataModel * curModel in array) {
  53. //图片 和视频 还原
  54. if(curModel.curUploadFileType == uploadFileTypeImage){
  55. }
  56. else if(curModel.curUploadFileType == uploadFileTypeVideo){
  57. if(curModel.curUploadStateType != uploadStateDone){
  58. }
  59. }
  60. if(curModel.curUploadStateType == uploadStateFail){
  61. [failArr addObject:curModel];
  62. }
  63. else if(curModel.curUploadStateType == uploadStateDone){
  64. [doneArr addObject:curModel];
  65. }
  66. else{
  67. [otherArr addObject:curModel];
  68. }
  69. }
  70. NSLock *lock = [NSLock new];
  71. [lock lock];
  72. NSMutableArray *newArr = [NSMutableArray new];
  73. [newArr addObject:otherArr];
  74. [newArr addObject:doneArr];
  75. [newArr addObject:failArr];
  76. self->_databaseArr = newArr;
  77. [lock unlock];
  78. complete(self->_databaseArr);
  79. }];
  80. // [lock unlock];
  81. });
  82. }
  83. //把TZAssetModel 转成 我们需要上传的model
  84. - (void)handlTZAssetModelToUploadFileDataFunBy:(NSMutableArray*)indexPathsForSelectedItems complete:(custom_complete_Arr)complete
  85. {
  86. if(!indexPathsForSelectedItems && indexPathsForSelectedItems.count == 0){
  87. complete(nil);
  88. return;
  89. }
  90. if(ksharedAppDelegate.DisabledFileTransferType){
  91. if(ksharedAppDelegate.isImageNewFor130){
  92. [[iToast makeText:NSLocalizedString(@"File_Transfer_Disable_tip",nil)] show];
  93. }
  94. else{
  95. [[iToast makeText:NSLocalizedString(@"File_Transfer_Disable_tip2",nil)] show];
  96. }
  97. complete(_databaseArr);
  98. return;
  99. }
  100. if(!_fileModelDataArr){
  101. _fileModelDataArr = [NSMutableArray new];
  102. }
  103. _isNewAddTaskType = YES;
  104. self.curUploadModelNumbers = indexPathsForSelectedItems.count;
  105. NSMutableArray *newUploadTaskArr = [NSMutableArray new];
  106. NSLock *lock = [NSLock new];
  107. for (TZAssetModel * model in indexPathsForSelectedItems) {
  108. uploadFileDataModel * curModel = [uploadFileDataModel new];
  109. curModel.asset = model.asset;
  110. curModel.localIdentifier = model.asset.localIdentifier;
  111. BOOL isRepeatingTasksType = NO;
  112. [lock lock];
  113. //查询上传任务是否存在
  114. if(_databaseArr && _databaseArr.count == 3 ){
  115. NSMutableArray *uploadingArr = [[NSMutableArray alloc] initWithArray:_databaseArr[0]];
  116. for (uploadFileDataModel *preModel in uploadingArr) {
  117. if([preModel.localIdentifier isEqualToString:curModel.localIdentifier]){
  118. isRepeatingTasksType = YES;
  119. break;
  120. }
  121. }
  122. }
  123. [lock unlock];
  124. if(isRepeatingTasksType){
  125. continue;
  126. }
  127. curModel.imageData = model.imageData;
  128. curModel.videoData = model.videoData;
  129. curModel.filename = [model.asset valueForKey:@"filename"];
  130. //curModel.curUploadStateType = uploadStateWait;
  131. curModel.curUploadStateType = uploadStateUploading;
  132. if(model.type == TZAssetModelMediaTypeVideo){
  133. curModel.curUploadFileType = uploadFileTypeVideo;
  134. [cachesFileManager getFileNameWithContent:curModel.videoData fileName:curModel.filename type:uploadFileTypeVideo];
  135. //curModel.totalBytes = [model.videoData length];
  136. curModel.totalBytes = model.totalBytes;
  137. curModel.videoData = [NSData new];//视频文件存储到文件后内存清空
  138. NSString *imgName1 = [curModel.filename stringByReplacingOccurrencesOfString:@"." withString:@"_"];
  139. curModel.videoFirstImageName = [[NSString alloc] initWithFormat:@"%@.png",imgName1];
  140. PHImageRequestOptions *options = [[PHImageRequestOptions alloc] init];
  141. options.networkAccessAllowed = YES; // Allow downloading from iCloud
  142. options.version = PHImageRequestOptionsVersionCurrent;
  143. options.deliveryMode = PHImageRequestOptionsDeliveryModeHighQualityFormat;
  144. //第一帧图片
  145. [[PHImageManager defaultManager] requestImageDataForAsset:curModel.asset options:options resultHandler:^(NSData * _Nullable imageData, NSString * _Nullable dataUTI, UIImageOrientation orientation, NSDictionary * _Nullable info) {
  146. // 直接得到最终的 NSData 数据
  147. if (imageData) {
  148. curModel.imageData = imageData;
  149. [cachesFileManager getFileNameWithContent:curModel.imageData fileName:curModel.videoFirstImageName type:uploadFileTypeImage];
  150. }
  151. }];
  152. }
  153. else{
  154. curModel.curUploadFileType = uploadFileTypeImage;
  155. curModel.totalBytes = model.totalBytes;
  156. if(curModel.imageData)
  157. {
  158. [cachesFileManager getFileNameWithContent:curModel.imageData fileName:curModel.filename type:uploadFileTypeImage];
  159. }
  160. else{
  161. // [[PHImageManager defaultManager] requestImageDataForAsset:curModel.asset options:nil resultHandler:^(NSData * _Nullable imageData, NSString * _Nullable dataUTI, UIImageOrientation orientation, NSDictionary * _Nullable info) {
  162. // // 直接得到最终的 NSData 数据
  163. // if (imageData) {
  164. // curModel.imageData = imageData;
  165. // [cachesFileManager getFileNameWithContent:curModel.imageData fileName:curModel.filename type:uploadFileTypeImage];;
  166. // }
  167. // }];
  168. }
  169. }
  170. //保存到数据库
  171. curModel.bg_tableName = upLoadFile_image_tableName;
  172. [newUploadTaskArr addObject:curModel];
  173. }
  174. KWeakSelf
  175. if(newUploadTaskArr.count == 0){
  176. [self getDataInDatabaseFun:YES complete:^(NSMutableArray * _Nonnull Arr) {
  177. complete(Arr);
  178. [weakSelf handleUploadFileModelBg_idFun:Arr];
  179. }];
  180. }
  181. else{
  182. [uploadFileDataModel bg_saveOrUpdateArrayAsync:newUploadTaskArr complete:^(BOOL isSuccess) {
  183. [weakSelf getDataInDatabaseFun:YES complete:^(NSMutableArray * _Nonnull Arr) {
  184. complete(Arr);
  185. [weakSelf handleUploadFileModelBg_idFun:Arr];
  186. }];
  187. }];
  188. }
  189. }
  190. #pragma mark 处理当前的model 加上bg_id
  191. - (void)handleUploadFileModelBg_idFun:(NSMutableArray*)totalArr
  192. {
  193. if(!totalArr || totalArr.count != 3){
  194. return;
  195. }
  196. _fileModelDataArr = [NSMutableArray arrayWithArray:totalArr[0]];
  197. [self beginUploadFileFun];
  198. }
  199. - (void)beginUploadFileFun
  200. {
  201. // if(_isUploadIngType && _reUploadIngSelectIndex <= 0){
  202. // return;
  203. // }
  204. BOOL isCanUseCellular = [HWDataManager getBoolWithKey:stringKeyAddSn(Const_file_Transfe_canUse_Cellular_all)];
  205. if(!isCanUseCellular){//不允许流量上传
  206. //
  207. if([AFNetworkReachabilityManager sharedManager].networkReachabilityStatus == AFNetworkReachabilityStatusReachableViaWWAN){
  208. mainBlock(^{
  209. [[NSNotificationCenter defaultCenter] postNotificationName:uploadFileSuspendAllNotification object:nil];
  210. [[iToast makeText:NSLocalizedString(@"File_Transfer_By_Cellular_tip",nil)] show];
  211. });
  212. return;
  213. }
  214. }
  215. if(ksharedAppDelegate.DisabledFileTransferType){
  216. if(ksharedAppDelegate.isImageNewFor130){
  217. [[iToast makeText:NSLocalizedString(@"File_Transfer_Disable_tip",nil)] show];
  218. }
  219. else{
  220. [[iToast makeText:NSLocalizedString(@"File_Transfer_Disable_tip2",nil)] show];
  221. }
  222. return;
  223. }
  224. if(_isSuspendType)
  225. {
  226. return;
  227. }
  228. _isUploadIngType = YES;
  229. _curUploadFileDataModel.imageData = nil;
  230. _curUploadFileDataModel.videoData = nil;
  231. _curUploadFileDataModel = nil;
  232. if(_fileModelDataArr.count == 0){
  233. [self didUploadAllTaskDoneFun];
  234. return;
  235. }
  236. if(_reUploadIngSelectIndex > 0 && _reUploadIngSelectIndex < _fileModelDataArr.count){
  237. _curUploadFileDataModel = _fileModelDataArr[_reUploadIngSelectIndex];
  238. }
  239. else
  240. {
  241. _curUploadFileDataModel = _fileModelDataArr.firstObject;
  242. //_curUploadFileDataModel = _fileModelDataArr.lastObject;
  243. }
  244. //做个异常处理
  245. if(_databaseArr.count == 3 && _isNewAddTaskType){
  246. NSMutableArray *uploadIngArr = _databaseArr[0];
  247. if(_fileModelDataArr.count != uploadIngArr.count){
  248. _databaseArr[0] = [NSMutableArray arrayWithArray:_fileModelDataArr];
  249. }
  250. }
  251. if(!_curUploadFileDataModel){
  252. return;
  253. }
  254. if(_curUploadFileDataModel.curUploadFileType == uploadFileTypeImage){
  255. NSString*pathStr = [cachesFileManager getFilePathWithName:_curUploadFileDataModel.filename type:_curUploadFileDataModel.curUploadFileType];
  256. _curUploadFileDataModel.imageData = [NSData dataWithContentsOfFile:pathStr];
  257. if(_curUploadFileDataModel.imageData && _curUploadFileDataModel.imageData.length >0){
  258. if(_curUploadFileDataModel.totalBytes == 0){
  259. _curUploadFileDataModel.totalBytes = _curUploadFileDataModel.imageData.length;
  260. }
  261. [[webSocketManager shareInstance] beginGotoUploadFileFunBy:_curUploadFileDataModel];
  262. return;
  263. }
  264. }
  265. else{
  266. //NSString*pathStr = [cachesFileManager getFilePathWithName:_curUploadFileDataModel.filename type:uploadFileTypeVideo];
  267. if([cachesFileManager checkFileIsSaveState:_curUploadFileDataModel.filename withType:uploadFileTypeVideo]){
  268. [[webSocketManager shareInstance] beginGotoUploadFileFunBy:_curUploadFileDataModel];
  269. return;
  270. }
  271. }
  272. if(!_curUploadFileDataModel.asset){
  273. NSString *curLocalIdentifier = _curUploadFileDataModel.localIdentifier;
  274. PHFetchResult *fetchResult = [PHAsset fetchAssetsWithLocalIdentifiers:@[curLocalIdentifier] options:nil];
  275. PHAsset *asset = fetchResult.firstObject;
  276. _curUploadFileDataModel.asset = asset;
  277. }
  278. if(!_curUploadFileDataModel.asset){
  279. [self getDataWrongToChangeFailFun];
  280. return;
  281. }
  282. KWeakSelf
  283. if(_curUploadFileDataModel.curUploadFileType == uploadFileTypeImage)
  284. {
  285. if(!_curUploadFileDataModel.imageData || _curUploadFileDataModel.imageData.length == 0){
  286. PHImageRequestOptions *options = [[PHImageRequestOptions alloc] init];
  287. options.networkAccessAllowed = YES; // Allow downloading from iCloud
  288. options.version = PHImageRequestOptionsVersionCurrent;
  289. options.deliveryMode = PHImageRequestOptionsDeliveryModeHighQualityFormat;
  290. [[PHImageManager defaultManager] requestImageDataForAsset:_curUploadFileDataModel.asset options:options resultHandler:^(NSData * _Nullable imageData, NSString * _Nullable dataUTI, UIImageOrientation orientation, NSDictionary * _Nullable info) {
  291. // 直接得到最终的 NSData 数据
  292. if (imageData) {
  293. self->_curUploadFileDataModel.imageData = imageData;
  294. [weakSelf afterGetImageDataFun];
  295. }
  296. else{
  297. [weakSelf getDataWrongToChangeFailFun];
  298. }
  299. }];
  300. }
  301. }
  302. else if(_curUploadFileDataModel.curUploadFileType == uploadFileTypeVideo){
  303. PHImageRequestOptions *options = [[PHImageRequestOptions alloc] init];
  304. options.networkAccessAllowed = YES; // Allow downloading from iCloud
  305. options.version = PHImageRequestOptionsVersionCurrent;
  306. options.deliveryMode = PHImageRequestOptionsDeliveryModeHighQualityFormat;
  307. //第一帧图片
  308. [[PHImageManager defaultManager] requestImageDataForAsset:_curUploadFileDataModel.asset options:options resultHandler:^(NSData * _Nullable imageData, NSString * _Nullable dataUTI, UIImageOrientation orientation, NSDictionary * _Nullable info) {
  309. // 直接得到最终的 NSData 数据
  310. if (imageData) {
  311. self->_curUploadFileDataModel.imageData = imageData;
  312. [weakSelf afterGetImageDataInVideoFun];
  313. }
  314. }];
  315. //真正的视频数据
  316. PHVideoRequestOptions *options2 = [[PHVideoRequestOptions alloc] init];
  317. options2.version = PHVideoRequestOptionsVersionOriginal;
  318. options2.networkAccessAllowed = YES; // Allow downloading from iCloud
  319. [[PHImageManager defaultManager] requestAVAssetForVideo:_curUploadFileDataModel.asset options:options2 resultHandler:^(AVAsset *asset, AVAudioMix *audioMix, NSDictionary *info) {
  320. if ([asset isKindOfClass:[AVURLAsset class]]) {
  321. AVURLAsset* urlAsset = (AVURLAsset*)asset;
  322. BOOL isSuc = [cachesFileManager copyVideoItemAtPath:[urlAsset.URL path] fileName:self->_curUploadFileDataModel.filename error:nil];
  323. //NSData *videoData = [NSData dataWithContentsOfURL:urlAsset.URL];
  324. if (isSuc) {
  325. //self->_curUploadFileDataModel.videoData = videoData;
  326. [weakSelf afterGetVideoDataFun];
  327. }
  328. else{
  329. [weakSelf getDataWrongToChangeFailFun];
  330. }
  331. }
  332. else{
  333. [weakSelf getDataWrongToChangeFailFun];
  334. }
  335. }];
  336. }
  337. }
  338. #pragma mark 获取数据失败 报错
  339. - (void)getDataWrongToChangeFailFun
  340. {
  341. [self changeUploadFileState:uploadStateFail withDidUploadBytes:_curUploadFileDataModel.didUploadBytes withModel:_curUploadFileDataModel complete:^(BOOL isSuccess) {
  342. }];
  343. }
  344. #pragma mark 根据 asset 获取到图片数据
  345. - (void)afterGetImageDataFun
  346. {
  347. [cachesFileManager getFileNameWithContent:_curUploadFileDataModel.imageData fileName:_curUploadFileDataModel.filename type:uploadFileTypeImage];
  348. [[webSocketManager shareInstance] beginGotoUploadFileFunBy:_curUploadFileDataModel];
  349. }
  350. - (void)afterGetImageDataInVideoFun
  351. {
  352. [cachesFileManager getFileNameWithContent:_curUploadFileDataModel.imageData fileName:_curUploadFileDataModel.videoFirstImageName type:uploadFileTypeImage];
  353. }
  354. - (void)afterGetVideoDataFun
  355. {
  356. [cachesFileManager getFileNameWithContent:_curUploadFileDataModel.videoData fileName:_curUploadFileDataModel.filename type:uploadFileTypeVideo];
  357. _curUploadFileDataModel.videoData = nil;
  358. [[webSocketManager shareInstance] beginGotoUploadFileFunBy:_curUploadFileDataModel];
  359. }
  360. //修改文件上传的状态
  361. - (void)changeUploadFileState:(uploadStateType)curUploadStateType withDidUploadBytes:(long)didUpLoadBytes withModel:(uploadFileDataModel*)model complete:(custom_complete_B)complete
  362. {
  363. if(model.bg_id.integerValue != _curUploadFileDataModel.bg_id.integerValue){
  364. complete(YES);
  365. return;
  366. }
  367. HLog(@"上传任务列表:%@",_fileModelDataArr);
  368. self.taskRenewTime = [iTools getNowTimeStamp];
  369. _curUploadFileDataModel.curUploadStateType = curUploadStateType;
  370. _curUploadFileDataModel.didUploadBytes = didUpLoadBytes;
  371. if(curUploadStateType == uploadStateDone){
  372. NSString *uploadDefaultPath = [HWDataManager getStringWithKey:stringKeyAddSn(Const_photo_upload_default_path)];
  373. uploadDefaultPath = [iTools changePathToShowPathBy:uploadDefaultPath];
  374. _curUploadFileDataModel.savePath = uploadDefaultPath;
  375. _curUploadFileDataModel.videoData = [NSData new];
  376. if(_curUploadFileDataModel.curUploadFileType == uploadFileTypeVideo){
  377. [cachesFileManager removeItemAtPath:_curUploadFileDataModel.filename type:uploadFileTypeVideo error:nil];
  378. }
  379. [self->_fileModelDataArr removeObject:self->_curUploadFileDataModel];
  380. [self handleDatabaseArrDeteleInUploading:_curUploadFileDataModel];
  381. [self handleDatabaseArrByInDone:_curUploadFileDataModel];
  382. }
  383. else if(curUploadStateType == uploadStateFail){
  384. [self->_fileModelDataArr removeObject:self->_curUploadFileDataModel];
  385. [self handleDatabaseArrDeteleInUploading:_curUploadFileDataModel];
  386. [self handleDatabaseArrByInFail:_curUploadFileDataModel];
  387. }
  388. [_curUploadFileDataModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  389. //HLog(@"%@ 写入 %@", model.filename, isSuccess ? @"成功":@"失败");
  390. }];
  391. if(!_isSuspendType || curUploadStateType != uploadStateSuspend){
  392. [[NSNotificationCenter defaultCenter] postNotificationName:uploadFileRefreshNotification object:model];
  393. }
  394. complete(YES);
  395. return;
  396. // NSString* where = nil;
  397. //
  398. // where = [NSString stringWithFormat:@"where %@=%@ ",bg_sqlKey(@"bg_id"),bg_sqlValue(model.bg_id)];
  399. // //HLog(@"ffff:%@",_curUploadFileDataModel.bg_id);
  400. //
  401. // KWeakSelf
  402. // dispatch_async(dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT,0), ^{
  403. // [uploadFileDataModel bg_findAsync:upLoadFile_image_tableName where:where complete:^(NSArray * _Nullable array) {
  404. // for (uploadFileDataModel * curModel in array) {
  405. // curModel.curUploadStateType = curUploadStateType;
  406. // curModel.didUploadBytes = didUpLoadBytes;
  407. //
  408. // if(curUploadStateType == uploadStateDone){
  409. // curModel.videoData = [NSData new];
  410. //
  411. // if(curModel.curUploadFileType == uploadFileTypeVideo){
  412. // [cachesFileManager removeItemAtPath:curModel.filename type:uploadFileTypeVideo error:nil];
  413. // }
  414. //
  415. // [weakSelf handleDatabaseArrByDelete:curModel];
  416. // [weakSelf handleDatabaseArrByInDone:curModel];
  417. // }
  418. // else if(curUploadStateType == uploadStateFail){
  419. // [weakSelf handleDatabaseArrByDelete:curModel];
  420. // [weakSelf handleDatabaseArrByInFail:curModel];
  421. // }
  422. //
  423. // [curModel bg_saveOrUpdateAsync:^(BOOL isSuccess) {
  424. // //HLog(@"%@ 写入 %@", model.filename, isSuccess ? @"成功":@"失败");
  425. // }];
  426. //
  427. // }
  428. //
  429. // complete(YES);
  430. // }];
  431. // });
  432. //
  433. }
  434. //修改文件上传的状态
  435. - (void)changeUploadState:(uploadStateType)curUploadStateType withModelArr:(NSArray*)modelArr complete:(custom_complete_B)complete
  436. {
  437. for (uploadFileDataModel * model in modelArr) {
  438. model.curUploadStateType = curUploadStateType;
  439. }
  440. if(modelArr.count == 0){
  441. return;
  442. }
  443. [uploadFileDataModel bg_saveOrUpdateArrayAsync:modelArr complete:^(BOOL isSuccess) {
  444. complete(YES);
  445. }];
  446. }
  447. - (void)handleDatabaseArrByDelete:(uploadFileDataModel*)delModel
  448. {
  449. if(_databaseArr && _databaseArr.count == 3 ){
  450. NSMutableArray *uploadingArr = nil;
  451. if(delModel.curUploadStateType == uploadStateUploading
  452. ||delModel.curUploadStateType == uploadStateSuspend){
  453. uploadingArr = _databaseArr[0];
  454. }
  455. else if(delModel.curUploadStateType == uploadStateDone)
  456. {
  457. uploadingArr = _databaseArr[1];
  458. }
  459. else{
  460. uploadingArr = _databaseArr[2];
  461. }
  462. [uploadingArr removeObject:delModel];
  463. }
  464. }
  465. - (void)handleDatabaseArrDeteleInUploading:(uploadFileDataModel*)uploadmodel
  466. {
  467. NSLock *lock = [NSLock new];
  468. [lock lock];
  469. if(!uploadmodel){
  470. [lock unlock];
  471. return;
  472. }
  473. if(_databaseArr && _databaseArr.count == 3 ){
  474. NSMutableArray *uploadingArr = _databaseArr[0];
  475. NSInteger count = uploadingArr.count;
  476. [uploadingArr removeObject:uploadmodel];
  477. NSInteger atferDelCount = uploadingArr.count;
  478. if(count == atferDelCount){
  479. for (uploadFileDataModel*model in uploadingArr) {
  480. if(model.bg_id.integerValue == uploadmodel.bg_id.integerValue){
  481. [uploadingArr removeObject:model];
  482. break;
  483. }
  484. }
  485. }
  486. NSInteger atferDelCount2 = uploadingArr.count;
  487. if(count == atferDelCount2){
  488. //HLog(@"\n\n\nerror:上传中的任务没有删除掉database\n\n\n\n");
  489. }
  490. }
  491. [lock unlock];
  492. }
  493. - (void)handleDatabaseArrByInDone:(uploadFileDataModel*)uploadmodel
  494. {
  495. NSLock *lock = [NSLock new];
  496. [lock lock];
  497. if(_databaseArr && _databaseArr.count == 3 ){
  498. NSMutableArray *uploadDoneArr = _databaseArr[1];
  499. if(uploadmodel){
  500. [uploadDoneArr insertObject:uploadmodel atIndex:0];
  501. }
  502. }
  503. [lock unlock];
  504. }
  505. - (void)handleDatabaseArrByInFail:(uploadFileDataModel*)uploadmodel
  506. {
  507. NSLock *lock = [NSLock new];
  508. [lock lock];
  509. if(_databaseArr && _databaseArr.count == 3 ){
  510. NSMutableArray *uploadFailArr = _databaseArr[2];
  511. if(uploadmodel){
  512. [uploadFailArr insertObject:uploadmodel atIndex:0];
  513. }
  514. }
  515. [lock unlock];
  516. }
  517. - (void)handleDatabaseArrAddModelInUploading:(uploadFileDataModel*)uploadmodel
  518. {
  519. NSLock *lock = [NSLock new];
  520. [lock lock];
  521. if(_databaseArr && _databaseArr.count == 3 ){
  522. NSMutableArray *uploadlingArr = _databaseArr[0];
  523. if(uploadmodel){
  524. [uploadlingArr insertObject:uploadmodel atIndex:0];
  525. }
  526. }
  527. [lock unlock];
  528. }
  529. //暂停上传完成
  530. - (void)suspendUploadFileFun:(BOOL)isSuspendAll withModel:(uploadFileDataModel*)model
  531. {
  532. if(!_fileModelDataArr || !_curUploadFileDataModel){
  533. return;
  534. }
  535. if(isSuspendAll){
  536. [[webSocketManager shareInstance] suspendUploadFileFun];
  537. for (uploadFileDataModel*model in _fileModelDataArr) {
  538. model.curUploadStateType = uploadStateSuspend;
  539. }
  540. _isSuspendType = YES;
  541. _isUploadIngType = NO;
  542. [self changeUploadState:uploadStateSuspend withModelArr:_fileModelDataArr complete:^(BOOL isSuccess) {
  543. [self->_fileModelDataArr removeAllObjects];
  544. }];
  545. }
  546. else{
  547. uploadFileDataModel *curModel = _curUploadFileDataModel;
  548. if(model){
  549. curModel = model;
  550. }
  551. else{
  552. [[webSocketManager shareInstance] suspendUploadFileFun];
  553. }
  554. curModel.curUploadStateType = uploadStateSuspend;
  555. [self changeUploadState:uploadStateSuspend withModelArr:@[curModel] complete:^(BOOL isSuccess) {
  556. [self->_fileModelDataArr removeObject:model];
  557. if(curModel.bg_id.integerValue == self->_curUploadFileDataModel.bg_id.integerValue){
  558. //开启下一个任务
  559. [self beginUploadFileFun];
  560. }
  561. }];
  562. }
  563. }
  564. //某个文件重新上传
  565. - (void)reUploadFileFunBy:(NSMutableArray*)Arr withAll:(BOOL)isAllType
  566. {
  567. if(!_fileModelDataArr){
  568. _fileModelDataArr = [NSMutableArray new];
  569. }
  570. _isSuspendType = NO;
  571. // for (uploadFileDataModel*model in Arr) {
  572. // model.curUploadStateType = uploadStateUploading;
  573. // }
  574. [self changeUploadState:uploadStateUploading withModelArr:Arr complete:^(BOOL isSuccess) {
  575. [self reUploadFileSecondBy:Arr withAll:isAllType];
  576. }];
  577. }
  578. - (void)reUploadFileSecondBy:(NSMutableArray*)Arr withAll:(BOOL)isAllType{
  579. if(isAllType)
  580. {
  581. _isNewAddTaskType = YES;
  582. _fileModelDataArr = [NSMutableArray arrayWithArray:Arr];
  583. _reUploadIngSelectIndex = -1;
  584. if(_fileModelDataArr.count >0){
  585. [[NSNotificationCenter defaultCenter] postNotificationName:uploadFileRefreshNotification object:_fileModelDataArr.firstObject];
  586. }
  587. [self beginUploadFileFun];
  588. return;
  589. }
  590. _isNewAddTaskType = NO;
  591. _reUploadIngSelectIndex = 0;
  592. NSMutableArray *curArr = [NSMutableArray arrayWithArray:Arr];
  593. //_fileModelDataArr = curArr;
  594. for (uploadFileDataModel*addModel in curArr){
  595. [_fileModelDataArr insertObject:addModel atIndex:0];
  596. }
  597. [self beginUploadFileFun];
  598. }
  599. - (void)uploadFileDoneFun
  600. {
  601. long totalSizeByte = _curUploadFileDataModel.totalBytes;
  602. [self changeUploadFileState:uploadStateDone withDidUploadBytes:totalSizeByte withModel:_curUploadFileDataModel complete:^(BOOL isSuccess) {
  603. self->_isUploadIngType = NO;
  604. if(self->_fileModelDataArr.count > 0){
  605. [self beginUploadFileFun];
  606. }
  607. else{
  608. [self didUploadAllTaskDoneFun];
  609. }
  610. }];
  611. }
  612. //文件上传失败
  613. - (void)uploadFileFailFun
  614. {
  615. [self changeUploadFileState:uploadStateFail withDidUploadBytes:_curUploadFileDataModel.didUploadBytes withModel:_curUploadFileDataModel complete:^(BOOL isSuccess) {
  616. if(self->_fileModelDataArr.count > 0){
  617. [self beginUploadFileFun];
  618. }
  619. else{
  620. [self didUploadAllTaskDoneFun];
  621. }
  622. }];
  623. mainBlock(^{
  624. //[[iToast makeText:NSLocalizedString(@"File_upload_fail",nil)] show];
  625. });
  626. }
  627. //删除本地数据库记录
  628. - (void)deleteUploadFileRecordBy:(NSMutableArray *)delArr withDelCache:(BOOL)isDelCache complete:(custom_complete_B)complete
  629. {
  630. //逻辑待优化
  631. BOOL isSuc = NO;
  632. BOOL isDelUploadingModel = NO;
  633. NSMutableArray *curDelArr = [NSMutableArray arrayWithArray:delArr];
  634. for (uploadFileDataModel *uploadFileDataMod in curDelArr) {
  635. NSMutableString* where = [[NSMutableString alloc] initWithString:@"where "];
  636. NSString *curStr = [NSString stringWithFormat:@"%@=%@ ",bg_sqlKey(@"bg_id"),bg_sqlValue(uploadFileDataMod.bg_id)];
  637. [where appendString:curStr];
  638. isSuc = [uploadFileDataModel bg_delete:upLoadFile_image_tableName where:where];
  639. [self handleDatabaseArrByDelete:uploadFileDataMod];
  640. //删除本地图片
  641. if(isSuc && isDelCache){
  642. //判断是否可以删除本地缓存
  643. if(_databaseArr && _databaseArr.count == 3 ){
  644. NSMutableArray *uploadingArr = _databaseArr[0];
  645. NSMutableArray *uploadDoneArr = _databaseArr[1];
  646. NSMutableArray *uploadFailArr = _databaseArr[2];
  647. BOOL isNeedDel = YES;
  648. for (uploadFileDataModel *baseUploadFileDataMod in uploadingArr) {
  649. if([uploadFileDataMod.filename isEqualToString:baseUploadFileDataMod.filename] ){
  650. isNeedDel = NO;
  651. break;
  652. }
  653. }
  654. if(isNeedDel){
  655. for (uploadFileDataModel *baseUploadFileDataMod in uploadDoneArr) {
  656. if([uploadFileDataMod.filename isEqualToString:baseUploadFileDataMod.filename] ){
  657. isNeedDel = NO;
  658. break;
  659. }
  660. }
  661. }
  662. if(isNeedDel){
  663. for (uploadFileDataModel *baseUploadFileDataMod in uploadFailArr) {
  664. if([uploadFileDataMod.filename isEqualToString:baseUploadFileDataMod.filename] ){
  665. isNeedDel = NO;
  666. break;
  667. }
  668. }
  669. }
  670. if(isNeedDel){
  671. if(uploadFileDataMod.curUploadFileType == uploadFileTypeVideo){
  672. [cachesFileManager removeItemAtPath:uploadFileDataMod.videoFirstImageName type:uploadFileTypeImage error:nil];
  673. [cachesFileManager removeItemAtPath:uploadFileDataMod.filename type:uploadFileTypeVideo error:nil];
  674. }
  675. else{
  676. [cachesFileManager removeItemAtPath:uploadFileDataMod.filename type:uploadFileTypeImage error:nil];
  677. }
  678. }
  679. }
  680. }
  681. if(isSuc && !isDelCache){//上传报云机已经存在文件了 查询本地是否有任务记录
  682. [self handleRetryUploadAndDelRecordFun:uploadFileDataMod];
  683. }
  684. //是否在删除上传中的任务
  685. if(_fileModelDataArr && _fileModelDataArr.count >0)
  686. {
  687. for (uploadFileDataModel *curUploadFileDataMod in _fileModelDataArr)
  688. {
  689. if(curUploadFileDataMod.bg_id.integerValue == uploadFileDataMod.bg_id.integerValue){
  690. [_fileModelDataArr removeObject:curUploadFileDataMod];
  691. //[self handleDatabaseArrDeteleInUploading:curUploadFileDataMod];
  692. break;
  693. }
  694. }
  695. }
  696. if(!isDelUploadingModel){
  697. if(_curUploadFileDataModel.bg_id.integerValue == uploadFileDataMod.bg_id.integerValue){
  698. isDelUploadingModel = YES;
  699. [[webSocketManager shareInstance] suspendUploadFileFun];
  700. }
  701. }
  702. }
  703. if(isDelUploadingModel){
  704. _isUploadIngType = NO;
  705. if(_fileModelDataArr.count >=1){
  706. [self beginUploadFileFun];
  707. }
  708. else{
  709. _curUploadFileDataModel = nil;
  710. _isSuspendType = NO;
  711. //_curUploadFileDataModel.curUploadStateType = uploadStateDone;
  712. }
  713. }
  714. uploadFileDataModel *RefreshUploadFileDataMod = nil;
  715. if(delArr.count >= 1){
  716. RefreshUploadFileDataMod = delArr.firstObject;
  717. }
  718. [[NSNotificationCenter defaultCenter] postNotificationName:uploadFileRefreshNotification object:RefreshUploadFileDataMod];
  719. complete(YES);
  720. }
  721. #pragma mark 处理重复上传文件的问题
  722. - (void)handleRetryUploadAndDelRecordFun:(uploadFileDataModel *)uploadFileDataMod
  723. {
  724. NSLock *lock = [NSLock new];
  725. [lock lock];
  726. if(!_databaseArr && _databaseArr.count != 3){
  727. return;
  728. }
  729. NSMutableArray * doneArr = _databaseArr[1];
  730. NSMutableArray * failArr = _databaseArr[2];
  731. BOOL isNeedDelType = YES;
  732. for (uploadFileDataModel *doneModel in doneArr) {
  733. if([doneModel.localIdentifier isEqualToString:uploadFileDataMod.localIdentifier]){
  734. isNeedDelType = NO;
  735. break;
  736. }
  737. }
  738. if(isNeedDelType){
  739. for (uploadFileDataModel *failModel in failArr) {
  740. if([failModel.localIdentifier isEqualToString:uploadFileDataMod.localIdentifier]){
  741. isNeedDelType = NO;
  742. break;
  743. }
  744. }
  745. }
  746. if(isNeedDelType){
  747. if(uploadFileDataMod.curUploadFileType == uploadFileTypeVideo){
  748. [cachesFileManager removeItemAtPath:uploadFileDataMod.videoFirstImageName type:uploadFileTypeImage error:nil];
  749. [cachesFileManager removeItemAtPath:uploadFileDataMod.filename type:uploadFileTypeVideo error:nil];
  750. }
  751. else{
  752. [cachesFileManager removeItemAtPath:uploadFileDataMod.filename type:uploadFileTypeImage error:nil];
  753. }
  754. }
  755. else{//视频文件是要删除的
  756. if(uploadFileDataMod.curUploadFileType == uploadFileTypeVideo){
  757. [cachesFileManager removeItemAtPath:uploadFileDataMod.filename type:uploadFileTypeVideo error:nil];
  758. }
  759. }
  760. [lock unlock];
  761. }
  762. //检查是否需要重新上传
  763. - (void)checkReUploadFileFun
  764. {
  765. if(!_curUploadFileDataModel
  766. || _curUploadFileDataModel.curUploadStateType == uploadStateDone
  767. || _curUploadFileDataModel.curUploadStateType == uploadStateSuspend){
  768. return;
  769. }
  770. if(_isSuspendType){
  771. return;
  772. }
  773. NSInteger curTime = [iTools getNowTimeStamp];
  774. if(self.taskRenewTime == 0){
  775. return;
  776. }
  777. if(curTime - self.taskRenewTime < 10){
  778. return;
  779. }
  780. [[webSocketManager shareInstance] beginGotoUploadFileFunBy:_curUploadFileDataModel];
  781. }
  782. - (BOOL)checkUploadingFun
  783. {
  784. if(!_curUploadFileDataModel){
  785. return NO;
  786. }
  787. if(_isSuspendType){
  788. return NO;
  789. }
  790. return YES;
  791. }
  792. - (void)checkHadUploadTaskWithComplete:(custom_complete_B)complete
  793. {
  794. [self getDataInDatabaseFun:NO complete:^(NSMutableArray * _Nonnull Arr) {
  795. mainBlock(^{
  796. if(!Arr || Arr.count != 3){
  797. complete(NO);
  798. }
  799. else{
  800. NSArray *firstArr = Arr[0];
  801. if(firstArr.count >0){
  802. complete(YES);
  803. }
  804. else{
  805. complete(NO);
  806. }
  807. }
  808. });
  809. }];
  810. }
  811. - (void)didUploadAllTaskDoneFun
  812. {
  813. self->_isUploadIngType = NO;
  814. self->_curUploadFileDataModel = nil;
  815. [[NSNotificationCenter defaultCenter] postNotificationName:uploadFileAllTaskDoneNotification object:nil];
  816. }
  817. @end