RAUploadManager.m 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777
  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. DebugLog(@"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. //
  379. @synchronized(self) {
  380. NSUserDefaults *defaults =[NSUserDefaults standardUserDefaults];
  381. // NSMutableArray* save_arr = [NSMutableArray new];
  382. // for(NSMutableDictionary* task in self.arr_queue)
  383. // {
  384. // NSMutableDictionary* savetask = [task mutableCopy];
  385. // savetask[@"operation"] = nil;
  386. // if([savetask[@"status"] intValue]==TaskStatusWait)
  387. // savetask[@"status"]=[NSNumber numberWithInteger:TaskStatusStop];
  388. // [save_arr addObject:savetask];
  389. // }
  390. [defaults setObject:self.arr_queue forKey:@"upload_queue"];
  391. [defaults synchronize];
  392. DebugLog(@"save upload queue %@",self.arr_queue);
  393. }
  394. }
  395. -(void) loadTasks
  396. {
  397. [self stopAllTasks];
  398. self.arr_queue = [NSMutableArray new];
  399. NSMutableArray* arr = [[[NSUserDefaults standardUserDefaults] objectForKey:@"upload_queue"] mutableCopy];
  400. // if(arr==nil)
  401. for(NSDictionary* task in arr)
  402. {
  403. NSMutableDictionary* newtask =[task mutableCopy];
  404. [self.arr_queue addObject:newtask];
  405. if([newtask[@"status"] intValue]==TaskStatusWait)
  406. newtask[@"status"] =[NSNumber numberWithInteger:TaskStatusStop];
  407. if(self.autoStart&&[task[@"status"] intValue]!=TaskStatusError&&[task[@"status"] intValue]!=TaskStatusFinish)
  408. [self startTask:newtask];
  409. }
  410. // [self.operation_queue start];
  411. // NSDictionary* ddd = arr[0] ;
  412. // NSMutableDictionary* eee = [ddd mutableCopy];
  413. //
  414. // eee=[NSMutableDictionary dictionaryWithDictionary:ddd];
  415. // if(arr!=nil)
  416. // self.arr_queue=arr;
  417. // else
  418. // self.arr_queue= [[NSMutableArray alloc] init];
  419. //
  420. // for(__strong NSMutableDictionary* task in arr)
  421. // {
  422. //
  423. // task=[NSMutableDictionary dictionaryWithDictionary:task];
  424. //// if([task[@"status"] intValue]==TaskStatusStart)
  425. //// task[@"status"]=[NSNumber numberWithInteger:TaskStatusWait];
  426. // // NSMutableDictionary *new_task = [task mutableCopy];
  427. // if(self.autoStart&&[task[@"status"] intValue]!=TaskStatusError&&[task[@"status"] intValue]!=TaskStatusFinish)
  428. // [self startTask:task];
  429. // }
  430. }
  431. -(void) stopAllTasks
  432. {
  433. for(NSMutableDictionary* task in self.arr_queue)
  434. {
  435. [self stopTask:task];
  436. }
  437. }
  438. -(void) stopTask:(NSMutableDictionary*) task
  439. {
  440. if([task[@"status"] intValue]!=TaskStatusStart&&[task[@"status"] intValue]!=TaskStatusWait)
  441. {
  442. task[@"operation"]=nil;
  443. return; // task 已停止,返回
  444. }
  445. DebugLog(@"stop task");
  446. RAUploadOperation* operation = task[@"operation"];
  447. [operation cancel];
  448. task[@"operation"]=nil;
  449. task[@"status"] = [NSNumber numberWithInt:TaskStatusStop] ;
  450. task[@"progress"] = [NSNumber numberWithDouble:0.0];
  451. task[@"msg"]= @"";
  452. // [self saveTasks];
  453. if (operation.updateUI) { // 网络状态改变时,首先是正在进行的网络活动中断,然后才通知网络改变
  454. operation.updateUI();
  455. }
  456. }
  457. -(void) startAllTasks
  458. {
  459. for(NSMutableDictionary* task in self.arr_queue)
  460. [self startTask:task];
  461. }
  462. -(void) startTask:(NSMutableDictionary*) task
  463. {
  464. if (self.onlyWiFi && self.reach.currentReachabilityStatus != ReachableViaWiFi) {
  465. return;
  466. }
  467. /*__block*/ NSMutableDictionary* block_task = task;
  468. __weak typeof(self) weakSelf = self;
  469. if([task[@"status"] intValue]==TaskStatusStart||[task[@"status"] intValue]==TaskStatusWait)
  470. return; // task 已在执行队列,返回
  471. RAUploadOperation* operation = [[RAUploadOperation alloc] initWithTaskinfo:task retry:self.maxRetry];
  472. [operation setCompletionBlock:^{
  473. // [self saveTasks];
  474. DebugLog(@"%@ %@",block_task[@"name"],block_task[@"status"]);
  475. switch ([block_task[@"status"] intValue]) {
  476. case TaskStatusFinish:
  477. {
  478. if(weakSelf.removeFinish)
  479. [self removeTask:block_task];
  480. [[NSNotificationCenter defaultCenter] postNotificationName:UPLOAD_FINISH_NOTIFICATION object:nil userInfo:block_task];
  481. }
  482. break;
  483. case TaskStatusError:
  484. {
  485. // if(weakSelf.removeError)
  486. // [self removeTask:block_task];
  487. }
  488. break;
  489. default:
  490. break;
  491. }
  492. }];
  493. task[@"msg"] = nil;
  494. task[@"status"] = [NSNumber numberWithInt:(int)self.newtaskStatus] ;
  495. // [self saveTasks];
  496. task[@"operation"] = operation;
  497. [self.operation_queue addOperation:operation];
  498. dispatch_async(dispatch_get_main_queue(), ^{
  499. [[NSNotificationCenter defaultCenter] postNotificationName:UPLOAD_REFRESH_UI_NOTIFICATION object:nil];
  500. });
  501. }
  502. - (void)restartAllTask {
  503. for(NSMutableDictionary* task in self.arr_queue) {
  504. TaskStatus status = [task[@"status"] intValue];
  505. if (status == TaskStatusWait || status == TaskStatusStop) {
  506. [self startTask:task];
  507. }
  508. }
  509. }
  510. //- (void) uploadTask:(NSMutableDictionary*) task url:(NSString*)url{
  511. //
  512. //// __weak typeof(self) weakself = self;
  513. //
  514. //
  515. //
  516. //
  517. //}
  518. - (BOOL)canUpload {
  519. if (self.onlyWiFi && self.reach.currentReachabilityStatus != ReachableViaWiFi) {
  520. return NO;
  521. }
  522. return YES;
  523. }
  524. - (BOOL)reachable {
  525. return self.reach.currentReachabilityStatus != NotReachable;
  526. }
  527. #pragma mark - Retry
  528. - (NSMutableArray<RAUploadOperation *> *)retryArr {
  529. if (!_retryArr) {
  530. _retryArr = [NSMutableArray array];
  531. }
  532. return _retryArr;
  533. }
  534. - (void)addRetryOperation:(RAUploadOperation *)operation {
  535. if (self.retryTimer == nil) {
  536. // self.retryTimer = [NSTimer scheduledTimerWithTimeInterval:60 target:self selector:@selector(checkRetryOperations) userInfo:nil repeats:YES];
  537. self.retryTimer = [NSTimer timerWithTimeInterval:60 target:self selector:@selector(checkRetryOperations) userInfo:nil repeats:YES];
  538. [[NSRunLoop mainRunLoop] addTimer:self.retryTimer forMode:NSRunLoopCommonModes];
  539. }
  540. [self.retryArr addObject:operation];
  541. // 升序
  542. [self.retryArr sortUsingComparator:^NSComparisonResult(RAUploadOperation* _Nonnull operation1, RAUploadOperation* _Nonnull opertation2) {
  543. if (operation1.theRestOfWaitTimeInterval < opertation2.theRestOfWaitTimeInterval) {
  544. return NSOrderedAscending;
  545. } else {
  546. return NSOrderedDescending;
  547. }
  548. }];
  549. }
  550. - (void)checkRetryOperations {
  551. // 检测是否有网络连接
  552. if (![self reachable]) {
  553. // 当前task状态为Waiting,是否考虑 monitoreNetwork == YES 的条件下修改为Stop?
  554. return;
  555. }
  556. // 检查是否有需要重试的任务
  557. if (self.retryArr.count == 0) {
  558. return;
  559. }
  560. // 检查Wi-Fi Only
  561. if (![self canUpload]) {
  562. return;
  563. }
  564. NSMutableArray<RAUploadOperation *> *nextStartOperations = [NSMutableArray array];
  565. // // 检查当前上传队列等待数量是否小于最大并发数
  566. // if (self.operation_queue.operationCount < self.maxThread) {
  567. //
  568. //// // 根据最大并发数将剩余Operation全部添加进去
  569. //// NSInteger minCount = MIN(self.retryArr.count, self.maxThread - self.operation_queue.operationCount);
  570. // for (int i = 0; i < self.retryArr.count; i++) {
  571. // RAUploadOperation *operation = [self.retryArr objectAtIndex:i];
  572. // operation.queuePriority = NSOperationQueuePriorityVeryHigh;
  573. // [self.operation_queue addOperation:operation];
  574. // [nextStartOperations addObject:operation];
  575. // }
  576. //
  577. // } else {
  578. // 调度等待时间到达的Operation
  579. __weak typeof(self) weakSelf = self;
  580. [self.retryArr enumerateObjectsUsingBlock:^(RAUploadOperation * _Nonnull operation, NSUInteger idx, BOOL * _Nonnull stop) {
  581. if (operation.theRestOfWaitTimeInterval <= 0) {
  582. operation.queuePriority = NSOperationQueuePriorityVeryHigh;
  583. [weakSelf.operation_queue addOperation:operation];
  584. [nextStartOperations addObject:operation];
  585. }
  586. }];
  587. // }
  588. [self.retryArr removeObjectsInArray:nextStartOperations];
  589. // if (self.retryArr.count == 0) {
  590. // [self destructRetryTimer];
  591. // }
  592. }
  593. - (void)destructRetryTimer {
  594. if (self.retryTimer != nil) {
  595. [self.retryTimer setFireDate:[NSDate distantFuture]];
  596. [self.retryTimer invalidate];
  597. self.retryTimer = nil;
  598. }
  599. }
  600. @end