自定义NSOperation

上一节，我对NSOperation的基本概念及使用进行了介绍，想要了解的，请点击这里。本节中，我介绍自定义NSOperation实现多线程异步下载图片，类似于SDWebImage。

自定义NSOperation的步骤很简单，重写 - (void)main方法，在里面实现想执行的任务。

重写 - (void)main方法注意点：

1.自己创建自动释放池（因为如果是异步操作，无法访问主线程的自动释放池）

2.经常通过 - (BOOL)isCancelled 方法检测操作是否被取消，对取消做出响应。

Demo最终的效果图：

上面的图片均是通过NSOperation异步下载的。

下面，我具体介绍代码的实现过程。

1.首先，我准备了数据源（data.plist文件）

2. 定义了对应的数据模型

@interface DataModel : NSObject/* 图片名称 */@property (nonatomic,copy) NSString *name;/* 图片url地址 */@property (nonatomic,copy) NSString *url;/* 下载次数 */@property (nonatomic,strong) NSNumber *downloadedCount;// 字典转化为模型- (instancetype)initWithDict:(NSDictionary *)dict;@end@implementation DataModel- (instancetype)initWithDict:(NSDictionary *)dict {    if (self = [super init]) {        // (KVC)字典转模型        [self setValuesForKeysWithDictionary:dict];    }    return  self;}@end

3. ViewController中得代码：

#import "ViewController.h"#import "DataModel.h"#import "LFDownloadOperation.h"@interface ViewController ()<LFDownloadOperationDelegate>/* 数据源 */@property (nonatomic,strong) NSMutableArray *dataLists;// Key:url, Value: UIImage (存放图片的缓存字典，有的话，直接取出使用;没有的话，下载)@property (nonatomic,strong) NSMutableDictionary *images;// Key:url, Value: LFDownloadOperation@property (nonatomic,strong) NSMutableDictionary *operations;// 下载队列，存放LFDownloadOperation@property (nonatomic,strong) NSOperationQueue *queue;@end@implementation ViewController#pragma mark - Lazy Load- (NSMutableArray *)dataLists {    if (!_dataLists) {        NSString *file = [[NSBundle mainBundle] pathForResource:@"data" ofType:@"plist"];        NSArray *dataArr = [NSArray arrayWithContentsOfFile:file];        NSMutableArray *tempLists = [NSMutableArray array];        for (NSDictionary *dict in dataArr) {            DataModel *model = [[DataModel alloc] initWithDict:dict];            [tempLists addObject:model];        }        _dataLists = tempLists;    }    return _dataLists;}- (NSMutableDictionary *)images {    if (!_images) {        _images = [NSMutableDictionary dictionary];    }    return _images;}- (NSMutableDictionary *)operations {    if( !_operations) {        _operations = [NSMutableDictionary dictionary];    }    return _operations;}- (NSOperationQueue *)queue {    if (!_queue) {        _queue = [[NSOperationQueue alloc] init];    }    return _queue;}- (void)viewDidLoad {    [super viewDidLoad];}#pragma mark - UITableView Delegate/DataSource- (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section {    return  self.dataLists.count;}- (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath {    static NSString *identifer = @"UITableViewCell";    UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:identifer];    if (!cell) {        cell = [[UITableViewCell alloc] initWithStyle:UITableViewCellStyleSubtitle reuseIdentifier:identifer];    }        DataModel *model = self.dataLists[indexPath.row];        cell.textLabel.text = model.name;    cell.detailTextLabel.text = [model.downloadedCount stringValue];        // 主要实现下载的代码    UIImage *image = self.images[model.url];    if (image) { // 图片下载完毕(在字典中)        cell.imageView.image = image;    } else {        // 创建下载Operation并设置默认占位图片        LFDownloadOperation *operation = self.operations[model.url];        cell.imageView.image = [UIImage imageNamed:@"placeholder.jpg"];                // 如果operation有值，说明正在下载        if (!operation) {            // 还没有下载，立刻创建operation并下载            operation = [[LFDownloadOperation alloc] init];            operation.delegate = self;            operation.url = model.url;            operation.indexPath = indexPath;            // 将每张图片的下载地址和每个opearion建立对应关系            self.operations[model.url] = operation;            // 不能在这里创建队列，应该只创建一个队列(所以使用懒加载)            //NSOperationQueue *queue = [[NSOperationQueue alloc] init];            // 加到队列中并下载            [self.queue addOperation:operation];        }    }        return  cell;}- (CGFloat)tableView:(UITableView *)tableView heightForRowAtIndexPath:(NSIndexPath *)indexPath {    return 130;}#pragma mark - LFDownloadOperationDelegate- (void)downloadOperation:(LFDownloadOperation *)operation didFinishedWithImage:(UIImage *)image {    NSIndexPath *indexPath = operation.indexPath;    DataModel *model = self.dataLists[indexPath.row];    // 当一张图片下载完成了，则应该移除这张图片对应的operation    [self.operations removeObjectForKey:model.url];    // 将下载完成的图片保存到images这个字典缓存池中    self.images[model.url] = image;    // 每次有图片下载完成，则刷新对应的cell并显示图片    [self.tableView reloadRowsAtIndexPaths:@[indexPath] withRowAnimation:UITableViewRowAnimationNone];}@end

4. 自定义的NSOperation

@class LFDownloadOperation;@protocol LFDownloadOperationDelegate <NSObject>@optional/* 代理返回下载完毕的图片 */- (void)downloadOperation:(LFDownloadOperation *)operation didFinishedWithImage:(UIImage *)image;@end@interface LFDownloadOperation : NSOperation@property (nonatomic,copy) NSString *url;@property (nonatomic,strong) NSIndexPath *indexPath;@property (nonatomic,assign) id<LFDownloadOperationDelegate> delegate;@end@implementation LFDownloadOperation// NSOperation中的main就是实现相应的异步操作,所以重写父类方法- (void)main{    // 因为LFDownloadOperation已经不在主线程的自动释放池了，所以要包含在@autoreleasepool中    @autoreleasepool {                // 取消操作发生在任何时刻都有可能，因此在执行任何操作之前，先检测该操作是否已经被取消        if (self.isCancelled) {            return;        }                NSURL *url = [NSURL URLWithString:self.url];        NSData *data = [NSData dataWithContentsOfURL:url];        UIImage *image = [UIImage imageWithData:data];                if (self.isCancelled) {            return;        }                if ([self.delegate respondsToSelector:@selector(downloadOperation:didFinishedWithImage:)]){            dispatch_async(dispatch_get_main_queue(), ^{                // 最后将异步下载完成的图片返回到主线程(dispatch_get_main_queue)                [self.delegate downloadOperation:self didFinishedWithImage:image];            });        }    }}@end