Swift小技巧通知转协议,让你再也不用通知做跨层传递

前言

iOS开发中,如果我们需要做页面的跨层级传递,比如A-B-C三层界面,想要C传递到A如果不想写多次闭包或者代理,那么我们一般会用NSNotification做传递,也就是通知。通知可以一对多,只要添加了观察者,然后在需要传递的地方postNotification就可以收到通知了,非常的方便。但是它也有它不好的地方,iOS9之前通知观察了因为是强引用,导致如果不调用removeObserver会内存泄露,当然现在已经不存在这个问题了。另外多次addObserver会接受到多次方法,导致代码可能出现问题,还有最后一个老生常谈的问题,如果Key写错了也容易导致出错。那么有没有一种方法能解决跨层传递,又不需要用到Notfication呢。答案是有的,我们自己封装一个MessageCenter类,用协议来传递不就可以了?下面我们说干就干。

实现过程

  • 定义一个消息协议

我们期望兼容OC,所以使用NSObjectProtocol。

less 复制代码
@objc protocol JFMessageProtocol: NSObjectProtocol {

@objc optional func messageProtocol_testMethod()

}
  • 定义消息中心类

定义消息中心类,内部的pushClients是一个弱引用的哈希Map用来装观察者,这样就无需担心没有remove观察者导致的内存泄露问题。obs方法则和通知的addObserver方法是一样的,用于添加观察者。sendMessage方法和postNotification方法类似,但是这里发送的是协议方法,这样这样添加了观察的类并且实现了该协议方法,就可以完成一次消息传递啦。

就这短短的20行代码,我们就从此可以丢掉通知了,改用协议做跨层传递。

swift 复制代码
class JFMessageCenter: NSObject {

@objc dynamic static let shared = JFMessageCenter()

private var pushClients = NSHashTable<AnyObject>.weakObjects()

@objc func obs(with target: AnyObject) {

self.pushClients.add(target)

}

//因为是弱引用,不手动调用也没事

@objc func removeObs(with target: AnyObject) {

self.pushClients.remove(target)

}

@objc public func sendMessage(selector: Selector ,completion: @escaping (JFMessageProtocol) -> Void) {

for client in self.pushClients.allObjects {

if client.responds(to: selector), let client = client as? JFMessageProtocol {

print("receive message")

if Thread.current != .main {

DispatchQueue.main.async {

completion(client)

}

} else {

completion(client)

}

}

}

}

}

使用方法

在需要观察的地方观察,并实现要调用的协议方法

scss 复制代码
override func viewDidLoad() {
    super.viewDidLoad()
    JFMessageCenter.shared.obs(with: self)
}

extension ViewController: JFMessageProtocol {
    func messageProtocol_testMethod() {
        JFPopup.toast {
            [                .hit("我是通知转协议"),                .position(.center)            ]
        }
//        JFPopup.toast(hit: "我是通知转协议")
        print("我是通知转协议")
        print(Thread.current)
    }
}

在想发送消息的地方,发送消息即可,非常方便

javascript 复制代码
JFMessageCenter.shared.sendMessage(selector: #selector(JFMessageProtocol.messageProtocol_testMethod)) { proto in

proto.messageProtocol_testMethod?()

}

是否要加锁的问题

上面我写的类是没有加锁的,考虑我自身是用来做页面的跨层传递,一般都是在控制器或者视图层的初始化里添加观察,都是主线程,必然优先观察到,所以就没有考虑加锁的问题。这里仅提供一个思路,如果想用该方法同学想严谨一点,可以自行补上。

测试Demo地址

MessageProtocolDemo

结尾

以上就是我说的Swift小技巧,如有说得不对的地方请多包涵,欢迎互相讨论🤚

相关推荐
杂雾无尘27 分钟前
开发者必看,全面解析应用更新策略,让用户无法拒绝你的应用更新!
ios·xcode·swift
xiangzhihong82 小时前
使用Universal Links与Android App Links实现网页无缝跳转至应用
android·ios
Digitally3 小时前
如何将iPhone备份到Mac/MacBook
macos·ios·iphone
帅次4 小时前
【iOS设计模式】深入理解MVC架构 - 重构你的第一个App
ios·swiftui·objective-c·iphone·swift·safari·cocoapods
东坡肘子9 小时前
高温与奇怪的天象 | 肘子的 Swift 周报 #092
人工智能·swiftui·swift
Swift社区9 小时前
Swift 解 LeetCode 320:一行单词有多少种缩写可能?用回溯找全解
开发语言·leetcode·swift
Frank学习路上20 小时前
【IOS】XCode创建firstapp并运行(成为IOS开发者)
开发语言·学习·ios·cocoa·xcode
瓜子三百克1 天前
CALayer的异步处理
macos·ios·cocoa
吴Wu涛涛涛涛涛Tao1 天前
一步到位:用 Very Good CLI × Bloc × go_router 打好 Flutter 工程地基
flutter·ios
杂雾无尘1 天前
开发者必看:如何在 iOS 应用中完美实现动态自定义字体!
ios·swift·apple