RAUploadManager.m 24 KB

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