RAUploadManager.m 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771
  1. //
  2. // RAUploadManager.m
  3. // test_autolayout
  4. //
  5. // Created by Ray on 02/05/2017.
  6. // Copyright © 2017 USAI. All rights reserved.
  7. //
  8. #import <UIKit/UIKit.h>
  9. #import "RAUploadManager.h"
  10. #import "NetworkUtils.h"
  11. #import "RAUtils.h"
  12. NSString *const kUploadSetting = @"UploadSettingKey";
  13. NSString *const UPLOAD_FINISH_NOTIFICATION = @"upload_finish_notification";
  14. NSString *const UPLOAD_COUNT_CHANGE_NOTIFICATION = @"upload_count_change_notification";
  15. NSString *const UPLOAD_REFRESH_UI_NOTIFICATION = @"upload_refresh_ui_notification";
  16. static RAUploadManager *sharedInstance;
  17. @interface RAUploadManager ()
  18. @property (nonatomic,strong) NSTimer *retryTimer;
  19. @property (nonatomic,strong) NSMutableArray<RAUploadOperation *> *retryArr;
  20. @property (nonatomic,assign) BOOL isNotReachable;
  21. @end
  22. @interface RAUPloadManagerConfigure ()
  23. @end
  24. @implementation RAUPloadManagerConfigure
  25. - (instancetype)init {
  26. if (self = [super init]) {
  27. NSDictionary *param = [[NSUserDefaults standardUserDefaults] objectForKey:kUploadSetting];
  28. if (param) {
  29. BOOL autoUpload = [[param objectForKey:@"auto_upload"] boolValue];
  30. BOOL autoRmFinish = [[param objectForKey:@"auto_rm_finish"] boolValue];
  31. BOOL autoRmErr = [[param objectForKey:@"auto_rm_error"] boolValue];
  32. BOOL onlyWiFi = [[param objectForKey:@"only_wifi"] boolValue];
  33. BOOL monitorNetwork = [[param objectForKey:@"monitor_network"] boolValue];
  34. BOOL compressImage = [[param objectForKey:@"compress_img"] boolValue];
  35. int retryCount = [[param objectForKey:@"retry_count"] intValue];
  36. NSNumber *timeInterval = [param objectForKey:@"retry_time_interval"];
  37. if (timeInterval != nil) {
  38. self.retryTimeIntetval = [timeInterval intValue];
  39. } else {
  40. self.retryTimeIntetval = 300;
  41. }
  42. self.autoUpload = autoUpload;
  43. self.autoRemoveFinish = autoRmFinish;
  44. self.autoRemoveError = autoRmErr;
  45. self.retryCount = retryCount;
  46. self.onlyWIFI = onlyWiFi;
  47. self.monitorNetwork = monitorNetwork;
  48. self.compressImage = compressImage;
  49. } else {
  50. self.autoUpload = YES;
  51. self.autoRemoveFinish = NO;
  52. self.autoRemoveError = NO;
  53. self.retryCount = 5;
  54. self.onlyWIFI = NO;
  55. self.retryTimeIntetval = 300;
  56. self.monitorNetwork = YES;
  57. self.compressImage = YES;
  58. }
  59. }
  60. return self;
  61. }
  62. - (void)save {
  63. NSDictionary *param = @{
  64. @"auto_upload" : @(self.autoUpload),
  65. @"auto_rm_finish" : @(self.autoRemoveFinish),
  66. @"auto_rm_error" : @(self.autoRemoveError),
  67. @"retry_count" : @(self.retryCount),
  68. @"only_wifi" : @(self.onlyWIFI),
  69. @"retry_time_interval" : @(self.retryTimeIntetval),
  70. @"monitor_network" : @(self.monitorNetwork),
  71. @"compress_img" : @(self.compressImage)
  72. };
  73. [[NSUserDefaults standardUserDefaults] setObject:param forKey:kUploadSetting];
  74. [[NSUserDefaults standardUserDefaults] synchronize];
  75. if (sharedInstance) {
  76. sharedInstance.autoStart = self.autoUpload;
  77. sharedInstance.removeFinish = self.autoRemoveFinish;
  78. // sharedInstance.removeError = self.autoRemoveError;
  79. sharedInstance.maxRetry = self.retryCount;
  80. sharedInstance.onlyWiFi = self.onlyWIFI;
  81. sharedInstance.monitorNetwork = self.monitorNetwork;
  82. sharedInstance.retryTimeInterval = self.retryTimeIntetval;
  83. }
  84. }
  85. @end
  86. @implementation RAUploadManager
  87. + (void)configureUploadManager:(void(^)(RAUPloadManagerConfigure *configure))configure {
  88. if (configure) {
  89. RAUPloadManagerConfigure *cfg = [[RAUPloadManagerConfigure alloc] init];
  90. configure(cfg);
  91. [cfg save];
  92. }
  93. }
  94. + (instancetype)sharedManager {
  95. static dispatch_once_t token;
  96. dispatch_once(&token, ^{
  97. sharedInstance = [[RAUploadManager alloc] init];
  98. });
  99. return sharedInstance;
  100. }
  101. /*!
  102. * Called by Reachability whenever status changes.
  103. */
  104. - (void) reachabilityChanged:(NSNotification *)note
  105. {
  106. [self handleReachability:self.reach];
  107. }
  108. - (void)handleReachability:(Reachability *)reachability {
  109. NetworkStatus netStatus = [reachability currentReachabilityStatus];
  110. if (self.monitorNetwork) {
  111. if (netStatus == NotReachable) { // 网络断开
  112. self.isNotReachable = YES;
  113. [self.retryArr removeAllObjects];
  114. [self stopAllTasks];
  115. } else if ((self.onlyWiFi && netStatus != ReachableViaWiFi)) { // 网络连接正常,但是onlyWifi
  116. self.isNotReachable = NO;
  117. [self.retryArr removeAllObjects];
  118. [self stopAllTasks];
  119. } else { // 网络正常连接
  120. if (self.isNotReachable) { // 重启任务
  121. [self restartAllTask];
  122. }
  123. self.isNotReachable = NO;
  124. }
  125. } else {
  126. if ((self.onlyWiFi && netStatus != ReachableViaWiFi)) {
  127. DebugLog(@"not wifi & stop task");
  128. [self.retryArr removeAllObjects];
  129. [self stopAllTasks];
  130. }
  131. }
  132. }
  133. - (instancetype)init {
  134. if (self = [super init]) {
  135. /*
  136. Observe the kNetworkReachabilityChangedNotification. When that notification is posted, the method reachabilityChanged will be called.
  137. */
  138. [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(reachabilityChanged:) name:kReachabilityChangedNotification object:nil];
  139. NSString *remoteHostName = @"www.apple.com";
  140. self.reach = [Reachability reachabilityWithHostName:remoteHostName];
  141. [self.reach startNotifier];
  142. [self handleReachability:self.reach];
  143. self.queue_status = QueueStatusDefault;
  144. NSDictionary *param = [[NSUserDefaults standardUserDefaults] objectForKey:kUploadSetting];
  145. if (param) {
  146. BOOL autoUpload = [[param objectForKey:@"auto_upload"] boolValue];
  147. BOOL autoRmFinish = [[param objectForKey:@"auto_rm_finish"] boolValue];
  148. // BOOL autoRmErr = [[param objectForKey:@"auto_rm_error"] boolValue];
  149. BOOL onlyWiFi = [[param objectForKey:@"only_wifi"] boolValue];
  150. int retryCount = [[param objectForKey:@"retry_count"] intValue];
  151. NSNumber *timeInterval = [param objectForKey:@"retry_time_interval"];
  152. BOOL monitorNetwork = [[param objectForKey:@"monitor_network"] boolValue];
  153. if (timeInterval != nil) {
  154. self.retryTimeInterval = [timeInterval intValue];
  155. } else {
  156. self.retryTimeInterval = 300;
  157. }
  158. self.autoStart = autoUpload;
  159. self.removeFinish = autoRmFinish;
  160. // self.removeError = autoRmErr;
  161. self.maxRetry = retryCount;
  162. self.onlyWiFi = onlyWiFi;
  163. self.monitorNetwork = monitorNetwork;
  164. } else {
  165. self.autoStart = YES;
  166. self.removeFinish = NO;
  167. // self.removeError = NO;
  168. self.maxRetry = 5;
  169. self.onlyWiFi = NO;
  170. self.retryTimeInterval = 300;
  171. self.monitorNetwork = YES;
  172. }
  173. self.newtaskStatus = TaskStatusWait;
  174. self.maxThread = 3;
  175. // self.autoStart = true;
  176. // self.removeError=true;
  177. // self.removeFinish = true;
  178. self.operation_queue = [NSOperationQueue new];
  179. self.operation_queue.maxConcurrentOperationCount = self.maxThread;
  180. [self.operation_queue addObserver:self
  181. forKeyPath:@"operationCount"
  182. options:(NSKeyValueObservingOptionNew|NSKeyValueObservingOptionOld)
  183. context:@"queueFinished changed"];
  184. //load task 要在监听arr_queue之前
  185. [self loadTasks];
  186. [self addObserver:self
  187. forKeyPath:@"arr_queue"
  188. options:(NSKeyValueObservingOptionNew|NSKeyValueObservingOptionOld)
  189. context:@"arr_queue changed"];
  190. // self.arr_queue_lock = [NSLock new];
  191. // [self.arr_queue_lock lock];
  192. // [self.arr_queue_lock lock];
  193. // [self.arr_queue_lock unlock];
  194. // [self.arr_queue_lock unlock];
  195. // self.backgroundColor = [UIColor clearColor];
  196. }
  197. return self;
  198. }
  199. -(void) dealloc
  200. {
  201. [self destructRetryTimer];
  202. [[NSNotificationCenter defaultCenter] removeObserver:self name:kReachabilityChangedNotification object:nil];
  203. [self.operation_queue removeObserver:self forKeyPath:@"operationCount"];
  204. [self.operation_queue removeObserver:self forKeyPath:@"arr_queue"];
  205. }
  206. - (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context
  207. {
  208. NSLog(@"ovserver!!! %@",keyPath);
  209. // [self saveTasks];
  210. if([keyPath isEqualToString:@"operationCount"])
  211. {
  212. // [self checkRetryOperations];
  213. if(self.operation_queue.operationCount == 0 && self.retryArr.count == 0)
  214. {
  215. for(NSMutableDictionary* task in self.arr_queue)
  216. {
  217. if([task[@"status"] intValue] == TaskStatusError)
  218. {
  219. self.queue_status = QueueStatusFinishWithError;
  220. return;
  221. }
  222. }
  223. self.queue_status = QueueStatusFinish;
  224. }
  225. }
  226. if([keyPath isEqualToString:@"arr_queue"])
  227. {
  228. NSArray* newqueue=[change objectForKey:NSKeyValueChangeNewKey];
  229. NSArray* oldqueue=[change objectForKey:NSKeyValueChangeOldKey];
  230. if([newqueue isKindOfClass:[NSNull class]])
  231. newqueue=nil;
  232. if([oldqueue isKindOfClass:[NSNull class]])
  233. oldqueue=nil;
  234. if(newqueue.count>oldqueue.count)
  235. {
  236. self.queue_status = QueueStatusAdd;
  237. return;
  238. }
  239. [[NSNotificationCenter defaultCenter] postNotificationName:UPLOAD_COUNT_CHANGE_NOTIFICATION object:nil];
  240. }
  241. }
  242. //-(void) upload:(NSString*) url
  243. //{
  244. // while(self.activeThread<self.maxThread)
  245. // {
  246. // bool nomore=false;
  247. // for(int i=0;i<self.arr_queue.count;i++)
  248. // {
  249. // NSMutableDictionary* task = self.arr_queue[i];
  250. //
  251. // TaskStatus ts=[task[@"status"] intValue];
  252. // if (ts==TaskStatusWait) {
  253. // self.activeThread--;
  254. // [self uploadTask:task url:url];
  255. // break;
  256. // }
  257. // if(i==self.arr_queue.count-1)
  258. // nomore=true;
  259. //
  260. // }
  261. // if(nomore)
  262. // break;
  263. // }
  264. //}
  265. -(void) addTask:(NSMutableDictionary*) task
  266. {
  267. // [self.arr_queue addObject:task];
  268. // [self.arr_queue_lock lock];
  269. // [[self mutableArrayValueForKey:@"arr_queue"] addObject:task];
  270. // [self.arr_queue_lock unlock];
  271. @synchronized(self) {
  272. // [[self mutableArrayValueForKey:@"arr_queue"] addObject:task];
  273. [[self mutableArrayValueForKey:@"arr_queue"] insertObject:task atIndex:0];
  274. // [self saveTasks];
  275. }
  276. if(self.newtaskStatus==TaskStatusWait)
  277. {
  278. [self startTask:task];
  279. }
  280. // self.queue_status = QueueStatusAdd;
  281. }
  282. -(void) addTasks:(NSMutableArray*) tasks
  283. {
  284. // [self.arr_queue_lock lock];
  285. // [[self mutableArrayValueForKey:@"arr_queue"] addObjectsFromArray:tasks];
  286. // [self.arr_queue_lock unlock];
  287. @synchronized(self) {
  288. // [[self mutableArrayValueForKey:@"arr_queue"] addObjectsFromArray:tasks];
  289. // NSIndexSet *indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange([self mutableArrayValueForKey:@"arr_queue"].count,tasks.count )];
  290. NSIndexSet *indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange(0,tasks.count )];
  291. [[self mutableArrayValueForKey:@"arr_queue"] insertObjects:tasks atIndexes:indexSet];
  292. }
  293. if(self.newtaskStatus==TaskStatusWait)
  294. {
  295. for(NSMutableDictionary* task in tasks)
  296. [self startTask:task];
  297. }
  298. // self.queue_status = QueueStatusAdd;
  299. }
  300. -(void) removeTasks:(NSMutableArray*) tasks
  301. {
  302. for(NSMutableDictionary* task in tasks)
  303. {
  304. [self stopTask:task];
  305. // 文件上传成功才移除文件
  306. // NSFileManager* fmanager = [NSFileManager new];
  307. // NSError* error = nil;
  308. // bool bdel=[fmanager removeItemAtPath:[self filePath:task] error:&error];
  309. // if(!bdel)
  310. // DebugLog(@"file delete failed path:%@",[self filePath:task]);
  311. }
  312. // [self.arr_queue_lock lock];
  313. // [[self mutableArrayValueForKey:@"arr_queue"] removeObjectsInArray:tasks];
  314. // [self.arr_queue_lock unlock];
  315. @synchronized(self) {
  316. // [[self mutableArrayValueForKey:@"arr_queue"] removeObjectsInArray:tasks];
  317. // NSIndexSet *indexSet = [NSIndexSet indexSetWithIndexesInRange:NSMakeRange([self mutableArrayValueForKey:@"arr_queue"].count,tasks.count )];
  318. NSIndexSet *indexSet = [self.arr_queue indexesOfObjectsPassingTest:^BOOL(id _Nonnull obj, NSUInteger idx, BOOL * _Nonnull stop)
  319. {
  320. return [tasks containsObject:obj];
  321. // return [tasks indexOfObject:obj];
  322. }];
  323. // [NSIndexSet new];
  324. // for(NSMutableDictionary* task in tasks)
  325. // {
  326. //
  327. // [self.arr_queue indexOfObject:task];
  328. // indexSet set
  329. // }
  330. [[self mutableArrayValueForKey:@"arr_queue"] removeObjectsAtIndexes:indexSet];
  331. // -removeObjectFrom<Key>AtIndex
  332. // [ insertObjects:tasks atIndexes:indexSet];
  333. }
  334. // if (self.delegate && [self.delegate respondsToSelector:@selector(uploadManager:didRemoveTasks:)]) {
  335. // [self.delegate uploadManager:self didRemoveTasks:tasks];
  336. // }
  337. }
  338. -(void) removeTask:(NSMutableDictionary*) task
  339. {
  340. // [self.arr_queue addObject:task];
  341. [self stopTask:task];
  342. // 上传成功才删除文件
  343. // NSFileManager* fmanager = [NSFileManager new];
  344. // NSError* error = nil;
  345. // bool bdel=[fmanager removeItemAtPath:[self filePath:task] error:&error];
  346. // if(!bdel)
  347. // DebugLog(@"file delete failed path:%@",[self filePath:task]);
  348. // [self.arr_queue_lock lock];
  349. // DebugLog(@"lock");
  350. // [[self mutableArrayValueForKey:@"arr_queue"] removeObject:task];
  351. // DebugLog(@"unlock");
  352. // [self.arr_queue_lock unlock];
  353. // [self.arr_queue_lock lock];
  354. // [[self mutableArrayValueForKey:@"arr_queue"] removeObject:task];
  355. // [self.arr_queue_lock unlock];
  356. @synchronized(self) {
  357. [[self mutableArrayValueForKey:@"arr_queue"] removeObject:task];
  358. // [self saveTasks];
  359. }
  360. // if (self.delegate && [self.delegate respondsToSelector:@selector(uploadManager:didRemoveTasks:)]) {
  361. // [self.delegate uploadManager:self didRemoveTasks:@[task]];
  362. // }
  363. }
  364. //-(NSString*)filePath:(NSMutableDictionary*)task
  365. //{
  366. //
  367. // NSString *path = [RAUtils appCacheDirectory];
  368. // path= [path stringByAppendingPathComponent:task[@"path"]];
  369. // path= [path stringByAppendingPathComponent:task[@"file"]];
  370. // return path;
  371. //}
  372. -(void) saveTasks
  373. {
  374. @synchronized(self) {
  375. NSUserDefaults *defaults =[NSUserDefaults standardUserDefaults];
  376. // NSMutableArray* save_arr = [NSMutableArray new];
  377. // for(NSMutableDictionary* task in self.arr_queue)
  378. // {
  379. // NSMutableDictionary* savetask = [task mutableCopy];
  380. // savetask[@"operation"] = nil;
  381. // if([savetask[@"status"] intValue]==TaskStatusWait)
  382. // savetask[@"status"]=[NSNumber numberWithInteger:TaskStatusStop];
  383. // [save_arr addObject:savetask];
  384. // }
  385. [defaults setObject:self.arr_queue forKey:@"upload_queue"];
  386. [defaults synchronize];
  387. }
  388. }
  389. -(void) loadTasks
  390. {
  391. [self stopAllTasks];
  392. self.arr_queue = [NSMutableArray new];
  393. NSMutableArray* arr = [[[NSUserDefaults standardUserDefaults] objectForKey:@"upload_queue"] mutableCopy];
  394. // if(arr==nil)
  395. for(NSDictionary* task in arr)
  396. {
  397. NSMutableDictionary* newtask =[task mutableCopy];
  398. [self.arr_queue addObject:newtask];
  399. if([newtask[@"status"] intValue]==TaskStatusWait)
  400. newtask[@"status"] =[NSNumber numberWithInteger:TaskStatusStop];
  401. if(self.autoStart&&[task[@"status"] intValue]!=TaskStatusError&&[task[@"status"] intValue]!=TaskStatusFinish)
  402. [self startTask:newtask];
  403. }
  404. // [self.operation_queue start];
  405. // NSDictionary* ddd = arr[0] ;
  406. // NSMutableDictionary* eee = [ddd mutableCopy];
  407. //
  408. // eee=[NSMutableDictionary dictionaryWithDictionary:ddd];
  409. // if(arr!=nil)
  410. // self.arr_queue=arr;
  411. // else
  412. // self.arr_queue= [[NSMutableArray alloc] init];
  413. //
  414. // for(__strong NSMutableDictionary* task in arr)
  415. // {
  416. //
  417. // task=[NSMutableDictionary dictionaryWithDictionary:task];
  418. //// if([task[@"status"] intValue]==TaskStatusStart)
  419. //// task[@"status"]=[NSNumber numberWithInteger:TaskStatusWait];
  420. // // NSMutableDictionary *new_task = [task mutableCopy];
  421. // if(self.autoStart&&[task[@"status"] intValue]!=TaskStatusError&&[task[@"status"] intValue]!=TaskStatusFinish)
  422. // [self startTask:task];
  423. // }
  424. }
  425. -(void) stopAllTasks
  426. {
  427. for(NSMutableDictionary* task in self.arr_queue)
  428. {
  429. [self stopTask:task];
  430. }
  431. }
  432. -(void) stopTask:(NSMutableDictionary*) task
  433. {
  434. if([task[@"status"] intValue]!=TaskStatusStart&&[task[@"status"] intValue]!=TaskStatusWait)
  435. {
  436. task[@"operation"]=nil;
  437. return; // task 已停止,返回
  438. }
  439. DebugLog(@"stop task");
  440. RAUploadOperation* operation = task[@"operation"];
  441. [operation cancel];
  442. task[@"operation"]=nil;
  443. task[@"status"] = [NSNumber numberWithInt:TaskStatusStop] ;
  444. task[@"progress"] = [NSNumber numberWithDouble:0.0];
  445. task[@"msg"]= @"";
  446. // [self saveTasks];
  447. if (operation.updateUI) { // 网络状态改变时,首先是正在进行的网络活动中断,然后才通知网络改变
  448. operation.updateUI();
  449. }
  450. }
  451. -(void) startAllTasks
  452. {
  453. for(NSMutableDictionary* task in self.arr_queue)
  454. [self startTask:task];
  455. }
  456. -(void) startTask:(NSMutableDictionary*) task
  457. {
  458. if (self.onlyWiFi && self.reach.currentReachabilityStatus != ReachableViaWiFi) {
  459. return;
  460. }
  461. /*__block*/ NSMutableDictionary* block_task = task;
  462. __weak typeof(self) weakSelf = self;
  463. if([task[@"status"] intValue]==TaskStatusStart||[task[@"status"] intValue]==TaskStatusWait)
  464. return; // task 已在执行队列,返回
  465. RAUploadOperation* operation = [[RAUploadOperation alloc] initWithTaskinfo:task retry:self.maxRetry];
  466. [operation setCompletionBlock:^{
  467. // [self saveTasks];
  468. NSLog(@"%@ %@",block_task[@"name"],block_task[@"status"]);
  469. switch ([block_task[@"status"] intValue]) {
  470. case TaskStatusFinish:
  471. {
  472. if(weakSelf.removeFinish)
  473. [self removeTask:block_task];
  474. [[NSNotificationCenter defaultCenter] postNotificationName:UPLOAD_FINISH_NOTIFICATION object:nil userInfo:block_task];
  475. }
  476. break;
  477. case TaskStatusError:
  478. {
  479. // if(weakSelf.removeError)
  480. // [self removeTask:block_task];
  481. }
  482. break;
  483. default:
  484. break;
  485. }
  486. }];
  487. task[@"msg"] = nil;
  488. task[@"status"] = [NSNumber numberWithInt:self.newtaskStatus] ;
  489. // [self saveTasks];
  490. task[@"operation"] = operation;
  491. [self.operation_queue addOperation:operation];
  492. dispatch_async(dispatch_get_main_queue(), ^{
  493. [[NSNotificationCenter defaultCenter] postNotificationName:UPLOAD_REFRESH_UI_NOTIFICATION object:nil];
  494. });
  495. }
  496. - (void)restartAllTask {
  497. for(NSMutableDictionary* task in self.arr_queue) {
  498. TaskStatus status = [task[@"status"] intValue];
  499. if (status == TaskStatusWait || status == TaskStatusStop) {
  500. [self startTask:task];
  501. }
  502. }
  503. }
  504. //- (void) uploadTask:(NSMutableDictionary*) task url:(NSString*)url{
  505. //
  506. //// __weak typeof(self) weakself = self;
  507. //
  508. //
  509. //
  510. //
  511. //}
  512. - (BOOL)canUpload {
  513. if (self.onlyWiFi && self.reach.currentReachabilityStatus != ReachableViaWiFi) {
  514. return NO;
  515. }
  516. return YES;
  517. }
  518. - (BOOL)reachable {
  519. return self.reach.currentReachabilityStatus != NotReachable;
  520. }
  521. #pragma mark - Retry
  522. - (NSMutableArray<RAUploadOperation *> *)retryArr {
  523. if (!_retryArr) {
  524. _retryArr = [NSMutableArray array];
  525. }
  526. return _retryArr;
  527. }
  528. - (void)addRetryOperation:(RAUploadOperation *)operation {
  529. if (self.retryTimer == nil) {
  530. // self.retryTimer = [NSTimer scheduledTimerWithTimeInterval:60 target:self selector:@selector(checkRetryOperations) userInfo:nil repeats:YES];
  531. self.retryTimer = [NSTimer timerWithTimeInterval:60 target:self selector:@selector(checkRetryOperations) userInfo:nil repeats:YES];
  532. [[NSRunLoop mainRunLoop] addTimer:self.retryTimer forMode:NSRunLoopCommonModes];
  533. }
  534. [self.retryArr addObject:operation];
  535. // 升序
  536. [self.retryArr sortUsingComparator:^NSComparisonResult(RAUploadOperation* _Nonnull operation1, RAUploadOperation* _Nonnull opertation2) {
  537. if (operation1.theRestOfWaitTimeInterval < opertation2.theRestOfWaitTimeInterval) {
  538. return NSOrderedAscending;
  539. } else {
  540. return NSOrderedDescending;
  541. }
  542. }];
  543. }
  544. - (void)checkRetryOperations {
  545. // 检测是否有网络连接
  546. if (![self reachable]) {
  547. // 当前task状态为Waiting,是否考虑 monitoreNetwork == YES 的条件下修改为Stop?
  548. return;
  549. }
  550. // 检查是否有需要重试的任务
  551. if (self.retryArr.count == 0) {
  552. return;
  553. }
  554. // 检查Wi-Fi Only
  555. if (![self canUpload]) {
  556. return;
  557. }
  558. NSMutableArray<RAUploadOperation *> *nextStartOperations = [NSMutableArray array];
  559. // // 检查当前上传队列等待数量是否小于最大并发数
  560. // if (self.operation_queue.operationCount < self.maxThread) {
  561. //
  562. //// // 根据最大并发数将剩余Operation全部添加进去
  563. //// NSInteger minCount = MIN(self.retryArr.count, self.maxThread - self.operation_queue.operationCount);
  564. // for (int i = 0; i < self.retryArr.count; i++) {
  565. // RAUploadOperation *operation = [self.retryArr objectAtIndex:i];
  566. // operation.queuePriority = NSOperationQueuePriorityVeryHigh;
  567. // [self.operation_queue addOperation:operation];
  568. // [nextStartOperations addObject:operation];
  569. // }
  570. //
  571. // } else {
  572. // 调度等待时间到达的Operation
  573. __weak typeof(self) weakSelf = self;
  574. [self.retryArr enumerateObjectsUsingBlock:^(RAUploadOperation * _Nonnull operation, NSUInteger idx, BOOL * _Nonnull stop) {
  575. if (operation.theRestOfWaitTimeInterval <= 0) {
  576. operation.queuePriority = NSOperationQueuePriorityVeryHigh;
  577. [weakSelf.operation_queue addOperation:operation];
  578. [nextStartOperations addObject:operation];
  579. }
  580. }];
  581. // }
  582. [self.retryArr removeObjectsInArray:nextStartOperations];
  583. // if (self.retryArr.count == 0) {
  584. // [self destructRetryTimer];
  585. // }
  586. }
  587. - (void)destructRetryTimer {
  588. if (self.retryTimer != nil) {
  589. [self.retryTimer setFireDate:[NSDate distantFuture]];
  590. [self.retryTimer invalidate];
  591. self.retryTimer = nil;
  592. }
  593. }
  594. @end