iOS - 自定义KVO

2018-12-26  本文已影响4人  ForScanf

之前我们已经了解过了KVO的底层实现原理,不过呢,在我们开始实现自定义KVO之前再来简单回顾下KVO的实现原理

1.创建子类
2.重写一个setter方法(其实是添加一个setter方法)
3.修改isa指针指向新创建的子类
4.调用父类的setName方法
5.将观察者保存到当前对象

接下来我们开始自定义KVO

我们先创建一个NSObject分类;然后实现对应的方法

#import <Foundation/Foundation.h>

NS_ASSUME_NONNULL_BEGIN

@interface NSObject (KVO)

- (void)ZXY_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(nullable void *)context;

@end

NS_ASSUME_NONNULL_END

1.创建子类

在.m文件里#import <objc/message.h>
创建完子类记得要注册这个类,

- (void)ZXY_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(void *)context{
    
    //1.创建子类
    NSString *oldClassName = NSStringFromClass(self.class);
    NSString *newClassName = [@"ZXYKVO_" stringByAppendingString:oldClassName];
    
    Class myClass = objc_allocateClassPair(self.class, newClassName.UTF8String, 0);
    
    // 注册
    objc_registerClassPair(myClass);    
    
}

2.重写一个setter方法(其实是添加一个setter方法)

- (void)ZXY_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(void *)context{
    
    //1.创建子类
    NSString *oldClassName = NSStringFromClass(self.class);
    NSString *newClassName = [@"ZXYKVO_" stringByAppendingString:oldClassName];
    
    Class myClass = objc_allocateClassPair(self.class, newClassName.UTF8String, 0);
    
    // 注册
    objc_registerClassPair(myClass);

    //2.重新setNanme方法(其实是添加一个setName方法)
    class_addMethod(myClass, @selector(setName:), (IMP)setMethod, "v@:@");
}

void setMethod(id self,SEL _cmd,NSString *newName){
   
    NSLog(@"来了%@",newName);
}

3.修改isa指针指向新创建的子类

- (void)ZXY_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(void *)context{
    
    //1.创建子类  
    NSString *oldClassName = NSStringFromClass(self.class);
    NSString *newClassName = [@"ZXYKVO_" stringByAppendingString:oldClassName];
    
    Class myClass = objc_allocateClassPair(self.class, newClassName.UTF8String, 0);
    
    // 注册
    objc_registerClassPair(myClass);

    //2.重新setNanme方法(其实是添加一个setName方法)
    class_addMethod(myClass, @selector(setName:), (IMP)setMethod, "v@:@");

    //3.修改imp地址
    object_setClass(self, myClass);
}

这个时候我们在控制器中注册这个观察者( 其中person是需要观察属性变化的实体类)

- (void)viewDidLoad {
    [super viewDidLoad];

    person = [[Person alloc]init];
    [person ZXY_addObserver:self forKeyPath:@"name" options:(NSKeyValueObservingOptionNew) context:nil];

}
//点击屏幕
- (void)touchesBegan:(NSSet<UITouch *> *)touches withEvent:(UIEvent *)event{
    
    static int a;
    person.name = [NSString stringWithFormat:@"%d",a++];
}

然后我们跑起来,点击屏幕: 控制台打印


屏幕快照 2018-12-25 下午4.33.43.png

4.调用父类的setName方法

到目前这个阶段,我们已经能获取到修改的值了,接下来我们需要调用父类的set方法,去修改属性值,这样person的name就修改了.

void setName(id self,SEL _cmd,NSString *newName){
    
    NSLog(@"来了%@",newName);
    
    //5. 调用父类的setName方法
    Class class = [self class];//拿到当前类型

    object_setClass(self, class_getSuperclass(class));// 将当前类型设置为父类类型

    objc_msgSend(self, @selector(setName:),newName);//给父类的set方法发送消息,传递修改的值

    // 改为子类
    object_setClass(self, class);
}
注意:在这一步一定要将当前的类型设置为父类类型,设置完后也一定要改为子类的类型

5.将观察者保存到当前对象

现在我们需要通知外部方法,已经将属性修改了,并将之传递出去,所以我们设置一个观察者

- (void)ZXY_addObserver:(NSObject *)observer forKeyPath:(NSString *)keyPath options:(NSKeyValueObservingOptions)options context:(void *)context{
    
    //1.创建子类   
    NSString *oldClassName = NSStringFromClass(self.class);
    NSString *newClassName = [@"ZXYKVO_" stringByAppendingString:oldClassName];
    
    Class myClass = objc_allocateClassPair(self.class, newClassName.UTF8String, 0);
    
    // 注册
    objc_registerClassPair(myClass);

    //2.重新setNanme方法(其实是添加一个setName方法)
    /**参数
     *class 给哪个类添加方法
     *sel   方法编号
     *imp   方法实现(函数指针)
     *type  返回值类型 ()
     */
    class_addMethod(myClass, @selector(setName:), (IMP)setMethod, "v@:@");

    //3.修改imp地址
    object_setClass(self, myClass);

    //4.将观察者保存到当前对象
   /**
     id object                     :表示关联者,是一个对象,变量名理所当然也是object
     const void *key               :获取被关联者的索引key
     id value                      :被关联者,这里是一个block
     objc_AssociationPolicy policy : 关联时采用的协议,有assign,retain,copy等协议,一般使用OBJC_ASSOCIATION_RETAIN_NONATOMIC,这里我们使用Weak协议,避免循环引用
     */
    objc_setAssociatedObject(self, "observer", observer, OBJC_ASSOCIATION_ASSIGN);//
}
在这里就有可能就有人问了,明明set方法只有一个返回类型,为什么要写“v@:@”,这里解释下,V表示的是返回void类型,第二个@表示的是一个id对象,第三个:表示的是方法SEL,最后一个@表示的传入的变量值.
⚠️注意:以为OC方法会默认两个参数,一个id对象,一个方法SEL

然后我们在set方法中获取到这个观察者

void setName(id self,SEL _cmd,NSString *newName){

    NSLog(@"来了%@",newName);
    
    //5. 调用父类的setName方法
    Class class = [self class];//拿到当前类型

    object_setClass(self, class_getSuperclass(class));

    objc_msgSend(self, @selector(setName:),newName);

    // 观察者
    id observer = objc_getAssociatedObject(self, "observer");

    // 改为子类
    object_setClass(self, class);
}

给系统的observeValueForKeyPath方法发送消息

void setName(id self,SEL _cmd,NSString *newName){

    NSLog(@"来了%@",newName);
    
    //5. 调用父类的setName方法
    Class class = [self class];//拿到当前类型

    object_setClass(self, class_getSuperclass(class));

    objc_msgSend(self, @selector(setName:),newName);

    // 观察者
    id observer = objc_getAssociatedObject(self, "observer");

    if (observer) {
        objc_msgSend(observer, @selector(observeValueForKeyPath:ofObject:change:context:),@"name",self,@{@"new:":newName,@"kind:":@1},nil);
    }

    // 改为子类
    object_setClass(self, class);
}

其中observeValueForKeyPath方法的参数说明:
keyPath: 属性名字
object: 哪个对象
change: 一个字典,它描述对键路径keyPath中的属性值相对于对象所做的更改
context: 当注册观察者以接收键值观察通知时提供的值,这里写为nil就行.

最后

现在我们在外部控制器中打印修改后的值

- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary<NSKeyValueChangeKey,id> *)change context:(void *)context{
    
    NSLog(@"%@",change);
    
}
屏幕快照 2018-12-26 下午2.23.06.png

到这里,自定义KVO就简单实现了,但是目前里面的属性变量名都是固定,如果能自定义的变化,那就更完美了.降下来,我们就来慢慢的研究他.

上一篇下一篇

猜你喜欢

热点阅读