来源：wazrx
链接：http://www.jianshu.com/p/70b2503d5fd1
写在前面
每次使用KVO和通知我就觉得是一件麻烦的事情，即便谈不上麻烦，也可说是不方便吧，对于KVO，你需要注册，然后实现监听方法，最后还要移除，通知当然也需要移除操作，这使得相关逻辑的代码过于分散，控制器搞得乱乱的，而且总有时候会忘记移除什么的，总之感觉不太好，所以我想如果能有方法添加一个KVO或者通知后能够省略后面移除或者实现监听方法步骤的话会多好，所以我就尝试写了一个分类，这个分类的目的在于尽可能简化KVO和通知的步骤，对于KVO，你只需要一句代码就可完成监听，无需自己手动移除，通知也差不多,接口如下：
/**
* 通过Block方式注册一个KVO，通过该方式注册的KVO无需手动移除，其会在被监听对象销毁的时候自动移除
*
* @param keyPath 监听路径
* @param block KVO回调block，obj为监听对象，oldVal为旧值，newVal为新值
*/
- (void)xw_addObserverBlockForKeyPath:(NSString*)keyPath block:(void (^)(id obj, id oldVal, id newVal))block;
/**
* 通过block方式注册通知，通过该方式注册的通知无需手动移除，同样会自动移除
*
* @param name 通知名
* @param block 通知的回调Block，notification为回调的通知对象
*/
- (void)xw_addNotificationForName:(NSString *)name block:(void (^)(NSNotification *notification))block;
使用也很简单咯，github地址如下:https://github.com/wazrx/XWEasyKVONotification.git
你只需要导入NSObject+XWAdd这个分类，然后调用上面两个接口即可完成KVO和通知，事例代码如下
//监听_objA的name属性
[_objA xw_addObserverBlockForKeyPath:@"name" block:^(id obj, id oldVal, id newVal) {
NSLog(@"kvo，修改name为%@", newVal);
}];
[self xw_addNotificationForName:@"XWTestNotificaton" block:^(NSNotification *notification) {
NSLog(@"收到通知：%@", notification.userInfo);
}];
是不是非常简单，再也不用关心忘记移除导致的崩溃了，而且代码也集中，看着也更舒服了
原理
1、由于KVO和通知都差不多，原理部分通过KVO的接口的的实现原理进行说明，考虑到代码的统一我首先考虑到使用block，同时为了block能回调，我们需要一个内部的对象target的来实现KVO的代码，在监听到值改变的时候通过这个对象来回调block，同时一个target应该对应一个keyPath，并且可应该对应多个Block，因为我们可能对一个keyPath进行多处监听，这个类的具体代码大致如下：
@interface _XWBlockTarget : NSObject
/**添加一个KVOblock*/
- (void)xw_addBlock:(void(^)(__weak id obj, id oldValue, id newValue))block;
@end
@implementation _XWBlockTarget{
//保存所有的KVOblock
NSMutableSet *_blockSet;
}
- (instancetype)init
{
self = [super init];
if (self) {
_blockSet = [NSMutableSet new];
}
return self;
}
- (void)xw_addBlock:(void(^)(__weak id obj, id oldValue, id newValue))block{
[_blockSet addObject:[block copy]];
}
//KVO的真正实现
- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)object change:(NSDictionary *)change context:(void *)context{
if (!_blockSet.count) return;
BOOL prior = [[change objectForKey:NSKeyValueChangeNotificationIsPriorKey] boolValue];
//只接受值改变时的消息
if (prior) return;
NSKeyValueChange changeKind = [[change objectForKey:NSKeyValueChangeKindKey] integerValue];
if (changeKind != NSKeyValueChangeSetting) return;
id oldVal = [change objectForKey:NSKeyValueChangeOldKey];
if (oldVal == [NSNull null]) oldVal = nil;
id newVal = [change objectForKey:NSKeyValueChangeNewKey];
if (newVal == [NSNull null]) newVal = nil;
//当KVO触发，值改变的时候执行该target下的所有block
[_blockSet enumerateObjectsUsingBlock:^(void (^block)(__weak id obj, id oldVal, id newVal), BOOL * _Nonnull stop) {
block(object, oldVal, newVal);
}];
}
@end
2、实际进行KVO的监听的对象有了，我们就可以开始书写逻辑了，我们给每一个对象绑定一个targets的字典，每次调用该API注册KVO的就去判断有没有对应的keyPath下的target(target和keyPath一一对应)，没有就创建，同时注册这个keyPath的KVO，有就把block加入这个target以便回调，具体代码如下：
- (void)xw_addObserverBlockForKeyPath:(NSString*)keyPath block:(void (^)(id obj, id oldVal, id newVal))block {
if (!keyPath || !block) return;
//取出存有所有KVOTarget的字典
NSMutableDictionary *allTargets = objc_getAssociatedObject(self, XWKVOBlockKey);
if (!allTargets) {
//没有则创建
allTargets = [NSMutableDictionary new];
//绑定在该对象中
objc_setAssociatedObject(self, XWKVOBlockKey, allTargets, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
//获取对应keyPath中的所有target
_XWBlockTarget *targetForKeyPath = allTargets[keyPath];
if (!targetForKeyPath) {
//没有则创建
targetForKeyPath = [_XWBlockTarget new];
//保存
allTargets[keyPath] = targetForKeyPath;
//如果第一次，则注册对keyPath的KVO监听
[self addObserver:targetForKeyPath forKeyPath:keyPath options:NSKeyValueObservingOptionNew | NSKeyValueObservingOptionOld context:NULL];
}
[targetForKeyPath xw_addBlock:block];
//对第一次注册KVO的类进行dealloc方法调剂
[self _xw_swizzleDealloc];
}
3、上一段代码的最后一个方法是对dealloc方法进行调剂，因为我们想要能够在合适的时候自动注销KVO，何为合适的地方呢，当然是被监听对象销毁的时候才是最合适的地方，所以dealloc方法里面是最合适的地方，我们期望能交换被监听对象的dealloc方法然后自己在该方法中实现注销KVO的逻辑，最先能想到的方式是通常我们使用的runtime中的swizzle黑魔法直接进行方法交换，但遗憾的是swizzle黑魔法只能在本类中交换本类的方法，而无法在一个类中对另一个类的方法进行调剂，所以需要另想调剂方法，我们采取直接对变监听对象所在的类修改或者添加dealloc方法来达到调剂目的，我结合代码进行说明：
/**
* 调剂dealloc方法，由于无法直接使用运行时的swizzle方法对dealloc方法进行调剂，所以稍微麻烦一些
*/
- (void)_xw_swizzleDealloc{
//我们给每个类绑定上一个值来判断dealloc方法是否被调剂过，因为一个类只需要调剂一次，如果调剂过了就无需再次调剂了
BOOL swizzled = [objc_getAssociatedObject(self.class, deallocHasSwizzledKey) boolValue];
//如果调剂过则直接返回
if (swizzled) return;
//开始调剂
Class swizzleClass = self.class;
//获取原有的dealloc方法
SEL deallocSelector = sel_registerName("dealloc");
//初始化一个函数指针用于保存原有的dealloc方法
__block void (*originalDealloc)(__unsafe_unretained id, SEL) = NULL;
//实现我们自己的dealloc方法，通过block的方式
id newDealloc = ^(__unsafe_unretained id objSelf){
//在这里我们移除所有的KVO
[objSelf xw_removeAllObserverBlocks];
//根据原有的dealloc方法是否存在进行判断
if (originalDealloc == NULL) {//如果不存在，说明本类没有实现dealloc方法，则需要向父类发送dealloc消息(objc_msgSendSuper)
//构造objc_msgSendSuper所需要的参数，.receiver为方法的实际调用者，即为类本身，.super_class指向其父类
struct objc_super superInfo = {
.receiver = objSelf,
.super_class = class_getSuperclass(swizzleClass)
};
//构建objc_msgSendSuper函数
void (*msgSend)(struct objc_super *, SEL) = (__typeof__(msgSend))objc_msgSendSuper;
//向super发送dealloc消息
msgSend(&superInfo, deallocSelector);
}else{//如果存在，表明该类实现了dealloc方法，则直接调用即可
//调用原有的dealloc方法
originalDealloc(objSelf, deallocSelector);
}
};
//根据block构建新的dealloc实现IMP
IMP newDeallocIMP = imp_implementationWithBlock(newDealloc);
//尝试添加新的dealloc方法，如果该类已经复写的dealloc方法则不能添加成功，反之则能够添加成功
if (!class_addMethod(swizzleClass, deallocSelector, newDeallocIMP, "v@:")) {
//如果没有添加成功则保存原有的dealloc方法，用于新的dealloc方法中，执行原有的系统的dealloc逻辑
Method deallocMethod = class_getInstanceMethod(swizzleClass, deallocSelector);
originalDealloc = (void(*)(__unsafe_unretained id, SEL))method_getImplementation(deallocMethod);
originalDealloc = (void(*)(__unsafe_unretained id, SEL))method_setImplementation(deallocMethod, newDeallocIMP);
}
//标记该类已经调剂过了
objc_setAssociatedObject(self.class, deallocHasSwizzledKey, @(YES), OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}
/**移除所有的KVO*/
- (void)xw_removeAllObserverBlocks {
NSMutableDictionary *allTargets = objc_getAssociatedObject(self, XWKVOBlockKey);
if (!allTargets) return;
[allTargets enumerateKeysAndObjectsUsingBlock:^(id key, _XWBlockTarget *target, BOOL *stop) {
[self removeObserver:target forKeyPath:key];
}];
[allTargets removeAllObjects];
}
通过如上方式，我们就完成了对dealloc方法的调剂，新的dealloc方法执行的时候回注销注册的KVO，这样就免去了手动注销的麻烦事情咯！
写在最后
通知的大致实现方式和KVO一样，详情请自行查看代码咯，我就不多做说明了，现在终于能优雅愉快的使用KVO和通知了，复习一下github地址:https://github.com/wazrx/XWEasyKVONotification.git