iOS中KVO的模拟实现

iOS中KVO的底层实现原理

在开发中我们经常使用addObserver:forKeyPath:options:context:方法来观察类的某个属性的改变,然后在observeValueForKeyPath:ofObject:change:context:方法中监听改变的回调,其底层的实现原理大致如下:

  • 利用Runtime动态的生成一个子类,类名是NSKVONotifying_为前缀。
  • 苹果为了隐藏KVO的实现,重写了子类class方法,返回的是父类的类对。
  • 重写了被监听属性的setter方法,这是实现KVO的关键,其实内部调用了Foundation框架下的_NSSet***ValueAndNotify方法,看具体监听属性的类型是什么,方法的调用名略有区别,这个方法的实现是KVO的核心,其大致实现逻辑如下:
    • 调用- (void)willChangeValueForKey:(NSString *)key方法表明属性即将发生改变。
    • 调用父类原来的setter方法的实现。
    • 调用- (void)didChangeValueForKey:(NSString *)key方法表明属性已改变,其中这个方法里面会调用observeValueForKeyPath: ofObject: change: context:方法告知父类监听的属性发生了改变。

上面只是大致了说了下底层的实现流程,其实当然还有一些其他的善后工作要做,这里我们不在深入研究,有兴趣的可以利用查看源码并用Runtime打印监听前后类的方法列表以及实现进行跟踪验证。

自定义KVO的实现

上面已经简要介绍了KVO的实现原理,现在我们仿照上面的原理自己写一个KVO的实现,也大致分为以下几个步骤:

  • 动态生成一个子类。
  • 重写setter方法,在方法中,调用supersetter实现,并通知观察者。

首先定义一个NSObject(KVO)的分类,然后仿照苹果一样定义一个- (void)wy_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(nullable void *)context方法来监听属性,方法的具体实现如下:

- (void)wy_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(nullable void *)context{
   //1.利用 runtime,动态生成一个类
   //1.1 创建self的子类
    NSString *oldClassName = NSStringFromClass([self class]);
    NSString *newClassName = [@"wykvo_" stringByAppendingString:oldClassName];
    const char *newName = [newClassName UTF8String];
    //创建一个类的class
    Class MyClass = objc_allocateClassPair([self class], newName, 0);
    //注册类
    objc_registerClassPair(MyClass);
    //2.添加一个set方法
    class_addMethod(MyClass, @selector(setName:), (IMP)setName, "v@:@");
    //3.改变isa指针(这个好像不利于把方法写成活的,采用方法交换可能更好)
    object_setClass(self, MyClass);
    //4.保存观察者对象
    objc_setAssociatedObject(self, @"objc", observer, OBJC_ASSOCIATION_RETAIN_NONATOMIC); 
}

首先利用Runtimeobjc_allocateClassPair方法来动态生成一个子类,并添加一个setName的方法,并改变isa指针的指向为新生成的子类,同时利用关联对象为分类添加一个objc的属性保存着观察者对象以便后面通知观察者属性发生了改变。

这里有几个点需要注意,由于这里只是简单的模拟name属性的改变,所以set属性的方法名是写死的,实际上应该根据keypath来动态确定,这里不在深入;为了实现当调用set方法能调用到新类的set方法上,采用了改变isa的指针来实现,这样在调用时会根据isa的指向找到新类的实现;同时由于分类中需要保存观察者,由于分类是不能添加属性的,这里采用了关联对象来保存观察者对象。

void setName(id self,SEL _cmd,NSString *newName){
    //调用父类的set方法,需要在build打开容许消息机制
    //保存子类类型
    id class = [self class];
    //改变self的isa指针
    object_setClass(self, class_getSuperclass(class));
    ((void (*)(id, SEL, NSString *))objc_msgSend)(self, @selector(setName:), newName);
    //拿到通知观察者
    id objc = objc_getAssociatedObject(self, @"objc");
    // 通知观察者
    ((void (*)(id, SEL, id, NSString *, id, void *))objc_msgSend)(objc, @selector(observeValueForKeyPath:ofObject:change:context:),self,@"name",nil,nil);
    //改回子类类型
    object_setClass(self, class);
}

setName的实现中,由于需要首先调用原来的set实现,所以再次将isa指针指向原来的被观察对象,同时利用objc_msgSend消息发送机制调用set方法,这样会根据isa指针首先找到观察类的set实现,然后通过关联对象拿到观察者,利用objc_msgSend调用相应的方法完成通知。

注意点

在使用KVO的过程中,判断某个属性设置会不会触发KVO需要看是否调用了set方法,比如如果直接对成员变量进行赋值则不会触发KVO机制,比如Person类里面一个dog对象属性,dog类有个name属性吗,当我们监听dog属性时,如何用person.dog.namedogname进行赋值时则不会调用dogset方法,是不会触发KVO的,但是可以手动在person.dog.name的前后调用上面提到的willChangeValueForKeydidChangeValueForKey方法来触发KVO机制。

总结

根据KVO的底层的实现原理,利用Runtime的消息机制,isa指针和关联对象等相关底层知识模仿实现了KVO实现,这有助于进一步理解底层KVO的实现原理,并加深对Runtime的相关知识的理解。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 206,839评论 6 482
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 88,543评论 2 382
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 153,116评论 0 344
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 55,371评论 1 279
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 64,384评论 5 374
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 49,111评论 1 285
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,416评论 3 400
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,053评论 0 259
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,558评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,007评论 2 325
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,117评论 1 334
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,756评论 4 324
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,324评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,315评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,539评论 1 262
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,578评论 2 355
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,877评论 2 345

推荐阅读更多精彩内容