NSDictionary的runtime的一些探讨

上一篇文章提到，将NSDictionary 的扩展方法，通过runtime替换系统方法，以实现AOP, 即，面向切面编程，不做侵入式开发。
前文链接：
NSDictionary中读取到'null'值时候怎么办：转成nil
http://www.jianshu.com/p/2cc4023526b4

实际上也就是将 -(id)objectForKey: 方法替换成 -(id)objectForKeyNotNull:(id)key 以达到不用倒处修改代码，即可替换实现方法。
具体实现代码如下：

//
//  NSDictionary+null.h
//  DicTest
//
//

#import <Foundation/Foundation.h>

@interface NSDictionary(null)

@end

//
//  NSDictionary+null.m
//  DicTest
//
//

#import "NSDictionary+null.h"
#include <objc/runtime.h>

@implementation NSDictionary(null)

/**
一个objc程序启动后，需要进行类的初始化、调用方法时的cache初始化，再发送消息的时候就直接走缓存
（引申：+load方法和+initialize方法。load方法是首次加载类时调用，绝对只调用一次；
initialize方法是首次给类发消息时调用，通常只调用一次，
但如果它的子类初始化时未定义initialize方法，则会再调用一次它的initialize方法）。
 */
+ (void) load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        Class class = [self class];
        // 选择器
        SEL originalSEL = @selector(objectForKey:);
        //@selector(objectForKey:);
        SEL SwizzledSEL = @selector(cb_objectForKey:);
         
        /**
        class_getInstanceMethod方法
        (使用NSDictionary、NSArray，必须使用与其对应的加入内存时的元类
        __NSDictionaryI  __NSArrayI)
        __NSDictionaryI——NSDictionary
        __NSDictionaryM——NSMutableDictionary
        __NSArrayI——NSArray
        __NSArrayM——NSMutableArray
         */
        Method originalMethod = class_getInstanceMethod(NSClassFromString(@"__NSDictionaryI"), originalSEL);
        //class_getClassMethod(class, originalSEL);备注的是获取静态方法
        Method SwizzledMethod = class_getInstanceMethod(NSClassFromString(@"__NSDictionaryI"), SwizzledSEL);
        //class_getClassMethod(class, SwizzledSEL);

        // 方法的实现
        IMP originalIMP = method_getImplementation(originalMethod);
        //class_getMethodImplementation(class, originalSEL);
        IMP SwizzledIMP = method_getImplementation(SwizzledMethod);
        //class_getMethodImplementation(class, SwizzledSEL);
       
        // 是否添加成功方法:添加了初始方法，实现内容指向目标方法体
        BOOL isSuccess = class_addMethod(class, originalSEL, SwizzledIMP, method_getTypeEncoding(SwizzledMethod));
        
        if (isSuccess) {
            // 初始指向目标，那么把目标的内容指向初始
            class_replaceMethod(class, SwizzledSEL, originalIMP, method_getTypeEncoding(originalMethod));
        }
        else{
            // 没有添加成功说明已经存在，就交换
            // 注意，这里交换的是IMP 实现
            method_exchangeImplementations(originalMethod, SwizzledMethod);
        }
    });
}

- (id)cb_objectForKey:(id)aKey{
    id object = [self cb_objectForKey:aKey];
    if ([object isKindOfClass:[NSNull class]]) {
        return nil;
    }else{
        return object;
    }
}

@end

这样就完成了，将NSDictionary的系统静态方法，通过runtime，替换为cb_objectForKey 。
运行以下代码，得到了预期的结果：

    NSString *string = @"{\"errCode\":\"NA\",\"errDesc\":null}";
    NSError *error = nil;
    NSData *data = [string dataUsingEncoding:NSUTF8StringEncoding];
    NSDictionary *dic = [NSJSONSerialization JSONObjectWithData:data options:NSJSONReadingAllowFragments error:&error];
    NSString *nullS = [dic objectForKey:@"errDesc"];
    if (nullS == nil) {
        NSLog(@"nullS 是 nil");
    }else{
        NSLog(@"nullS == %@", nullS);
    }
    NSLog(@"dic == %@", [dic description]);

运行结果：

2017-07-03 19:36:21.220 DicTest[6892:477878] nullS 是 nil
2017-07-03 19:36:21.221 DicTest[6892:477878] dic == {
    errCode = NA;
    errDesc = (null);
}

但需要注意的是，
NSDictionary作为基础类，app运行的时候用到的地方很多，贸然的修改基础类，往往会遇到意想不到的问题。
在调试过程中，我就遇见了这么一个情况：
最开始，我的cb_objectForKey 方法是这样的：

- (id)cb_objectForKey:(id)aKey{
    id object = [self cb_objectForKey:aKey];
    if ([object isKindOfClass:[NSNumber class]] ||
        [object isKindOfClass:[NSString class]] ||
        [object isKindOfClass:[NSArray class]] ||
        [object isKindOfClass:[NSDictionary class]])
    {
        return object;
    }
    return @"";
}

程序启动后,app运行到一个位置后，不再继续往下走了。
断点调试都没有用，因为压根就没走到断点的位置就停下来了。
深入一想，这肯定就是加的这个NSDictionary分类的问题，将 +(void)load{...}注释掉，果然app运行恢复正常了，这也证实了猜想。
于是加了两行log：

- (id)cb_objectForKey:(id)aKey{
    NSLog(@"aKey = %@",aKey);
    id object = [self hObjectForKey:aKey];
    NSLog(@"object = %@",object);

运行结果如下：

2017-07-04 09:42:53.543 DictRuntime[7305:535790] aKey = CLASSIC
2017-07-04 09:42:53.543 DictRuntime[7305:535790] object = 0
2017-07-04 09:42:53.554 DictRuntime[7305:535790] aKey = SIMULATOR_EXTENDED_DISPLAY_PROPERTIES
2017-07-04 09:42:53.554 DictRuntime[7305:535790] object = /Users/caobo/Library/Developer/CoreSimulator/Devices/627677FC-A4B3-471C-8A37-C42F20EFD64E/data/Library/Application Support/Simulator/extended_display.plist
2017-07-04 09:42:53.556 DictRuntime[7305:535790] aKey = com.apple.frontboard.systemappservices
2017-07-04 09:42:53.556 DictRuntime[7305:535790] object = (null)
2017-07-04 09:42:53.556 DictRuntime[7305:535854] aKey = XPC_SERVICE_NAME
2017-07-04 09:42:53.557 DictRuntime[7305:535854] object = UIKitApplication:Douglas.DictRuntime[0x4713][4890]
2017-07-04 09:42:53.557 DictRuntime[7305:535790] aKey = com.apple.frontboard.workspace
2017-07-04 09:42:53.557 DictRuntime[7305:535790] object = (null)

这样，就注意到 com.apple.frontboard.systemappservices 和 com.apple.frontboard.workspace 两个key 对应的 value值是nil，但在runtime替换方法后，ruturn的是@“”，额，这就尴尬了。
于是将这个 return @""; 替换为 return nil;
再运行一次，正常了，整个都世界安静了。
这样，事实就很明显了，在app运行的时候，大量的用到了Foundation中的各种基层代码，如果贸然的用runtime来修改基础类，大概率的可能性会遇到意想不到的问题。
所以，通过runtime来修改基础类库，一定要谨慎谨慎再谨慎。
因此，最终的代码如下：

- (id)cb_objectForKey:(id)aKey{
    id object = [self cb_objectForKey:aKey];
    if ([object isKindOfClass:[NSNull class]]) {
        return nil;
    }else{
        return object;
    }
}

只是将 NSNull这种类型转换为 nil，减少出错的概率。

NSDictionary的runtime的一些探讨

推荐阅读更多精彩内容