iOS-自定义实现通知功能

此篇为模仿系统NSNotificationCenter的底层实现，为我自己的构想，具体底层如何实现，不得而知

首先，自定义文件名为SFTongzhiCenter，因为通知是多对多的关系，所以先定义一个可变字典保存印射关系，

@interface SFTongzhiCenter ()

@property (nonatomic, strong) NSMutableDictionary *classMap;

@end

@implementation SFTongzhiCenter
+ (instancetype)defaultCenter{
    static dispatch_once_t onceToken;
    static SFTongzhiCenter *tongzhi = nil;
    dispatch_once(&onceToken, ^{
        tongzhi = [SFTongzhiCenter new];
        tongzhi.classMap = [NSMutableDictionary dictionary];
    });
    return tongzhi;
}
@end

因为涉及到通知传参，所以我们自定义一个类SFNotification来保存通知，里面可以涉及三个参数，一个通知名，一个参数信息，一个通知object！因为这些参数外界只能查看，不能更改，所以都定义为readonly属性，还有一个可以快速构建类的方法！

@interface SFNotification : NSObject
@property (readonly, copy) NSNotificationName name;
@property (nullable, readonly, retain) id object;
@property (nullable, readonly, copy) NSDictionary *userInfo;

- (instancetype)initWithName:(NSString *)name object:(nullable id)object userInfo:(nullable NSDictionary *)userInfo;

@end

在.m中可以重写属性，实现可读可写

@interface SFNotification ()

@property (nonatomic, copy) NSString *name;
@property (nonatomic, retain) id object;
@property (nonatomic, copy) NSDictionary *userInfo;

@end

@implementation SFNotification

- (instancetype)initWithName:(NSString *)name object:(nullable id)object userInfo:(nullable NSDictionary *)userInfo{
    SFNotification *notification = [SFNotification new];
    notification.name = name;
    notification.object = object;
    notification.userInfo = userInfo;
    return notification;
}

@end

对外开放添加通知和发送通知以及删除通知的方法，因为通知可传参，也可不传参，故暴露两个方法，一个有参，一个无参

@interface SFTongzhiCenter : NSObject

+ (instancetype)defaultCenter;

- (void)addObserver:(id)observer selector:(SEL)aSelector name:(nullable NSString *)aName object:(nullable id)anObject;

- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject;
- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject userInfo:(nullable NSDictionary *)aUserInfo;

- (void)removeObserver:(id)observer;
- (void)removeObserver:(id)observer name:(nullable NSNotificationName)aName object:(nullable id)anObject;

@end

首先是添加通知的逻辑，因为添加通知时，会传进来三个参数，一个是对应的实例类，一个是实例方法，还有一个是通知object，所以我们要把这三个信息都保存下来，并且通知是多对多，所以我们用字典来印射他们的关系，以通知名为key，一个可变数组为value，这个value中则保存通知aName所属的所有类信息

- (void)addObserver:(id)observer selector:(SEL)aSelector name:(nullable NSString *)aName object:(nullable id)anObject{
    NSMutableArray *array = self.classMap[aName];
    if (!array) {
        array = [NSMutableArray array];
    }
    [array addObject:@{@"class":observer,@"selector":NSStringFromSelector(aSelector),@"object":anObject}];
    [self.classMap setObject:array forKey:aName];
}

发送通知逻辑即为找到添加了名为aName的所有通知的类信息进行遍历，依次进行方法调用，系统底层我猜测应该直接是用objc_msgSend调用的，这里我们用系统封装好的NSInvocation类进行调用

- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject{
    [self postNotificationName:aName object:anObject userInfo:nil];
}
- (void)postNotificationName:(NSString *)aName object:(nullable id)anObject userInfo:(nullable NSDictionary *)aUserInfo{
    NSMutableArray *array = self.classMap[aName];
    for (NSDictionary *mapDic in array) {
        //当mapDic中的object与anObject一致时，才调用方法
        if ([mapDic[@"object"] isEqual:anObject]) {
            //NSInvocation;用来包装方法和对应的对象，它可以存储方法的名称，对应的对象，对应的参数,
            /*
             NSMethodSignature：签名：再创建NSMethodSignature的时候，必须传递一个签名对象，签名对象的作用：用于获取参数的个数和方法的返回值
             */
            //创建签名对象的时候不是使用NSMethodSignature这个类创建，而是方法属于谁就用谁来创建
            NSMethodSignature *signature = [[mapDic[@"class"] class] instanceMethodSignatureForSelector:NSSelectorFromString(mapDic[@"selector"])];
            //1、创建NSInvocation对象
            NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:signature];
            invocation.target = mapDic[@"class"];
            //invocation中的方法必须和签名中的方法一致。
            invocation.selector = NSSelectorFromString(mapDic[@"selector"]);
            /*第一个参数：需要给指定方法传递的值
                   第一个参数需要接收一个指针，也就是传递值的时候需要传递地址*/
            //第二个参数：需要给指定方法的第几个参数传值
            //注意：设置参数的索引时不能从0开始，因为0已经被self占用，1已经被_cmd占用
            SFNotification *noti = [[SFNotification alloc] initWithName:aName object:anObject userInfo:aUserInfo];
            [invocation setArgument:&noti atIndex:2];
            //2、调用NSInvocation对象的invoke方法
            //只要调用invocation的invoke方法，就代表需要执行NSInvocation对象中制定对象的指定方法，并且传递指定的参数
            [invocation invoke];
        }
    }
}

删除通知时，找到对应的类，在classMap中删除即可

- (void)removeObserver:(id)observer{
    NSMutableDictionary *tmpDict = [NSMutableDictionary dictionary];
    [self.classMap enumerateKeysAndObjectsUsingBlock:^(id  _Nonnull key, id  _Nonnull obj, BOOL * _Nonnull stop) {
        NSMutableArray *tmpArray = [obj mutableCopy];
        for (NSDictionary *mapDic in obj) {
            if ([mapDic[@"class"] isKindOfClass:[observer class]]) {
                [tmpArray removeObject:mapDic];
            }
        }
        [tmpDict setObject:tmpArray forKey:key];
    }];
    self.classMap = tmpDict;
}
- (void)removeObserver:(id)observer name:(nullable NSNotificationName)aName object:(nullable id)anObject{
    NSMutableArray *array = self.classMap[aName];
    NSMutableArray *tmpArray = [array mutableCopy];
    for (NSDictionary *mapDic in array) {
        if ([mapDic[@"class"] isKindOfClass:[observer class]]) {
            if ([mapDic[@"object"] isEqual:anObject] || !anObject) {
                [tmpArray removeObject:mapDic];
            }
        }
    }
    [self.classMap setObject:tmpArray forKey:aName];
}

由此我们的自定义通知功能就完成了，使用和体验与系统原生通知毫无区别，Done

附Demo地址