uploadFileManager.m 35 KB

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