NSArray、NSMutableArray、NSDictionary、NSMutableDictionary、是我们的在iOS开发中非常常用的类。当然,在享受这些类的便利的同时,它们也给我们带来一些困扰。粗心我们可能会调用addObject:传入一个nil, 也有可能是会objectAtIndex:传入一个越界的index。尤其是在数据基本依赖于服务端返回的的情况,这种crash大幅增加。最近项目上经常出现NSDictionary的setObject:forKey:的nil object的崩溃。我们希望为这种崩溃找一个解决办法。

解决方案

函数包装

我们希望能够用一个统一的方法解决粗心的程序员可能传入的nil object。我们最先想到的想法是对这些函数进行一个包装,比如objectAtIndex,我们写一个如下的函数



- (id)safeObjectAtIndex:(NSUInteger)index {   
    if (index >= self.count) {
        return nil;
    }
    return [self objectAtIndex:index];
}

以后所有调用objectAtIndex的地方统统替换为safeObjectAtIndex。 不过,这显然不是我想要的,我不希望改变现有的调用方式,大家可能是通过[]或objectAtIndex(不推荐)的方式获取数组元素,如果要做替换的话改动势必非常大,而且不便于以后的移植。我们希望代码调用objectAtIndex的时候,能够被我们先捕获到,进行处理之后再调用Cocoa的这个方法。Objective-C作为一门动态语言,有强大的动态加载的能力,提供了Method swizzling实现这样的功能。现在,黑魔法起飞。

Method swizzling

关于Method swizzling的具体实现,不打算多说,这里谈一下我遇到的问题。最初在做Method swizzling的时候,我尝试去替换NSArrat的objectAtIndex:,但我始终没有办法替换掉这个方法。后来,搞了半天才发现,我们使用的NSAarray或者NSMutableArray并不是我们所看到样子,它们是class cluster。objectAtIndex:并不是他们的方法,而是他们背后的concrete class: __NSArrayI __NSArrayM的方法。解决了这个问题,剩下的就很简单了。

 

使用

直接把XTSafeCollection.hXTSafeCollection.m拖入工程,NSArrayNSMutableArrayNSDictionaryNSMutableDictionary这些类的API以前是怎么调用的,还怎么写,完全不用修改。Demo里,我全部以传统的会引起crash的方式调用代码,以下是我的Demo的代码和输出

NSArray *array = @[@"a", @"b"];
NSMutableArray *mutableArray = [@[@"aa", @"bb"] mutableCopy];
     
// Object at index
NSLog(@"%@", array[10]);
NSLog(@"%@", mutableArray[100]);
     
// add object
[mutableArray addObject:nil];
     
// Insert object
[mutableArray insertObject:nil atIndex:0];
[mutableArray insertObject:@"cc" atIndex:10];
     
// Replace object
[mutableArray replaceObjectAtIndex:0 withObject:nil];
[mutableArray replaceObjectAtIndex:10 withObject:@"cc"];
     
// Dictionary
NSMutableDictionary *mutableDictionary = [NSMutableDictionary dictionary];
mutableDictionary[nil] = @"1";
mutableDictionary[@"1"] = nil;
2015-08-25 18:10:23.932 XTSafeCollection[29067:443479] [__NSArrayI objectAtIndex:] index {10} beyond bounds [0...1]
2015-08-25 18:10:23.933 XTSafeCollection[29067:443479] (null)
2015-08-25 18:10:23.933 XTSafeCollection[29067:443479] [__NSArrayM objectAtIndex:] index {100} beyond bounds [0...1]
2015-08-25 18:10:23.933 XTSafeCollection[29067:443479] (null)
2015-08-25 18:10:23.934 XTSafeCollection[29067:443479] [__NSArrayM addObject:], NIL object.
2015-08-25 18:10:23.934 XTSafeCollection[29067:443479] [__NSArrayM insertObject:atIndex:] NIL object.
2015-08-25 18:10:23.934 XTSafeCollection[29067:443479] [__NSArrayM insertObject:atIndex:] index {10} beyond bounds [0...1].
2015-08-25 18:10:23.934 XTSafeCollection[29067:443479] [__NSArrayM replaceObjectAtIndex:withObject:] NIL object.
2015-08-25 18:10:23.934 XTSafeCollection[29067:443479] [__NSArrayM replaceObjectAtIndex:withObject:] index {10} beyond bounds [0...1].
2015-08-25 18:10:23.934 XTSafeCollection[29067:443479] [__NSDictionaryM setObject:forKey:] NIL key.


安装

源码: https://github.com/wuwen1030/XTSafeCollection

已知问题

替换NSMuatbelArray的objectAtIndex:引起键盘展示状态态切换后台的崩溃,抛出*** -[UIKeyboardLayoutStar release]: message sent to deallocated instance 0x7f883beac9c0 在这里找到了解决方法