自定義NSOperation實現取消正在執行下載的操作

無論是掛起，還是取消全部，都無法取消正在執行的操作。

NSOperationQueue *queue = [[NSOperationQueue alloc] init];

// 移除隊列里面所有的操作，但正在執行的操作無法移除
[queue cancelAllOperations];

// 掛起隊列，使隊列任務不再執行，但正在執行的操作無法掛起
_queue.suspended = YES;

 

我們可以自定義NSOperation，實現取消正在執行的操作。其實就是攔截main方法。

main方法：

1、任何操作在執行時，首先會調用start方法，start方法會更新操作的狀態（過濾操作,如過濾掉處於“取消”狀態的操作）。

2、經start方法過濾后，只有正常可執行的操作，就會調用main方法。

3、重寫操作的入口方法(main)，就可以在這個方法里面指定操作執行的任務。

4、main方法默認是在子線程異步執行的。

 

示例代碼：

頭文件

#import <Foundation/Foundation.h>
#import <UIKit/UIKit.h>

/**
 自定義下載圖片器：下載圖片，回調給VC
 */
@interface YSDownloadOperation : NSOperation

/**
 類方法實例化自定義操作

 @param urlString 圖片地址
 @param finishBlock 完成回調
 @return 自定義操作
 */
+ (instancetype)downloadImageWithURLString:(NSString *)urlString andFinishBlock:(void(^)(UIImage*))finishBlock;

@end

.m文件

#import "YSDownloadOperation.h"

@interface YSDownloadOperation()

/**
 圖片地址
 */
@property(copy,nonatomic) NSString *urlString;

/**
 回調Block,在主線程執行
 */
@property(copy,nonatomic) void(^finishBlock)(UIImage*);

@end

@implementation YSDownloadOperation


/**
 重寫自定義操作的入口方法
 任何操作在執行時都會默認調用這個方法
 默認在子線程執行
 當隊列調度操作執行時，才會進入main方法
 */
- (void)main{
    // 默認在子線程執行
    // NSLog(@"%@",[NSThread currentThread]);
    
    NSAssert(self.urlString != nil, @"請傳入圖片地址");
    NSAssert(self.finishBlock != nil, @"請傳入下載完成回調Block");
    
    // 越晚執行越好，一般寫在耗時操作后面(可以每行代碼后面寫一句)
    if(self.isCancelled){
        return;
    }
    
    // 下載圖片
    NSURL *imgURL = [NSURL URLWithString:self.urlString];
    NSData *imgData = [NSData dataWithContentsOfURL:imgURL];
    UIImage *img = [UIImage imageWithData:imgData];
    
    // 越晚執行越好，一般寫在耗時操作后面(可以每行代碼后面寫一句)
    if(self.isCancelled){
        return;
    }
    
    // 傳遞至VC
    [[NSOperationQueue mainQueue] addOperationWithBlock:^{
        self.finishBlock(img);
    }];
}

+ (instancetype)downloadImageWithURLString:(NSString *)urlString andFinishBlock:(void(^)(UIImage*))finishBlock{
    YSDownloadOperation *op = [[self alloc] init];
    
    op.urlString = urlString;
    op.finishBlock = finishBlock;
    
    return op;
}

@end