iOS的null值处理:Method-Forwarding方式

方法源码

https://github.com/FlashHand/BuffKit/tree/master/BuffKit/NullBuff

常常会从服务器拿到蛋疼的null值，说不定就导致闪退了，考虑到数据解析后的结果大不了是NSNumber，NSString，NSDictionary，NSArray这四种，我就写了个NSNull的Extension.

像这样：

-(NSInteger)length
{
  return 0;
}
-(NSInteger)count
{
  return 0;
}
//...
添加了各种NSNumber，NSString，NSDictionary，NSArray里的对象方法。

这样可以足够覆盖实际需求了，但总归是有限的，细想了下决定用消息转发处理：

//
// Created by BoWang on 16/5/16.
// Copyright (c) 2016 BoWang. All rights reserved.
//

#import <Foundation/Foundation.h>
//NSString,NSArray,NSDictionary
@interface NSNull (NullBuff)

@end

@interface NullBuff : NSObject
@end

//
// Created by BoWang on 16/5/16.
// Copyright (c) 2016 BoWang. All rights reserved.
//

#import "NullBuff.h"
#import <objc/runtime.h>

@implementation NSNull (NullBuff)
- (id)forwardingTargetForSelector:(SEL)aSelector {
    //函数说明：class_getInstanceMethod用来获取某个类的某个selector的Method结构体
    //method_copyReturnType用来获取该方法的返回值类型，如果没有该方法则返回空。
    //处理NSNumber,NSString,NSArray,NSDictionary，这里会遍历这四个类，当returnType为真得时候，说明当前遍历到的类含有该selector.
    //由于OC良好的命名机制，同样的selector不会有不同类的返回值。
    NSArray *supporttedTypes = @[@"NSNumber" , @"NSString" , @"NSArray" , @"NSDictionary"];
    for (int i = 0; i < 4; ++i) {
        Method m = class_getInstanceMethod(NSClassFromString(supporttedTypes[i]) , aSelector);
        const char *returnType = method_copyReturnType(m);
        if (returnType) {
            NSString *returnTypeStr = [[NSString alloc] initWithCString:returnType encoding:NSUTF8StringEncoding];
            free(returnType);
            switch (i) {
                case 0:
                    return @(0);
                case 1:
                    return @"";
                case 2:
                    return @[];
                case 3:
                    return @{};
                default:
                    break;
            }
        }

    }
    return [super forwardingTargetForSelector:aSelector];
}
@end

@implementation NullBuff {

}
@end

原理很简单：NSNull是继承于NSObject，而NSObject的

• (id)forwardingTargetForSelector:(SEL)aSelector 可以被重写。

该方法的可以将消息转发给它的return value.