一步一步实现自定义KVO

听起来KVO很强大,其实自己实现也是可以的。但是想做到像系统实现的那样还是比较费事的,这次我们就搞一个简单一点的,但是功能和系统提供的几乎一致。

首先说一下原理,其实就是在调用方法

addObserver:(NSObject*)observer forKeyPath:(NSString*)keyPath options:(NSKeyValueObservingOptions)options context:(nullablevoid*)context

的时候,系统在这个方法内部通过runtime创建了需要监听对象的类的子类,然后将isa指针指向了这个子类,同时又添加了需要监听属性的set方法、dealloc、_isKOVA、重写了class方法。

因为被监听对象的isa指针指向了动态创建的子类,所以当修改属性的时候就调用了子类的set方法,然后在set方法内部做一些处理,比如获取旧值,设置新值,调用 willChangeValueForKey、didChangeValueForKey,向observeValueForKeyPath:ofObject:change:context:发送消息等等一些操作。

重写class方法主要是为了让开发者调用[xx class]的时候返回的还是原来的类,但是如果使用object_getClass(xx)这个时候获取到的就是真正的类了。

当然想看一下动态创建子类的所有方法可以使用runtime的class_copyMethodList方法获取。

下面开始自己实现KVO。

先来观察两张图片

通过这两张图就可以看出来苹果也是在添加观察者的时候自动创建了一个被观察对象的子类,知道了原理就好办了

其实核心就是使用runtime,好了废话不多说直接开撸.

1,首先创建一个NSObject分类,  在.h里边搞一个供外部添加观察者的方法

@interfaceNSObject (YHKVO)

- (void)YH_addObserver:(NSObject*)observer forKeyPath:(NSString*)keyPath options:(NSKeyValueObservingOptions)options context:(nullablevoid*)context;

@end


2,    在.m实现该方法

首先通过原类创建一个原类的子类,创建好了以后不要忘记注册,系统创建的子类前缀是NSKVONotifying_,我们也搞一个和系统差不多的YHKVONotifying_

    NSString *oldName = NSStringFromClass(self.class);

    NSString *newName = [@"YHKVONotifying_" stringByAppendingString:oldName];

    ClassMyClass =objc_getClass(newName.UTF8String);

    if(!MyClass) {

        MyClass =objc_allocateClassPair(self.class, newName.UTF8String, 0);

        //注册类

        objc_registerClassPair(MyClass);

    }


3,需要在子类中添加一个方法,获取添加方法需要的所有参数

    //set方法首字母大写

    NSString *keyPathChange = [[[keyPath substringToIndex:1] uppercaseString] stringByAppendingString:[keyPath substringFromIndex:1]];

    NSString*setNameStr = [NSStringstringWithFormat:@"set%@", keyPathChange];

    SEL setSEL = NSSelectorFromString([setNameStr stringByAppendingString:@":"]);

    //添加子类的set方法

    Method getMethod = class_getInstanceMethod([self class], @selector(keyPath));

    constchar*types =method_getTypeEncoding(getMethod);

    /**

     *class 给哪个类添加方法

     *sel 方法编号

     *imp 方法实现(函数指针)

     *type返回值类型以及参数

     */

    class_addMethod(MyClass, setSEL, (IMP)setMethod, types);


4,修改当前指针指向子类,修改了以后在添加观察者的时候打断点就可以看到指针的指向改变了

    //修改isa指针    

    object_setClass(self, MyClass);


5,保存接下来需要使用的方法

    //保存方法

    objc_setAssociatedObject(self, YH_observer, observer, OBJC_ASSOCIATION_ASSIGN);

    objc_setAssociatedObject(self, YH_settter, setNameStr, OBJC_ASSOCIATION_COPY_NONATOMIC);

    objc_setAssociatedObject(self, YH_getter, keyPath, OBJC_ASSOCIATION_COPY_NONATOMIC);

6,实现void setMethod方法

所有的objc_msgSend()最少需要两个参数,第一个参数self, 第二个_cmd

    void setMethod(id self, SEL _cmd, id newVlaue) {

    NSString *setName = objc_getAssociatedObject(self, YH_settter);

    NSString *getName = objc_getAssociatedObject(self, YH_getter);

    Classclass = [selfclass];//获取当前类型


    object_setClass(self, class_getSuperclass(class));//修改isa指针到父类


    //调用原类get, 获取旧值

    id oldValue = objc_msgSend(self, NSSelectorFromString(getName));

    //调用原类,设置新值

    objc_msgSend(self, NSSelectorFromString([setName stringByAppendingString:@":"]), newVlaue);


    id observer = objc_getAssociatedObject(self, YH_observer);


    NSMutableDictionary * change = [NSMutableDictionary dictionary];

    if(newVlaue) {

        change[NSKeyValueChangeNewKey] = newVlaue;

    }

    if(oldValue) {

        change[NSKeyValueChangeOldKey] = oldValue;

    }


    if(observer) {

                objc_msgSend(observer,@selector(observeValueForKeyPath:ofObject:change:context:), getName,self, change,nil);

    }

    //修改回子类,供下次使用

    object_setClass(self, class);

}

测试:不出意外一切正常

- (void)viewDidLoad {

    [super viewDidLoad];

    _p= [[Persionalloc]init];

    [_p YH_addObserver:self forKeyPath:NSStringFromSelector(@selector(age)) options:(NSKeyValueObservingOptionNew | NSKeyValueObservingOptionOld) context:nil];

}

- (void)observeValueForKeyPath:(NSString*)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void*)context {

    NSLog(@"%@",_p.age);

    NSLog(@"%@", change);

}

- (void)touchesBegan:(NSSet *)touches withEvent:(UIEvent*)event {

    staticinta =0;

    a++;

    _p.age = [NSString stringWithFormat:@"%d", a];

}


如有不足请指正

源码地址: GitHub - yeshibuzhong/iOS

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容