底层原理-2. iOS-KVO

313 阅读30分钟

Key-Value Observing 键值观察通过侦听特定键路径上的更改,可以在特定对象实例之间进行特殊的事件自省。

例如:一个 ProgressView 可以观察 网络请求的 numberOfBytesRead 来更新它自己的 progress 属性。
Delegate 是一个流行的传递事件的设计模式,通过定义一系列的方法来传递给指定的处理对象。
例如:UIScrollView 每次它的 scroll offset 改变的时候都会发送 scrollViewDidScroll: 
到它的代理Callbacks 不管是像 NSOperation 里的 completionBlock(当 isFinished==YES 
的时候会触发),
还是 C 里边的函数指针,传递一个函数钩子比如 SCNetworkReachabilitySetCallback(3);
在执行 addObserver:forKeyPath:options 时 runtime 会替换实列的 isa,变成 NSKVONotifyin_xxx
其 set 方法其实调用了 Fundation 框架中的 C 函数
_NSSetIntValueAndNotify
其 set 函数中,先执行 willChangeValueForKey,
再调用父类的 set 方法
最后执行 didChangeValueForKey

Follow up:
如何验证上面的这些说法?
1是在 addObserver 后用 lldb po 类
2是 %p 打印 [instance methodForSelector:@selector(setXXX:)] 的 IMP 内存地址,然后 lldb p (IMP)内存地址
如你手动触发?
手动调用 willChangeValueForKey 和 didChangeValueForKey
Follow up: 如何用 block 或 delegate 实现 KVO?
juejin.cn/post/684490…
主要是用几个 runtime 方法新建一个子类来做这些事


一、KVO 初探

根据苹果官方文档的定义,KVO (Key Value Observing) 键值观察是建立在 KVC 基础之上的,所以如果对 KVC 不是很了解的读者可以查看上一篇 KVC 底层探索的文章。

我相信大多数开发者应该对于 KVO 都能熟练掌握,不过我们还是回顾一下官网对于 KVO 的解释吧。

1.1 什么是 KVO?

KVO 提供了一种当其他对象的属性发生变化就会通知观察者对象的机制。根据官网的定义,属性的分类可以分为下列三种:

  • Attributes: 简单属性,比如基本数据类型,字符串和布尔值,而诸如 NSNumber 和其它一些不可变类型比如 NSColor 也可以被认为是简单属性
  • To-one relationships: 这些是具有自己属性的可变对象属性。即对象的属性可以更改,而无需更改对象本身。例如,一个 Account 对象可能具有一个 owner 属性,该属性是 Person 对象的实例,而 Person 对象本身具有 address 属性。owner 的地址可以更改,但却而无需更改 Account 持有的 owner 属性。也就是说 Accountowner 属性未被更改,只是 address 被更改了。
  • To-many relationships: 这些是集合对象属性。尽管也可以使用自定义集合类,但是通常使用 NSArrayNSSet 的实例来持有此集合。

KVO 对于这三种属性都能适用。下面举一个例子:

如上所示,Person 对象有一个 Account 属性,而 Account 对象又有 balanceinterestRate 两个属性。并且这两个属性对于 Person 对象来说都是可读写的。如果想实现一个功能:当余额或利率变化的时候需要通知到用户。一般来说可以使用轮询的方式,Person 对象定期从 Account 属性中取出 balanceinterestRate。但这种方式是效率低下且不切实际的,更好的方式是使用 KVO,类似于余额或利率变动时, Person 对象收到了通知一样。

要实现 KVO 的前提是要确保被观察对象是符合 KVO 机制的。一般来说,继承于 NSObject 根类的对象及其属性都自动符合 KVO 这一机制。当然也可以自己去实现 KVO 符合。也就是说实际上 KVO 机制分为自动符合手动符合

一旦确定了对象和属性是 KVO 符合的话,就需要历经三个步骤:

  • 观察者注册

Person 对象需要将自己注册到 Account 的某一个具体属性上。这个过程是通过 addObserver:forKeyPath:options:context: 实现的,这个方法需要指定监听者(observer)、监听谁(keypath)、监听策略(options)、监听上下文(context)

  • 被观察者触发回调

Person 对象要接收 Account 被监听属性改动后发出的通知,需要自身实现 observeValueForKeyPath:ofObject:change:context: 方法来接收通知。

  • 观察者取消注册

在观察者不需要再监听或自身生命周期结束的时候,需要取消注册。具体实现是通过向被观察对象发出 removeObserver:forKeyPath: 消息。

KVO 机制的最大好处你不需要自己去实现一个机制来获取对象属性何时改变以及改变后的结果。

1.2 KVO 三大流程解析

1.2.1 观察者注册

- (void)addObserver:(NSObject *)observer
         forKeyPath:(NSString *)keyPath
            options:(NSKeyValueObservingOptions)options
            context:(void *)context
复制代码

observer:注册 KVO 通知的对象。观察者必须实现 key-value observing 方法 observeValueForKeyPath:ofObject:change:context:keyPath:被观察者的属性的 keypath,相对于接受者,值不能是 niloptions: NSKeyValueObservingOptions 的组合,它指定了观察通知中包含了什么 context:在 observeValueForKeyPath:ofObject:change:context: 传给 observer 参数的上下文

前两个参数很好理解,而 optionscontext 参数则需要额外注意。

options 代表 NSKeyValueObservingOptions 的位掩码,需要注意 NSKeyValueObservingOptionNew & NSKeyValueObservingOptionOld,因为这些是你经常要用到的,可以跳过 NSKeyValueObservingOptionInitial & NSKeyValueObservingOptionPrior:

NSKeyValueObservingOptionNew: 表明通知中的更改字典应该提供新的属性值,如何可以的话。 NSKeyValueObservingOptionOld: 表明通知中的更改字典应该包含旧的属性值,如何可以的话。 NSKeyValueObservingOptionInitial: 这个枚举值比较特殊,如果指定了这个枚举值, 在属性发生变化后立即通知观察者,这个过程甚至早于观察者注册。如果在注册的时候配置了 NSKeyValueObservingOptionNew,那么在通知的更改字典中也会包含 NSKeyValueChangeNewKey,但是不会包括 NSKeyValueChangeOldKey。(在初始通知中,观察到的属性值可能是旧的,但是对于观察者来说是新的)其实简单来说就是这个枚举值会在属性变化前先触发一次 observeValueForKeyPath 回调。 NSKeyValueObservingOptionPrior: 这个枚举值会先后连续出发两次 observeValueForKeyPath 回调。同时在回调中的可变字典中会有一个布尔值的 key - notificationIsPrior 来标识属性值是变化前还是变化后的。如果是变化后的回调,那么可变字典中就只有 new 的值了,如果同时制定了 NSKeyValueObservingOptionNew 的话。如果你需要启动手动 KVO 的话,你可以指定这个枚举值然后通过 willChange 实例方法来观察属性值。在出发 observeValueForKeyPath 回调后再去调用 willChange 可能就太晚了。

这些选项允许一个对象在发生变化的前后获取值。在实践中,这不是必须的,因为从当前属性值获取的新值一般是可用的 也就是说 NSKeyValueObservingOptionInitial 对于在反馈 KVO 事件的时候减少代码路径是很有好处的。比如,如果你有一个方法,它能够动态的使一个基于 text 值的按钮有效,传 NSKeyValueObservingOptionInitial 可以使事件随着它的初始化状态触发一旦观察者被添加进去的话。

如何设置一个好的 context 值呢?这里有个建议:

static void * XXContext = &XXContext;
复制代码

就是这么简单:一个静态变量存着它自己的指针。这意味着它自己什么也没有,使 <NSKeyValueObserving> 更完美。

我们简单测试一下在注册观察者时指定不同的枚举值会有怎么样的结果:

  • 只指定 NSKeyValueObservingOptionNew

  • 只指定 NSKeyValueObservingOptionOld

  • 指定 NSKeyValueObservingOptionInitial

可以看到,只指定了 NSKeyValueObservingOptionInitial 后触发了两个回调,并且一次是在属性值变化前,一次是在属性值变化后。同时并且没有新值和旧值返回,我们加一个 NSKeyValueObservingOptionNewNSKeyValueObservingOptionOld:

在我们加上新值和旧值的枚举之后,新值在两次回调后被返回,但是第一次的新值其实是最开始的属性值,第二次才是改变之后的属性值,而旧值在第二次真正属性值被改变后返回。

  • 指定 NSKeyValueObservingOptionPrior

可以看到,NSKeyValueObservingOptionPrior 枚举值是在属性值发生变化后触发了两次回调,同时也没有新值和旧值的返回,我们加一个 NSKeyValueObservingOptionNewNSKeyValueObservingOptionOld:

可以看到,在第一次回调里没有新值,第二次才有,而旧值在两次回调里面都有。

  • keyPath 字符串问题

我们在注册观察者的时候,要求传入的 keyPath 是字符串类型,如果我们拼写错误的话,编译器是不能帮我们检查出来的,所有最佳实践应该是使用 NSStringFromSelector(SEL aSelector),比如我们要观察 tableViewcontentSize 属性,我们可以这样使用:

NSStringFromSelector(@selector(contentSize))
复制代码

1.2.2 观察者接收通知

- (void)observeValueForKeyPath:(NSString *)keyPath
                      ofObject:(id)object
                        change:(NSDictionary *)change
                       context:(void *)context
复制代码

这个方法就是观察者接收通知的地方,除了 change 参数之外,其他三个参数都与观察者注册的时候传入的三个参数一一对应。

  • 不同对象监听相同的 keypath

默认情况下,我们在 addObserver:forKeyPath:options:context: 方法的最后一个参数传入的是 NULL,因为这个方法签名中最后一个参数 contextvoid *,所以需要传入一个空指针,而根据下图我们可知,nil 只是一个对象的字面零值,这里需要的是一个指针,所以需要传 NULL

但是如果是不同的对象都监听同一属性,我们就需要给 context 传入一个可以区分不同对象的字符串指针:

static void *StudentNameContext = &StudentNameContext;
static void *PersonNameContext = &PersonNameContext;

[self.person addObserver:self forKeyPath:@"name" options:(NSKeyValueObservingOptionNew) context:PersonNameContext];
[self.student addObserver:self forKeyPath:@"name" options:(NSKeyValueObservingOptionNew) context:StudentNameContext];

- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary<NSKeyValueChangeKey,id> *)change context:(void *)context{
  
  if (context == PersonNameContext) {
  
  } else if (context == StudentNameContext) {
    
  }
  
}

复制代码
  • 需要自己处理 superclassobserve 事务

对于 Objective-C,很多时候 Runtime 系统都会自动帮助处理 superclass 的方法。譬如对于 dealloc,假设类 Father 继承自 NSObject,而类 Son 继承自Father,创建一个 Son 的实例 aSon,在 aSon 被释放的时候,Runtime 会先调用 Son#dealloc,之后会自动调用 Father#dealloc,而无需在 Son#dealloc 中显式执行 [super dealloc];。但 KVO 不会这样,所以为了保证父类(父类可能也会自己 observe 事务要处理)的 observe 事务也能被处理。

- (void)observeValueForKeyPath:(NSString *)keyPath
                     ofObject:(id)object
                       change:(NSDictionary *)change
                      context:(void *)context {
   
   if (object == _tableView && [keyPath >isEqualToString:@"contentSize"]) {
       [self configureView];
   } else {
       [super observeValueForKeyPath:keyPath ofObject:object >change:change context:context];
   }
}
复制代码

1.2.3 取消注册

- (void)removeObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath context:(void *)context;
- (void)removeObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath;
复制代码

取消注册有两个方法,不过建议还是跟注册和通知两个流程统一,选用带有 context 参数的方法。

  • 取消注册与注册是一对一的关系

一旦对某个对象上的属性注册了键值观察,可以选择在收到属性值变化后取消注册,也可以在观察者声明周期结束之前(比如:dealloc 方法) 取消注册,如果忘记调用取消注册方法,那么一旦观察者被销毁后,KVO 机制会给一个不存在的对象发送变化回调消息导致野指针错误。

  • 不能重复取消注册

取消注册也不能对同一个观察者重复多次,为了避免 crash,可以把取消注册的代码包裹在 try&catch 代码块中:

static void * ContentSizeContext = &ContentSizeContext;
    
- (void)viewDidLoad {
    
    [super viewDidLoad];
    
    // 1. subscribe
    [_tableView addObserver:self
                 forKeyPath:NSStringFromSelector(@selector(contentSize))
                    options:NSKeyValueObservingOptionNew
                    context:ContentSizeContext];
}
    
// 2. responding
- (void)observeValueForKeyPath:(NSString *)keyPath
                      ofObject:(id)object
                        change:(NSDictionary *)change
                       context:(void *)context {
    if (context == ContentSizeContext) {
        // configure view
    } else {
        [super observeValueForKeyPath:keyPath ofObject:object change:change context:context];
    }
}
    
- (void)dealloc {
    @try {
        // 3. unsubscribe
        [_tableView removeObserver:self
                        forKeyPath:NSStringFromSelector(@selector(contentSize))
                           context:ContentSizeContext];
    }
    @catch (NSException *exception) {
        
    }
}
复制代码

1.3 "自动挡" 和 "手动挡"

默认情况下,我们只需要按照前面说的 三步曲 的方式来实现对属性的键值观察,不过这属于是 「自动挡」,什么意思呢?就是说属性值变化完全是由系统控制,我们只需要告诉系统监听什么属性,然后就直接等系统告诉我们就完事了。而实际上,KVO 还支持「手动挡」。

要让系统知道我们想开启手动挡,需要修改类方法 automaticallyNotifiesObserversForKey: 的返回值,这个方法如果返回 YES 就是自动挡,返回 NO 就是手动挡。同时该类方法还能精准实策,让我们选择对哪些属性是自动,哪些属性是手动。

+ (BOOL)automaticallyNotifiesObserversForKey:(NSString *)theKey {
 
    BOOL automatic = NO;
    if ([theKey isEqualToString:@"balance"]) {
        automatic = NO;
    }
    else {
        automatic = [super automaticallyNotifiesObserversForKey:theKey];
    }
    return automatic;
}
复制代码

同样的,如上代码所示,我们使用 automaticallyNotifiesObserversForKey 的最佳实践仍然需要把我们需要手动或自动的代码排除后去调用下父类的方法来确保不会有问题出现。

  • 自动 KVO 触发方式
// Call the accessor method.
[account setName:@"Savings"];
 
// Use setValue:forKey:.
[account setValue:@"Savings" forKey:@"name"];
 
// Use a key path, where 'account' is a kvc-compliant property of 'document'.
[document setValue:@"Savings" forKeyPath:@"account.name"];
 
// Use mutableArrayValueForKey: to retrieve a relationship proxy object.
Transaction *newTransaction = <#Create a new transaction for the account#>;
NSMutableArray *transactions = [account mutableArrayValueForKey:@"transactions"];
[transactions addObject:newTransaction];
复制代码

如上代码所示是自动 KVO 的触发方式

  • 手动 KVO 触发方式

其实手动 KVO 可以帮助我们将多个属性值的更改合并成一个,这样在回调的时候就有一次了,同时也能最大程度地减少处于应用程序特定原因而导致的通知发生。

- (void)setBalance:(double)theBalance {
    [self willChangeValueForKey:@"balance"];
    _balance = theBalance;
    [self didChangeValueForKey:@"balance"];
}
复制代码

如上代码所示,最朴素的手动 KVO 使用方法就是在属性值改变前对观察者发送 willChangeValueForKey 实例方法,在属性值改变之后对观察者发送 didChangeValueForKey 实例方法,参数都是所观察的键。 当然,上面这种方式不是最佳的,为了性能最佳,可以在属性的 setter 中判断是否要执行 will + did:

- (void)setBalance:(double)theBalance {
    if (theBalance != _balance) {
        [self willChangeValueForKey:@"balance"];
        _balance = theBalance;
        [self didChangeValueForKey:@"balance"];
    }
}
复制代码

但是,如果对一个属性的改变会影响到多个键的话,则需要如下的操作:

- (void)setBalance:(double)theBalance {
    [self willChangeValueForKey:@"balance"];
    [self willChangeValueForKey:@"itemChanged"];
    _balance = theBalance;
    _itemChanged = _itemChanged+1;
    [self didChangeValueForKey:@"itemChanged"];
    [self didChangeValueForKey:@"balance"];
}
复制代码

对于有序的一对多关系属性,不仅必须指定已更改的键,还必须指定更改的类型和所涉及对象的索引。 更改的类型是 NSKeyValueChange,它指定 NSKeyValueChangeInsertionNSKeyValueChangeRemovalNSKeyValueChangeReplacement,受影响的对象的索引作为 NSIndexSet 对象传递:

- (void)removeTransactionsAtIndexes:(NSIndexSet *)indexes {
    [self willChange:NSKeyValueChangeRemoval
        valuesAtIndexes:indexes forKey:@"transactions"];
 
    // Remove the transaction objects at the specified indexes.
 
    [self didChange:NSKeyValueChangeRemoval
        valuesAtIndexes:indexes forKey:@"transactions"];
}
复制代码

1.4 注册从属关系的 KVO

所谓从属关系,指的是一个对象的某个属性的值取决于另一个对象的一个或多个属性。对于不同类型的属性,有不同的方式来实现。

  • 一对一关系

要触发 一对一 类型属性的自动 KVO,有两种方式。一种是重写 keyPathsForValuesAffectingValueForKey 方法,一种是实现一个合适的方法。

- (NSString *)fullName {
    return [NSString stringWithFormat:@"%@ %@",firstName, lastName];
}
复制代码

比如上面的代码,fullNamefirstNamelastName 组成,所以重写 fullName 属性的 getter 方法。这样,不论是 firstName 还是 lastName 发生了改变,监听 fullName 属性的观察者都会收到通知。

+ (NSSet *)keyPathsForValuesAffectingValueForKey:(NSString *)key {
 
    NSSet *keyPaths = [super keyPathsForValuesAffectingValueForKey:key];
 
    if ([key isEqualToString:@"fullName"]) {
        NSArray *affectingKeys = @[@"lastName", @"firstName"];
        keyPaths = [keyPaths setByAddingObjectsFromArray:affectingKeys];
    }
    return keyPaths;
}
复制代码

如上代码所示,通过实现类方法 keyPathsForValuesAffectingValueForKey 来返回一个集合。值得注意的是,这里需要先对父类发送 keyPathsForValuesAffectingValueForKey 消息,以免干扰父类中对此方法的重写。

实际上还有一个便利的方法,就是 keyPathsForValuesAffecting<Key>Key 是属性的名称(需要首字母大写)。这个方法的效果和 keyPathsForValuesAffectingValueForKey 是一样的,但针对的某个具体属性。

+ (NSSet *)keyPathsForValuesAffectingFullName {
    return [NSSet setWithObjects:@"lastName", @"firstName", nil];
}
复制代码

相对来说,在分类中去使用 keyPathsForValuesAffectingFullName 更合理,因为分类中是不允许重载方法的,所以 keyPathsForValuesAffectingValueForKey 方法肯定是不能在分类中使用的。

  • 一对多关系

keyPathsForValuesAffectingValueForKey:方法不支持包含一对多关系的 Key Path。例如,假设你有一个 Department对象,该对象与 Employee 有一对多关系(即 employees 属性),而 Employee 具有 salary 属性。 如果需要在 Department 对象上增加totalSalary 属性,而该属性取决于关系中所有 Employees 的薪水。例如,您不能使用 keyPathsForValuesAffectingTotalSalary 和返回 employees.salary 作为键来执行此操作。

- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context {
 
    if (context == totalSalaryContext) {
        [self updateTotalSalary];
    }
    else
    // deal with other observations and/or invoke super...
}
 
- (void)updateTotalSalary {
    [self setTotalSalary:[self valueForKeyPath:@"employees.@sum.salary"]];
}
 
- (void)setTotalSalary:(NSNumber *)newTotalSalary {
 
    if (totalSalary != newTotalSalary) {
        [self willChangeValueForKey:@"totalSalary"];
        _totalSalary = newTotalSalary;
        [self didChangeValueForKey:@"totalSalary"];
    }
}
 
- (NSNumber *)totalSalary {
    return _totalSalary;
}
复制代码

如上代码所示,将 Department 实例对象注册为观察者,然后观察对象为 totalSalary 属性,但是在通知回调中会手动调用 totalSalary 属性的 setter 方法,并且传入值是通过 KVC 的集合运算符的方式取出 employees 属性所对应的集合中所有 sum 值之和。然后在 totalSalary 属性的 setter 方法中,会相应的调用 willChangeValueForKey:didChangeValueForKey: 方法。

如果使用的是 Core Data,你还可以把 Department 注册到 NSNotificationCenter 中来作为托管对象上下文的观察者。Department 应以类似于观察键值的方式响应 Employee 发布的相关变更通知。

二、KVO 原理探究

Automatic key-value observing is implemented using a technique called isa-swizzling. 【译】自动的键值观察的实现基于 isa-swizzling

The isa pointer, as the name suggests, points to the object's class which maintains a dispatch table. This dispatch table essentially contains pointers to the methods the class implements, among other data. 【译】isa 指针,顾名思义,指向的是对象所属的类,这个类维护了一个哈希表。这个哈希表基本上存储的是方法的 SELIMP 的键值对。

When an observer is registered for an attribute of an object the isa pointer of the observed object is modified, pointing to an intermediate class rather than at the true class. As a result the value of the isa pointer does not necessarily reflect the actual class of the instance.

【译】当一个观察者注册了对一个对象的某个属性键值观察之后,被观察对象的 isa 指针所指向的内容发生了变化,指向了一个中间类而不是真正的类。这也导致 isa 指针并不一定是指向实例所属的真正的类。

You should never rely on the isa pointer to determine class membership. Instead, you should use the class method to determine the class of an object instance.

【译】你永远不应依靠 isa 指针来确定类成员身份。相反,你应该使用 class 方法来确定对象实例所属的类。

2.1 中间类

根据官网文档的内容,我们初步判断,在 KVO 底层实现中,会有一个所谓的中间类生成。而这个中间类会让对象的 isa 指针发生变化。我们不妨测试一下:

如上图所示,person 对象和 personForTest 对象都是属于 JHPerson 类的,而 person 对象又实现了 KVO,但是在控制台打印结果里面可以看到它们二者的类都是 JHPerson 类。不是说会有一个中间类生成吗?难道是这个中间类生成又被干掉了?我们直接LLDB 大法测试一下:

Bingo~,所谓的中间类 NSKVONotifying_JHPerson 被我们找出来了。那么其实这里显然,系统是重写了中间类 NSKVONotifying_JHPersonclass 方法,让我们以为对象的 isa 指针一直指向的都是 JHPerson 类。那么这个中间类和原来的类是什么关系呢?我们可以测试一下:

其中 printClasses 实现如下:

- (void)printClasses:(Class)cls{
    int count = objc_getClassList(NULL, 0);
    NSMutableArray *mArray = [NSMutableArray arrayWithObject:cls];
    Class* classes = (Class*)malloc(sizeof(Class)*count);
    objc_getClassList(classes, count);
    for (int i = 0; i<count; i++) {
        if (cls == class_getSuperclass(classes[i])) {
            [mArray addObject:classes[i]];
        }
    }
    free(classes);
    NSLog(@"classes = %@", mArray);
}
复制代码

最终打印结果如下:

classes = (
    JHPerson
)
classes = (
    JHPerson,
    "NSKVONotifying_JHPerson"
)
复制代码

结果很清晰,中间类 NSKVONotifying_JHPerson 是作为原始真正的类 JHPerson 的子类的角色。

2.2 KVO 观察的是什么?

KVO 所关注的是属性值的变化,而属性值本质上是成员变量+getter+settergetter 是用来获取值的,而显然只有 setter 和成员变量赋值两种方式可以改变属性值。我们测试一下这两种方式:

// JHPerson.h
@interface JHPerson : NSObject {
    @public
    NSString *_nickName;
}
@property (nonatomic, copy) NSString *name;
@end
复制代码

如上图所示,setter 方法对属性 name 做了修改被 KVO 监听到了,而成员变量 _nickName 的修改并没有被监听到,说明 KVO 底层其实观察的是 setter 方法。

2.3 中间类重写了哪些方法?

我们可以通过打印原始类和中间类的方法列表来验证:

printClassAllMethod 方法实现如下:

- (void)printClassAllMethod:(Class)cls{
    NSLog(@"*********************");
    unsigned int count = 0;
    Method *methodList = class_copyMethodList(cls, &count);
    for (int i = 0; i<count; i++) {
        Method method = methodList[i];
        SEL sel = method_getName(method);
        IMP imp = class_getMethodImplementation(cls, sel);
        NSLog(@"%@-%p",NSStringFromSelector(sel),imp);
    }
    free(methodList);
}
复制代码

可以看到如上图所示,原始类和中间类都有 setter 方法。根据我们前面所探索的消息发送以及转发流程,这里的中间类应该是重写了 setName:classdealloc_isKVOA 方法。

由我们上一小节的测试结果可知,中间类重写的 class 方法结果仍然是返回的是原始类,显然系统这样做的目的就是隐藏中间类的存在,让调用者调用 class 方法结果前后一致。

2.4 KVO 中间类何时指回去?

我们推断 KVO 注册观察者到移除观察者这一个流程里面,被观察对象的 isa 指针才会指向中间类,我们用代码测试一下:

由上图可知,观察者的 dealloc 方法中的移除观察者之后,对象的 isa 指针已经指回了原始的类。那么是不是此时中间类就被销毁了呢,我们不妨打印一下此时原始类的所有子类信息:

结果表明中间类仍然存在,也就是说移除观察者并不会导致中间类销毁,显然这样对于多次添加和移除观察者来说性能上更好。

2.5 KVO 调用顺序

而我们前面说了,有一个中间类的存在,既然要生成中间类,肯定是有意义的,我们梳理一下整个 KVO 的流程,从注册观察者到观察者的回调通知,既然有回调通知,那么肯定是在某个地方发出回调的,而由于中间类是不能编译的,所以我们对中间类的父类也就是 JHPerson 类,我们重写一下相应的 setter 方法,我们不妨测试一下:

// JHPerson.m
- (void)setName:(NSString *)name
{
    _name = name;
}

- (void)willChangeValueForKey:(NSString *)key{
    [super willChangeValueForKey:key];
    NSLog(@"willChangeValueForKey");
}

- (void)didChangeValueForKey:(NSString *)key{
    NSLog(@"didChangeValueForKey - begin");
    [super didChangeValueForKey:key];
    NSLog(@"didChangeValueForKey - end");
}
复制代码

打印结果如下:

也就是说 KVO 的调用顺序是:

  • 调用 willChangeValueForKey:
  • 调用原来的 setter 实现
  • 调用 didChangeValueForKey:

也就是说 didChangeValueForKey: 内部必然是调用了 observerobserveValueForKeyPath:ofObject:change:context:方法。

三、自定义 KVO 如何实现

我们已经初步了解了 KVO 底层原理,接下来我们尝试自己简单实现一下 KVO。 我们直接跳转到 addObserver:forKeyPath:options:context: 方法的声明处:

可以看到,跟 KVC 一样,KVO 在底层也是以分类的形式加载的,这个分类叫做 NSKeyValueObserverRegistration。我们不妨也以这种方式来自定义实现一下 KVO

// NSObject+JHKVO.h
@interface NSObject (JHKVO)
// 观察者注册
- (void)jh_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(JHKeyValueObservingOptions)options context:(nullable void *)context;
// 回调通知观察者
- (void)jh_observeValueForKeyPath:(nullable NSString *)keyPath ofObject:(nullable id)object change:(nullable NSDictionary<NSKeyValueChangeKey, id> *)change context:(nullable void *)context;
// 移除观察者
- (void)jh_removeObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath context:(nullable void *)context;
@end
复制代码

这里为了避免与系统的方法冲突,所以添加了一个方法前缀。同时对于观察策略,为了简化实现,这里只声明了新值和旧值两种策略。

3.1 自定义观察者注册

在开始之前,我们回忆下自定义 KVC 的时候的第一个步骤就是判断 key 或者 keyPath,那么 KVO 是否也需要进行这样的判断呢?经过笔者实际测试,如果观察对象的一个不存在的属性的话,并不会报错,也不会来到 KVO 回调方法,由此可见,判断 keyPath 是否存在并没有必要。但是,我们回想一下上一节 KVO 底层原理,KVO 关注的是属性的 setter 方法,那其实判断对象所属的类是否有这样的 setter 就相当于同时判断了 keyPath 是否存在。接着我们就需要去动态的创建子类,创建子类的过程中包括了重写 setter 等一系列方法。然后就需要保存观察者和 keyPath 等信息,这里我们借助关联对象来实现,我们把传入的观察者对象、keyPath和观察策略封装成一个新的对象存储在关联对象中。因为同一个对象的属性可以被不同的观察者所观察,所以这里实质上是以对象数组的方式存储在关联对象里面。 话不多说,直接上代码:

// JHKVOInfo.h
typedef NS_OPTIONS(NSUInteger, JHKeyValueObservingOptions) {
    JHKeyValueObservingOptionNew = 0x01,
    JHKeyValueObservingOptionOld = 0x02,
};
@interface JHKVOInfo : NSObject
@property (nonatomic, weak) NSObject  *observer;
@property (nonatomic, copy) NSString  *keyPath;
@property (nonatomic, assign) JHKeyValueObservingOptions options;
- (instancetype)initWitObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(JHKeyValueObservingOptions)options;
@end

// JHKVOInfo.m
@implementation JHKVOInfo
- (instancetype)initWitObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(JHKeyValueObservingOptions)options
{
    if (self = [super init]) {
        _observer = observer;
        _keyPath = keyPath;
        _options = options;
    }
    return self;
}
@end
复制代码

上面的代码是自定义的 JHKVOInfo 对象。

static NSString *const kJHKVOPrefix = @"JHKVONotifying_";
static NSString *const kJHKVOAssiociateKey = @"kJHKVO_AssiociateKey";
- (void)jh_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(JHKeyValueObservingOptions)options context:(void *)context
{
    // 1.判断 getter 是否存在
    SEL setterSelector = NSSelectorFromString(setterForGetter(keyPath));
    Method setterMethod = class_getInstanceMethod([self class], setterSelector);
    if (!setterMethod) {
        NSString *reason = [NSString stringWithFormat:@"对象 %@ 的 key %@ 没有 setter 实现", self, keyPath];
        @throw [NSException exceptionWithName:NSInvalidArgumentException
                                       reason:reason
                                     userInfo:nil];
        return;
    }
    
    // 2.动态创建中间子类
    Class newClass = [self createChildClassWithKeyPath:keyPath];
    
    // 3.将对象的isa指向为新的中间子类
    object_setClass(self, newClass);
    
    // 4.保存观察者
    JHKVOInfo *info = [[JHKVOInfo alloc] initWitObserver:observer forKeyPath:keyPath options:options];
    NSMutableArray *observerArr = objc_getAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey));
    
    if (!observerArr) {
        observerArr = [NSMutableArray arrayWithCapacity:1];
        [observerArr addObject:info];
        objc_setAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey), observerArr, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
    }
}
复制代码

上面的代码是完整的添加观察者的流程:

  • 判断对象所属的类上是否有要观察的 keyPath 对应的 setter 方法

这里的 setterForGetter 实现如下:

static NSString * setterForGetter(NSString *getter)
{
   // 判断 getter 是否为空字符串
   if (getter.length <= 0) {
       return nil;
   }
   // 取出 getter 字符串的第一个字母并转大写 
   NSString *firstLetter = [[getter substringToIndex:1] uppercaseString];
   // 取出剩下的字符串内容
   NSString *remainingLetters = [getter substringFromIndex:1];
   // 将首字母大写的字母与剩下的字母拼接起来得到 `set<KeyPath>` 格式的字符串
   NSString *setter = [NSString stringWithFormat:@"set%@%@:", firstLetter, remainingLetters];
   return setter;
}
复制代码
  • 如果存在相应的 setter 方法,那么就创建有对应前缀的中间子类

这里的 createChildClassWithKeyPath 实现如下:

- (Class)createChildClassWithKeyPath:(NSString *)keyPath{
    // 获得原始类的类名
    NSString *oldClassName = NSStringFromClass([self class]);
    // 在原始类名前添加中间子类的前缀来获得中间子类名
    NSString *newClassName = [NSString stringWithFormat:@"%@%@",kJHKVOPrefix,oldClassName];
    // 通过中间子类名来判断是否创建过
    Class newClass = NSClassFromString(newClassName);
    // 如果创建过中间子类,直接返回
    if (newClass) return newClass;
    // 如果没有创建过,则需要创建一下, objc_allocateClassPair 方法的三个参数分别为: 1.父类 2.新类的名字 3.创建新类所需额外的空间
    newClass = objc_allocateClassPair([self class], newClassName.UTF8String, 0);
    // 注册中间子类
    objc_registerClassPair(newClass);
    // 从父类上拿到 `class` 方法的 `SEL` 以及类型编码,然后在中间子类上添加一个新的子类实现 `jh_class`
    SEL classSEL = NSSelectorFromString(@"class");
    Method classMethod = class_getInstanceMethod([self class], classSEL);
    const char *classTypes = method_getTypeEncoding(classMethod);
    class_addMethod(newClass, classSEL, (IMP)jh_class, classTypes);
    // 从父类上拿到 `getter` 方法的 `SEL` 以及类型编码,然后在中间子类上添加一个新的子类实现 `jh_setter`
    SEL setterSEL = NSSelectorFromString(setterForGetter(keyPath));
    Method setterMethod = class_getInstanceMethod([self class], setterSEL);
    const char *setterTypes = method_getTypeEncoding(setterMethod);
    class_addMethod(newClass, setterSEL, (IMP)jh_setter, setterTypes);
    return newClass;
}
复制代码

jh_class 的实现如下:

Class jh_class(id self,SEL _cmd) {
   // 通过 class_getSuperclass 来返回父类的 `Class`,达到对调用者隐藏中间子类的效果
   return class_getSuperclass(object_getClass(self));
}
复制代码

jh_setter 的实现如下:

static void jh_setter(id self,SEL _cmd,id newValue){
    // 因为 `_cmd` 作为方法的第二个参数其实就是 `setter` 的 `SEL`,这里反向获得对应 `getter` 字符串形式作为 `keyPath`,然后通过 `KVC` 来获取到旧的属性值
    NSString *keyPath = getterForSetter(NSStringFromSelector(_cmd));
    id oldValue       = [self valueForKey:keyPath];
    
    // 因为是重写父类的 `setter`,所以还需要通过消息发送的方式手动执行以下父类的 `setter` 方法
    // 通过强转的方式将 `objc_msgSendSuper` 转成 `jh_msgSendSuper` 函数指针,同时,由于 `objc_msgSendSuper` 要比我们常见的 `objc_msgSend` 多一个父类结构体参数,所以需要手动构建一下这个父类结构体,结构体有两个属性,分别是实例对象以及实例对象的类的父类
    void (*jh_msgSendSuper)(void *, SEL, id) = (void *)objc_msgSendSuper;
    // void /* struct objc_super *super, SEL op, ... */
    struct objc_super superStruct = {
        .receiver = self,
        .super_class = class_getSuperclass(object_getClass(self)),
    };
    // 准备工作完成后手动调用 `jh_msgSendSuper`,因为 `superStruct` 是结构体类型,而 `jh_msgSendSuper` 的第一个参数是空指针对象,所以这里需要加取地址符来把结构体地址赋值给指针对象
    jh_msgSendSuper(&superStruct, _cmd, newValue);
    // 调用完父类的 `setter` 之后,从关联对象中取出存储了自定义的对象数组
    NSMutableArray *observerArr = objc_getAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey));
    // 循环遍历自定义的对象
    for (JHKVOInfo *info in observerArr) {
    // 如果 `keyPath` 匹配则进入下一步
        if ([info.keyPath isEqualToString:keyPath]) {
            // 基于线程安全的考虑,使用 `GCD` 的全局队列异步执行下面的操作 
            dispatch_async(dispatch_get_global_queue(0, 0), ^{
                // 初始化一个通知字典
                NSMutableDictionary<NSKeyValueChangeKey,id> *change = [NSMutableDictionary dictionaryWithCapacity:1];
                // 判断存储的观察策略,如果是新值,则在通知字典中设置新值  
                if (info.options & JHKeyValueObservingOptionNew) {
                    [change setObject:newValue forKey:NSKeyValueChangeNewKey];
                }
                // 如果是旧值,在通知字典中设置旧值
                if (info.options & JHKeyValueObservingOptionOld) {
                    [change setObject:@"" forKey:NSKeyValueChangeOldKey];
                    if (oldValue) {
                        [change setObject:oldValue forKey:NSKeyValueChangeOldKey];
                    }
                }
                // 取得通知观察者方法的 `SEL`
                SEL observerSEL = @selector(jh_observeValueForKeyPath:ofObject:change:context:);
                // 通过 `objc_msgSend` 手动发送消息,达到观察者收到回调的效果
                ((void(*)(id, SEL, id, id, NSMutableDictionary *, void *))objc_msgSend)(info.observer, observerSEL, keyPath, self, change, NULL);
            });
        }
    }
}
复制代码

getterForSetter 实现如下:

static NSString *getterForSetter(NSString *setter){
    // 判断传入的 `setter` 字符串长度是否大于 0,以及是否有 `set` 的前缀和 `:` 的后缀
    if (setter.length <= 0 || ![setter hasPrefix:@"set"] || ![setter hasSuffix:@":"]) { return nil;}
    // 排除掉 `setter` 字符串中的 `set:` 部分以取得 getter 字符串
    NSRange range = NSMakeRange(3, setter.length-4);
    NSString *getter = [setter substringWithRange:range];
    // 对 getter 字符串首字母小写处理
    NSString *firstString = [[getter substringToIndex:1] lowercaseString];
    return  [getter stringByReplacingCharactersInRange:NSMakeRange(0, 1) withString:firstString];
}
复制代码

3.2 自定义移除观察者

我们接着开始自定义移除观察者,首先,我们需要把 isa 指回原来的类,然后需要对关联对象中存储的自定义对象数组对应的观察者移除掉。

- (void)jh_removeObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath context:(void *)context
{
    // 从关联对象中取出数组
    NSMutableArray *observerArr = objc_getAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey));
    // 如果数组中没有内容,说明没有添加过观察者,那么直接返回
    if (observerArr.count<=0) {
        return;
    }
    
    // 遍历取出的所有自定义对象
    for (JHKVOInfo *info in observerArr) {
        // 如果 `keyPath` 匹配上了 则从数组中移除响应对象,然后存储最新的数组到关联对象上
        if ([info.keyPath isEqualToString:keyPath]) {
            [observerArr removeObject:info];
            objc_setAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey), observerArr, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
            break;
        }
    }
    
    // 要将 `isa` 指回原来的类的前提条件是,被观察属性的对象已经没有任何观察者在观察了,那么就需要指回去
    if (observerArr.count<=0) {
        Class superClass = [self class];
        object_setClass(self, superClass);
    }
}
复制代码

3.3 实现自动移除观察者

现在我们自定义的 KVO 已经可以实现简单的通知观察者新值和旧值的变化了,但其实对于 api 的使用者来说,还是要严格的执行 addObserverremoveObserver 的配套操作,难免有些繁琐。虽然一般来说为了方便起见,都是在观察者的 dealloc 方法中去手动调用 removeObserver 方法,但还是太麻烦了。因此,我们可以借助 methodSwizzling 的技术来替换默认 dealloc 方法的实现,直接上代码:

+ (BOOL)jh_hookOrigInstanceMenthod:(SEL)oriSEL newInstanceMenthod:(SEL)swizzledSEL {
    // 获取 Class 对象
    Class cls = self;
    // 通过 `SEL` 获取原始方法
    Method oriMethod = class_getInstanceMethod(cls, oriSEL);
    // 通过 `SEL` 获取要替换的方法
    Method swiMethod = class_getInstanceMethod(cls, swizzledSEL);
    // 如果要替换的方法不存在,返回 NO
    if (!swiMethod) {
        return NO;
    }
    // 如果原始方法不存在,那么就直接在 Class 上添加要替换的方法,注意,添加的方法实现为要替换的方法,但是方法 `SEL` 还是原始方法的 `SEL`
    if (!oriMethod) {
        class_addMethod(cls, oriSEL, method_getImplementation(swiMethod), method_getTypeEncoding(swiMethod));
        method_setImplementation(swiMethod, imp_implementationWithBlock(^(id self, SEL _cmd){ }));
    }
    
    // 判断是否添加成功
    BOOL didAddMethod = class_addMethod(cls, oriSEL, method_getImplementation(swiMethod), method_getTypeEncoding(swiMethod));
    if (didAddMethod) {
    // 如果成功,说明 Class 上已经存在了要替换的方法的实现,那么就把原始方法实现替换掉 `swizzledSEL` 对应的方法实现
        class_replaceMethod(cls, swizzledSEL, method_getImplementation(oriMethod), method_getTypeEncoding(oriMethod));
    }else{
    // 如果不成功,说明原始方法已经存在,则直接交换方法实现
        method_exchangeImplementations(oriMethod, swiMethod);
    }
    return YES;
}


+ (void)load{
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        [self jh_hookOrigInstanceMenthod:NSSelectorFromString(@"dealloc") newInstanceMenthod:@selector(myDealloc)];
    });
}

- (void)myDealloc{
    Class superClass = [self class];
    object_setClass(self, superClass);
    // 这里并不会造成循环引用的递归,因为 `myDealloc` 的方法实现是真正的原始 `dealloc`
    [self myDealloc];
}
复制代码

通过实现自动移除观察者,api 的使用者可以完全放心的只使用 addObserver 来添加观察者以及 observeValueForKeyPath 来接收回调。

3.4 函数式编程思想重构

我们虽然已经实现了自动的移除观察者,但是从函数式编程思想来看,现在的设计还不是很完美,对同一个属性的观察的代码散落在不同的地方,如果业务一旦增多,对于可读性和可维护性都有很大的影响。所以,我们可以把现在这种回调的形式重构为 Block 的方式。

// NSObject+JHBlockKVO.h
typedef void(^JHKVOBlock)(id observer,NSString *keyPath,id oldValue,id newValue);
@interface NSObject (JHBlockKVO)
- (void)jh_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath block:(JHKVOBlock)block;
- (void)jh_removeObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath;
@end

// NSObject+JHBlockKVO.m
@interface JHBlockKVOInfo : NSObject
@property (nonatomic, weak) NSObject   *observer;
@property (nonatomic, copy) NSString   *keyPath;
@property (nonatomic, copy) JHKVOBlock  handleBlock;
@end

@implementation JHBlockKVOInfo
- (instancetype)initWitObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath handleBlock:(JHKVOBlock)block{
    if (self=[super init]) {
        _observer = observer;
        _keyPath  = keyPath;
        _handleBlock = block;
    }
    return self;
}
@end

@implementation NSObject (JHBlockKVO)
- (void)jh_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath block:(JHKVOBlock)block{
    
    // 1.判断 getter 是否存在
    SEL setterSelector = NSSelectorFromString(setterForGetter(keyPath));
    Method setterMethod = class_getInstanceMethod([self class], setterSelector);
    if (!setterMethod) {
        NSString *reason = [NSString stringWithFormat:@"对象 %@ 的 key %@ 没有 setter 实现", self, keyPath];
        @throw [NSException exceptionWithName:NSInvalidArgumentException
                                       reason:reason
                                     userInfo:nil];
        return;
    }
    
    // 2.动态创建中间子类
    Class newClass = [self createChildClassWithKeyPath:keyPath];
    
    // 3.将对象的isa指向为新的中间子类
    object_setClass(self, newClass);
    
    // 4.保存观察者
    JHBlockKVOInfo *info = [[JHBlockKVOInfo alloc] initWitObserver:observer forKeyPath:keyPath handleBlock:block];
    NSMutableArray *mArray = objc_getAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey));
    if (!mArray) {
        mArray = [NSMutableArray arrayWithCapacity:1];
        objc_setAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey), mArray, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
    }
    [mArray addObject:info];
} 
复制代码

这里我们直接通过传入分类一个 block,然后存储在对应的自定义观察对象中,然后我们还需要在重写 setter 方法中做出修改,原来是直接通过发送消息来实现回调,现在需要改成 block 回调

static void jh_setter(id self,SEL _cmd,id newValue){
    NSString *keyPath = getterForSetter(NSStringFromSelector(_cmd));
    id oldValue = [self valueForKey:keyPath];
    void (*jh_msgSendSuper)(void *,SEL , id) = (void *)objc_msgSendSuper;
    struct objc_super superStruct = {
        .receiver = self,
        .super_class = class_getSuperclass(object_getClass(self)),
    };
    jh_msgSendSuper(&superStruct,_cmd,newValue);
    
    NSMutableArray *mArray = objc_getAssociatedObject(self, (__bridge const void * _Nonnull)(kJHKVOAssiociateKey));
    for (JHBlockKVOInfo *info in mArray) {
        if ([info.keyPath isEqualToString:keyPath] && info.handleBlock) {
            info.handleBlock(info.observer, keyPath, oldValue, newValue);
        }
    }
}
复制代码

四、总结

经过探索 KVCKVO 的底层,我们可以看到 KVO 是建立在 KVC 基础之上的。KVO 作为观察者设计模式在 iOS 中的具体落地,其原理到实现我们都探索完了。其实我们可以看出来在早期设计 api 的时候,原生的 KVO 其实并不好用,所以诸如 FaceBook 的库 KVOController 会大受欢迎。当然本文的自定义 KVO 实现并不严谨,感兴趣的读者可以查看这两个代码库:

我们的 iOS 底层探索系列接下来将会进入多线程篇章,敬请期待~

参考资料

Key-Value Observing Programming Guide - Apple 官方文档

nil/Nil/Null/NSNull - NSHipster

Key-Value Observing - NSHipster


作者:leejunhui
链接:https://juejin.cn/post/6844904067794206733
来源:掘金
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。