RAUploadManager.m 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  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. @implementation RAUploadManager
  12. - (instancetype)init {
  13. if (self = [super init]) {
  14. self.arr_queue = [[NSMutableArray alloc] init];
  15. self.newtaskStatus = TaskStatusWait;
  16. self.maxThread = 3;
  17. self.removeError=true;
  18. self.removeFinish = true;
  19. self.operation_queue = [NSOperationQueue new];
  20. self.operation_queue.maxConcurrentOperationCount = self.maxThread;
  21. // self.backgroundColor = [UIColor clearColor];
  22. }
  23. return self;
  24. }
  25. //-(void) upload:(NSString*) url
  26. //{
  27. // while(self.activeThread<self.maxThread)
  28. // {
  29. // bool nomore=false;
  30. // for(int i=0;i<self.arr_queue.count;i++)
  31. // {
  32. // NSMutableDictionary* task = self.arr_queue[i];
  33. //
  34. // TaskStatus ts=[task[@"status"] intValue];
  35. // if (ts==TaskStatusWait) {
  36. // self.activeThread--;
  37. // [self uploadTask:task url:url];
  38. // break;
  39. // }
  40. // if(i==self.arr_queue.count-1)
  41. // nomore=true;
  42. //
  43. // }
  44. // if(nomore)
  45. // break;
  46. // }
  47. //}
  48. -(void) addTask:(NSMutableDictionary*) task
  49. {
  50. // [self.arr_queue addObject:task];
  51. if(self.newtaskStatus==TaskStatusWait)
  52. {
  53. [self startTask:task];
  54. }
  55. [[self mutableArrayValueForKey:@"arr_queue"] addObject:task];
  56. }
  57. -(void) addTasks:(NSMutableArray*) tasks
  58. {
  59. if(self.newtaskStatus==TaskStatusWait)
  60. {
  61. for(NSMutableDictionary* task in tasks)
  62. [self startTask:task];
  63. }
  64. [[self mutableArrayValueForKey:@"arr_queue"] addObjectsFromArray:tasks];
  65. }
  66. -(void) removeTasks:(NSMutableArray*) tasks
  67. {
  68. NSFileManager* fmanager = [NSFileManager new];
  69. for(NSMutableDictionary* task in tasks)
  70. {
  71. [self stopTask:task];
  72. NSError* error = nil;
  73. [fmanager removeItemAtPath:task[@"file_path"] error:&error];
  74. }
  75. [[self mutableArrayValueForKey:@"arr_queue"] removeObjectsInArray:tasks];
  76. }
  77. -(void) removeTask:(NSMutableDictionary*) task
  78. {
  79. NSFileManager* fmanager = [NSFileManager new];
  80. // [self.arr_queue addObject:task];
  81. [self stopTask:task];
  82. NSError* error = nil;
  83. [fmanager removeItemAtPath:task[@"file_path"] error:&error];
  84. [[self mutableArrayValueForKey:@"arr_queue"] removeObject:task];
  85. }
  86. -(void) stopTask:(NSMutableDictionary*) task
  87. {
  88. if([task[@"status"] intValue]!=TaskStatusStart&&[task[@"status"] intValue]!=TaskStatusWait)
  89. return; // task 已停止,返回
  90. RAUploadOperation* operation = task[@"operation"];
  91. [operation cancel];
  92. task[@"status"] = [NSNumber numberWithInt:TaskStatusStop] ;
  93. task[@"progress"] = [NSNumber numberWithDouble:0.0];
  94. }
  95. -(void) startTask:(NSMutableDictionary*) task
  96. {
  97. __block NSMutableDictionary* block_task = task;
  98. __weak typeof(self) weakSelf = self;
  99. if([task[@"status"] intValue]==TaskStatusStart||[task[@"status"] intValue]==TaskStatusWait)
  100. return; // task 已在执行队列,返回
  101. RAUploadOperation* operation = [[RAUploadOperation alloc] initWithTaskinfo:task retry:self.maxRetry];
  102. [operation setCompletionBlock:^{
  103. switch ([block_task[@"status"] intValue]) {
  104. case TaskStatusFinish:
  105. {
  106. if(weakSelf.removeFinish)
  107. [self removeTask:block_task];
  108. }
  109. break;
  110. case TaskStatusError:
  111. {
  112. if(weakSelf.removeError)
  113. [self removeTask:block_task];
  114. }
  115. break;
  116. default:
  117. break;
  118. }
  119. }];
  120. task[@"operation"] = operation;
  121. task[@"status"] = [NSNumber numberWithInt:self.newtaskStatus] ;
  122. [self.operation_queue addOperation:operation];
  123. }
  124. //- (void) uploadTask:(NSMutableDictionary*) task url:(NSString*)url{
  125. //
  126. //// __weak typeof(self) weakself = self;
  127. //
  128. //
  129. //
  130. //
  131. //}
  132. @end