RAUploadManager.m 24 KB

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