NSMutableDictionary / NSDictionary 向指定Key赋nil值发生崩溃的解决方案

• 发生场景及原因:
  绝大多数情况下，我们向NSNull对象发送消息，都会产生崩溃，NSNull对象常见于后台返回数据中可能会有null字段，很多JSON库都会转成NSNull对象，如下情况就会产生崩溃：

id obj = [NSNull null];
NSLog(@"%@", [objstringValue]);

对此我们利用运行时来可以重写
- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector和
- (void)forwardInvocation:(NSInvocation *)anInvocation这两个方法将没能力处理消息的方法签名转发给nil对象则不会产生崩溃

此外，常见的崩溃比如，NSArray取值越界，NSDictionary传了nil对象，这些问题产生的崩溃可以使用Runtime中的Method Swizzle,将原生的方法hook掉，如下：

#import "NSMutableDictionary+NullSafe.h"
#import 

@implementation NSMutableDictionary (NullSafe)

+ (void)load {

    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        id obj = [[self alloc] init];
        [obj swizzleMethod:@selector(setObject:forKey:)withMethod:@selector(safe_setObject:forKey:)];
    });
    
}


- (void)swizzleMethod:(SEL)origSelector withMethod:(SEL)newSelector
{
    //
    Class class = [self class];
    
    /** 得到类的实例方法 class_getInstanceMethod(Class  _Nullable __unsafe_unretained cls, SEL  _Nonnull name)
     _Nullable __unsafe_unretained cls  那个类
     _Nonnull name 按个方法
     
     补充: class_getClassMethod 得到类的 类方法
     */
    // 必须两个Method都要拿到
    Method originalMethod = class_getInstanceMethod(class, origSelector);
    Method swizzledMethod = class_getInstanceMethod(class, newSelector);

    /** 动态添加方法 class_addMethod(Class  _Nullable __unsafe_unretained cls, SEL  _Nonnull name, IMP  _Nonnull imp, const char * _Nullable types)
        class_addMethod  是相对于实现来的说的，将本来不存在于被操作的Class里的newMethod的实现添加在被操作的Class里，并使用origSel作为其选择子
     _Nonnull name  原方法选择子，
     _Nonnull imp 新方法选择子，
     
     */
    // 如果发现方法已经存在，会失败返回，也可以用来做检查用,我们这里是为了避免源方法没有实现的情况;如果方法没有存在,我们则先尝试添加被替换的方法的实现
    BOOL didAddMethod = class_addMethod(class,origSelector,method_getImplementation(swizzledMethod),method_getTypeEncoding(swizzledMethod));
    
    // 如果返回成功:则说明被替换方法没有存在.也就是被替换的方法没有被实现,我们需要先把这个方法实现,然后再执行我们想要的效果,用我们自定义的方法去替换被替换的方法. 这里使用到的是class_replaceMethod这个方法. class_replaceMethod本身会尝试调用class_addMethod和method_setImplementation，所以直接调用class_replaceMethod就可以了)
    if (didAddMethod) {
        class_replaceMethod(class,newSelector,method_getImplementation(originalMethod),method_getTypeEncoding(originalMethod));
        
    } else { // 如果返回失败:则说明被替换方法已经存在.直接将两个方法的实现交换即
        method_exchangeImplementations(originalMethod, swizzledMethod);
    }
}



- (void)safe_setObject:(id)value forKey:(NSString *)key {
    
    if (value) {
    [self safe_setObject:value forKey:key];
    
}else {
    
    NSLog(@"[NSMutableDictionarysetObject: forKey:], Object cannot be nil");
}
}

@end

这种解决方法可以避免诸如数组取值越界、字典传空值、removeObjectAtIndex等错误，如下的崩溃就可以避免：

id obj = nil;
NSMutableDictionary *m_dict =[NSMutableDictionary dictionary];
[dict setObject:obj forKey:@"666"];