erpsuiteios/common/upload/RAUploadManager.m

//
//  RAUploadManager.m
//  test_autolayout
//
//  Created by Ray on 02/05/2017.
//  Copyright © 2017 USAI. All rights reserved.
//
#import <UIKit/UIKit.h>
#import "RAUploadManager.h"
#import "NetworkUtils.h"
#import "RAUtils.h"
#import "UploadSettingController.h"

@implementation RAUploadManager

- (instancetype)init {
    if (self = [super init]) {
        
        NSDictionary *param = [[NSUserDefaults standardUserDefaults] objectForKey:kUploadSetting];
        if (param) {
            BOOL autoUpload = [[param objectForKey:@"auto_upload"] boolValue];
            int retryCount = [[param objectForKey:@"retry_count"] intValue];
            self.autoStart = autoUpload;
            self.maxRetry = retryCount;
        } else {
            self.autoStart = YES;
            self.maxRetry = 2;
        }

        
        [self loadTasks];

        self.newtaskStatus = TaskStatusWait;
        self.autoStart = true;
        self.maxThread = 3;
        self.removeError=true;
        self.removeFinish = true;
        
        self.operation_queue = [NSOperationQueue new];
        self.operation_queue.maxConcurrentOperationCount = self.maxThread;
//        self.backgroundColor = [UIColor clearColor];
    }
    return self;
}
//-(void) upload:(NSString*) url
//{
//    while(self.activeThread<self.maxThread)
//    {
//        bool nomore=false;
//        for(int i=0;i<self.arr_queue.count;i++)
//        {
//            NSMutableDictionary* task = self.arr_queue[i];
//            
//            TaskStatus ts=[task[@"status"] intValue];
//            if (ts==TaskStatusWait) {
//                self.activeThread--;
//                [self uploadTask:task url:url];
//                break;
//            }
//            if(i==self.arr_queue.count-1)
//                nomore=true;
//
//        }
//        if(nomore)
//            break;
//    }
//}
-(void) addTask:(NSMutableDictionary*) task
{
    
//    [self.arr_queue addObject:task];
   
    
    if(self.newtaskStatus==TaskStatusWait)
    {
        
        [self startTask:task];

        
    }
     [[self mutableArrayValueForKey:@"arr_queue"] addObject:task];
}
-(void) addTasks:(NSMutableArray*) tasks
{
    
    if(self.newtaskStatus==TaskStatusWait)
    {
        for(NSMutableDictionary* task in tasks)
            [self startTask:task];
        
        
    }
    [[self mutableArrayValueForKey:@"arr_queue"] addObjectsFromArray:tasks];
}

-(void) removeTasks:(NSMutableArray*) tasks
{
    
    NSFileManager* fmanager = [NSFileManager new];
    for(NSMutableDictionary* task in tasks)
    {
        
        [self stopTask:task];
        NSError* error = nil;
        [fmanager removeItemAtPath:[self filePath:task] error:&error];
    }
    [[self mutableArrayValueForKey:@"arr_queue"] removeObjectsInArray:tasks];
//    if (self.delegate && [self.delegate respondsToSelector:@selector(uploadManager:didRemoveTasks:)]) {
//        [self.delegate uploadManager:self didRemoveTasks:tasks];
//    }
}

-(void) removeTask:(NSMutableDictionary*) task
{
    NSFileManager* fmanager = [NSFileManager new];
    //    [self.arr_queue addObject:task];
    [self stopTask:task];
    NSError* error = nil;
    [fmanager removeItemAtPath:[self filePath:task] error:&error];
    
    [[self mutableArrayValueForKey:@"arr_queue"] removeObject:task];
//    if (self.delegate && [self.delegate respondsToSelector:@selector(uploadManager:didRemoveTasks:)]) {
//        [self.delegate uploadManager:self didRemoveTasks:@[task]];
//    }
    
    

}


-(NSString*)filePath:(NSMutableDictionary*)task
{
    
    NSString *path = [RAUtils appCacheDirectory];
    path= [path stringByAppendingPathComponent:task[@"path"]];
    path= [path stringByAppendingPathComponent:task[@"file"]];
    return path;
}
-(void) saveTasks
{
    NSUserDefaults *defaults =[NSUserDefaults standardUserDefaults];
    [defaults setObject:self.arr_queue forKey:@"upload_queue"];
    [defaults synchronize];
}
-(void) loadTasks
{
    [self stopAllTasks];
    
    NSMutableArray* arr = [[NSUserDefaults standardUserDefaults] objectForKey:@"upload_queue"];
    if(arr!=nil)
        self.arr_queue=arr;
    else
        self.arr_queue= [[NSMutableArray alloc] init];
    
    for(NSMutableDictionary* task in arr)
    {
//        if([task[@"status"] intValue]==TaskStatusStart)
//            task[@"status"]=[NSNumber numberWithInteger:TaskStatusWait];
        NSMutableDictionary *new_task = [task mutableCopy];
        if(self.autoStart&&[task[@"status"] intValue]!=TaskStatusError&&[task[@"status"] intValue]!=TaskStatusFinish)
           [self startTask:new_task];
    }
    
    
}
-(void) stopAllTasks
{
    for(NSMutableDictionary* task in self.arr_queue)
    {
        
        [self stopTask:task];
    }
    
}
-(void) stopTask:(NSMutableDictionary*) task
{
    
    if([task[@"status"] intValue]!=TaskStatusStart&&[task[@"status"] intValue]!=TaskStatusWait)
        return; // task 已停止，返回
    
    RAUploadOperation* operation = task[@"operation"];
    [operation cancel];
    task[@"operation"]=nil;
    task[@"status"] = [NSNumber numberWithInt:TaskStatusStop] ;
    task[@"progress"] = [NSNumber numberWithDouble:0.0];
    
    
}  
-(void) startAllTasks
{

        
    for(NSMutableDictionary* task in self.arr_queue)
        [self startTask:task];

}
-(void) startTask:(NSMutableDictionary*) task
{
    __block NSMutableDictionary* block_task = task;
    __weak typeof(self) weakSelf = self;

    if([task[@"status"] intValue]==TaskStatusStart||[task[@"status"] intValue]==TaskStatusWait)
        return; // task 已在执行队列，返回
    
        RAUploadOperation* operation = [[RAUploadOperation alloc] initWithTaskinfo:task retry:self.maxRetry];
    [operation setCompletionBlock:^{
        switch ([block_task[@"status"] intValue]) {
            case TaskStatusFinish:
            {
                if(weakSelf.removeFinish)
                   [self removeTask:block_task];
            }
                break;
            case TaskStatusError:
            {
                if(weakSelf.removeError)
                    [self removeTask:block_task];
            }
                break;
            default:
                break;
        }
        
    }];
        task[@"operation"] = operation;
        task[@"status"] = [NSNumber numberWithInt:self.newtaskStatus] ;
        [self.operation_queue addOperation:operation];
    
        
}
//- (void) uploadTask:(NSMutableDictionary*) task url:(NSString*)url{
//    
////    __weak typeof(self) weakself = self;
//    
//    
//  
//    
//}
@end