IOS消息机制应用实例--异常处理

Posted 2020-09-28

ios消息机制应用实例--异常处理

最近发现了一个在项目中常用的异常处的工具NullSafe，分析了它的实现原理，不小心发现了一个小Bug，现将其分享出来，关于这篇文章的Demo已经上传至GitHub，看完如有收获，欢迎Star，如有疑问欢迎issue，大家一起学习。在IOS开发中我们可能会遇到下面的情景:服务器给我们返回得某个字段是null,比如someValue:null，这个时候我们利用第三方工具转化之后会得到someValue = <null>,这个时候如果我们判断这个someValue的类型，会看到其为:NSNull。那么问题来了，如果这个someValue是要给控件赋值，比如:someLabel.text = someVlaue，这个时候相当于someLabel.text = nil,显然，是不会有问题的。但是有时候我们可能会给这个貌似是NSString的对象发送消息(因为我们在Model里定义了NSString * someValue)，比如:[someValue length]。这个时候由于null这个对象没有这个方法,也就是是说：null 这个对象不能处理这个消息所有就会Crash，让程序闪退。那么我们怎样处理来避免这种Crash呢？我们怎样处理这个消息呢？
首先我们来看一下NSObject.h中我们不常用到的几个方法,以及它们的含义:

    + (BOOL)resolveClassMethod:(SEL)sel; // 判断是否发现了这个method，如果发了，将其添加给该对象，并且返回YES,如果没有返回NO。
    + (BOOL)resolveInstanceMethod:(SEL)sel;// 同上类似

    - (id)forwardingTargetForSelector:(SEL)aSelector;  // 这个方法来指定未被识别的消息首先要指向得对象 

    + (NSMethodSignature *)instanceMethodSignatureForSelector:(SEL)aSelector;
    // 遍历一个类的实例方法得到这个消息得NSMethodSignature，这个返回值包含了对这个method相关的描述，如果这个method不能找到，那么返回nil.

     - (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector;//遍历一个类的实例方法或者类方法来得到这个消息的NSMethodSignature

    - (void)forwardInvocation:(NSInvocation *)anInvocation; // NSObject的子类可以覆盖这个方法来讲消息转发给其它的对象。当一个对象发送一个消息，但是这个对象不能响应这个消息，那么RunTime会给这个对象一个来转发这个消息的机会。
 这个对象通过创建一个NSInvocation对象作为参数来调用这个forwardInvocation方法，然后改对象会调用这个方法来将消息转发给其它对象。
 这个NSInvocation对象其实是有上一个方法methodSignatureForSelector中返回的NSMethodSignature来得到的，所以在重写这个方法之前我们必须重写methodSignatureForSelector方法。

    + (BOOL)instancesRespondToSelector:(SEL)aSelector;// 这个类的实例是否具有相应这个selector的能力，也就是说这个类有没有这样一个方法

    - (IMP)methodForSelector:(SEL)aSelector;//遍历一个类的实例方法或者类方法 得到这个方法的IMP(函数指针，指向这个方法的具体实现)

    + (IMP)instanceMethodForSelector:(SEL)aSelector;//遍历一个类的实例方法列表，得到这个方法IMP           
    - (void)doesNotRecognizeSelector:(SEL)aSelector; // 如果一个对象收到了一个消息，但是它不能处理这个消息，并且这个消息没有被转发，那么系统将会调用这个方法。
    同时这个方法会引发一个NSInvalidArgumentException，并且引发error.

那么这几个方法，在系统中是怎样的调用顺序呢？我们来看下图:

IOS中消息处理得流程

从中可以看出在给一个对象发送消息的时候，如果对象没有对应的IML，那么会调用对象所属类的

+ (BOOL)resolveInstanceMethod:(SEL)sel

方法，然后看对于这个SEL对象是否可以执行，如果不可以执行则会调用

- (id)forwardingTargetForSelector:(SEL)aSelector

来找到一个对象处理这个方法(我们可以返回一个对象，这个对象可以处理这个方法)，如果这个方法返回的是nil，那么会调用这个对象的

- (NSMethodSignature *)methodSignatureForSelector:(SEL)aSelector

这个方法，如果这方法调用之后，仍然没有找到对应的NSMethodSignature，那么会调用

- (void)doesNotRecognizeSelector:(SEL)aSelector

这个方法，并且抛出异常，如果这个时候返回了一个有效的NSMethodSignature，那么会调用

- (void)forwardInvocation:(NSInvocation *)anInvocation

到这里消息的处理结束。
那么对于NSNull这个对象，我们如何让它处理一个其自身不能处理的消息呢？这时候，我们肯定会想到，将这个消息传递给其它可以处理的对象。那么问题来了：我们如何能找到这样一个对象呢？这时候我们想到了RunTime,利用RunTime的objc_getClassList方法，我们可以获取整个项目中注册得所有类（只要在项目中添加了这个类文件，无论这个类是否被使用），这个时候我们可以过滤掉用不到的父类，以节约循环得次数，因为子类已经继承了父类的方法，所以具有处理这个消息得能力，之后首先利用上文提到的instancesRespondToSelector来判断这个类是否可以响应这个消息，如果可以响应，那么可以利用上文提到的instanceMethodSignatureForSelector来得到这个NSMethodSignature,并且返回。通过上述分析，系统会调用这个forwardInvocation,这个是时候我们调用NSInvocation的invokeWithTarget这个方法来将这个消息发送给nil,在OC中向一个nil发送任何消息都不会引起程序Crash,至此一个由于服务器返回数据异常而导致的Crash被解决了。这显然增加了系统的容错能力，在项目调试阶段，可能由于数据不完善，所以可以利用这个方法来规避Crash,但是在数据基本完善之后，我们可以去掉这种方法以便我们在程序Crash的时候，及时提醒后台人员来完善数据。

附：NullSafe的实现详解：

@implementation NSNull (NullSafe)

 - (NSMethodSignature *)methodSignatureForSelector:(SEL)selector
{
    @synchronized([self class])
    {
        // 寻找 method signature
        NSMethodSignature *signature = [super methodSignatureForSelector:selector];
        if (!signature)
        {
            //改消息不能被NSNull处理，所以我们要寻找其它的可以处理的类               
            static NSMutableSet *classList = nil;
            static NSMutableDictionary *signatureCache = nil;// 缓存这个找到的 method signature，以便下次寻找
            if (signatureCache == nil)
            {
                classList = [[NSMutableSet alloc] init];
                signatureCache = [[NSMutableDictionary alloc] init];

            // 获取项目中的所有类，并且去除有子类的类。
            // objc_getClassList：这个方法会将所有的类缓存，以及这些类的数量。我们需要提供一块足够大得缓存来存储它们，所以我们必须调用这个函数两次。第一次来判断buffer的大小，第二次来填充这个buffer。
            int numClasses = objc_getClassList(NULL, 0);             
            Class *classes = (Class *)malloc(sizeof(Class) * (unsigned long)numClasses);
            numClasses = objc_getClassList(classes, numClasses);



            NSMutableSet *excluded = [NSMutableSet set];
            for (int i = 0; i < numClasses; i++)
            {

                Class someClass = classes[i];

                // 筛选出其中含有子类的类，加入:excluded中
                Class superclass = class_getSuperclass(someClass);
                while (superclass)
                {
                    if (superclass == [NSObject class]) // 如果父类是NSObject,则跳出循环，并且加入classList
                    {
                        // 将系统中用到的所有类都加到了ClassList中
                        [classList addObject:someClass];
                        break;
                    }

                    //父类不是NSObject,将其父类添加到excluded
                    [excluded addObject:superclass];
                     superclass = class_getSuperclass(superclass);
                }
            }

            // 删除所有含有子类的类
            for (Class someClass in excluded)
            {
                [classList removeObject:someClass];
            }

            //释放内存
            free(classes);
        }

        // 首先检测缓存是否有这个实现
        NSString *selectorString = NSStringFromSelector(selector);
        signature = signatureCache[selectorString];
        if (!signature)
        {
            //找到方法的实现
            for (Class someClass in classList)
            {
                if ([someClass instancesRespondToSelector:selector])
                {
                    signature = [someClass instanceMethodSignatureForSelector:selector];
                    break;
                }
            }

            //缓存以备下次使用
            signatureCache[selectorString] = signature ?: [NSNull null];
        }
        else if ([signature isKindOfClass:[NSNull class]])
        {
            signature = nil;
        }
    }
    return signature;
    }
    }

   - (void)forwardInvocation:(NSInvocation  *)invocation
   {
    // 让nil来处理这个invocation
    [invocation invokeWithTarget:nil];

    }

  @end

在原文中作者是这样写的: [excluded addObject:NSStringFromClass(superclass)];
这样得话ClassList中存放的是Class，而excluded中存放的是String,将会失去其过滤掉不必要的类的作用，所以，我将其改为了: [excluded addObject:superclass];不知道作者是不是考虑了其他问题，也可能是由于其大意。

作者：击水湘江
链接：http://www.jianshu.com/p/b1de9404d7d9
來源：简书
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。