RAUploadManager.m 24 KB

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