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小技巧,如有说得不对的地方请多包涵,欢迎互相讨论🤚

相关推荐
YungFan2 小时前
iOS26适配指南之UINavigationController
ios·swift
Swift社区3 小时前
Swift 解 LeetCode 321:拼接两个数组中的最大数,贪心 + 合并全解析
开发语言·leetcode·swift
木叶丸3 小时前
Flutter 生命周期完全指南
android·flutter·ios
Keya7 小时前
解决 pod install 警告信息显示不完全的方法
ios·xcode·cocoapods
随笔记9 小时前
uniapp开发的小程序输入框在ios自动填充密码,如何欺骗苹果手机不让自动填充
前端·ios·app
unicrom_深圳市由你创科技10 小时前
Unity开发如何解决iOS闪退问题
unity·ios·蓝桥杯
TE-茶叶蛋10 小时前
iOS打包流程
ios
大熊猫侯佩11 小时前
Swift 数学计算:用 Accelerate 框架让性能“加速吃鸡”
算法·swift
大熊猫侯佩11 小时前
Swift 6.2 并发江湖:两大神功破局旧制,代码运行经脉革新(下)
swiftui·swift·wwdc