整體結(jié)構(gòu)
我們知道Delegate是一對一，NSNotification可以一對多，現(xiàn)在我們也想用代理實現(xiàn)多個對象同時監(jiān)聽，所以就有了多重代理這個東西。
我已經(jīng)對多重代理進(jìn)行了封裝，只需要把MulicastDelegate文件夾拖進(jìn)你的項目中就可以用了，包含MulticastDelegate.swift和Weak.swift兩個文件，git地址MulticastDelegate
直接上代碼，注釋的比較詳細(xì)，整體還是比較簡單的
import Foundation
/**
* `MulticastDelegate` 可以輕松地為給定的協(xié)議或類創(chuàng)建一個多重代理。
*/
open class MulticastDelegate<T>: NSObject {
/**
* 用來存儲delegate的數(shù)組
*/
fileprivate var delegates = [Weak]()
/**
* 用這個屬性來判斷delegates是否為空
*
* 如果數(shù)組delegates中一個元素都沒有返回true
*/
public var isEmpty: Bool {
return delegates.count == 0
}
/**
* 這個方法用來向delegates中添加新的delegate
*/
public func add(_ delegate: T) {
guard !self.contains(delegate) else {
return
}
delegates.append(Weak(value: delegate as AnyObject))
}
/**
* 這個方法用來刪除delegates中某個delegate
*/
public func remove(_ delegate: T) {
let weak = Weak(value: delegate as AnyObject)
if let index = delegates.index(of: weak) {
delegates.remove(at: index)
}
}
/**
* 這個方法用來觸發(fā)代理方法
*/
public func invoke(_ invocation: @escaping (T) -> ()) {
clearNil()
delegates.forEach({
if let delegate = $0.value as? T {
invocation(delegate)
}
})
}
/**
* 這個方法用來判斷delegates中是否已經(jīng)存在某個delegate
*/
private func contains(_ delegate: T) -> Bool {
return delegates.contains(Weak(value: delegate as AnyObject))
}
/**
* 這個方法用來移除delegates中無效的delegate
*/
private func clearNil() {
delegates = delegates.filter{ $0.value != nil }
}
}
/**
* 自定義操作符實現(xiàn)delegate的添加，同add
*
* - 參數(shù) left: The multicast delegate
* - 參數(shù) right: The delegate to be added
*/
public func +=<T>(left: MulticastDelegate<T>, right: T) {
left.add(right)
}
/**
* 自定義操作符實現(xiàn)delegate的刪除，同remove
*
* - 參數(shù) left: The multicast delegate
* - 參數(shù) right: The delegate to be removed
*/
public func -=<T>(left: MulticastDelegate<T>, right: T) {
left.remove(right)
}
/**
* 自定義類型添加Sequence，需要添加返回迭代器的方法 -> makeIterator()，實現(xiàn)for循環(huán)功能
*/
extension MulticastDelegate: Sequence {
public func makeIterator() -> AnyIterator<T> {
clearNil()
var iterator = delegates.makeIterator()
return AnyIterator {
while let next = iterator.next() {
if let delegate = next.value {
return delegate as? T
}
}
return nil
}
}
}
為了避免循環(huán)這里用了Proxy，這個在我那篇關(guān)于CADisplayLink內(nèi)存泄漏的文章也有提到，原理都是一樣的。
import Foundation
/**
* `Weak` 代理用來避免內(nèi)存泄漏
*/
class Weak: Equatable {
weak var value: AnyObject?
init(value: AnyObject) {
self.value = value
}
static func ==(lhs: Weak, rhs: Weak) -> Bool {
return lhs.value === rhs.value
}
}
這里需要提一下我看到有些人用NSHashTable來避免內(nèi)存泄漏。
private let delegates: NSHashTable<AnyObject> = NSHashTable.weakObjects()
NSHashTable和NSSet(NSMutableSet)很相似，但又有所不同，提供了比NSSet更多的操作選項，尤其是在對弱引用關(guān)系的支持上，NSHashTable在對象/內(nèi)存處理時更加的靈活。相較于NSSet，NSHashTable具有以下特性：
- NSSet(NSMutableSet)持有其元素的強(qiáng)引用，同時這些元素是使用hash值及isEqual:方法來做hash檢測及判斷是否相等的。
- NSHashTable是可變的，它沒有不可變版本。
- 它可以持有元素的弱引用，而且在對象被銷毀后能正確地將其移除。而這一點(diǎn)在NSSet是做不到的。
- 它的成員可以在添加時被拷貝。
- 它的成員可以使用指針來標(biāo)識是否相等及做hash檢測。
- 它可以包含任意指針，其成員沒有限制為對象。我們可以配置一個NSHashTable實例來操作任意的指針，而不僅僅是對象。
NSHashTable吸引人的地方在于可以持有元素的弱引用，而且在對象被銷毀后能正確地將其移除
在添加對象的操作中，NSHashTable所有的時間差不多是NSMutableSet的2倍，而在其它操作中，性能大體和NSSet相近。所以我選擇用了數(shù)組，其實差別不大，看個人喜好。
用途
說了這么多，看著很吊的樣子那到底錘子用哦，我拋磚迎玉提幾個場景，大家可以根據(jù)自己需要用到合適的地方。
1.類似IM庫，當(dāng)接受到消息，在幾個不同地方做回調(diào)，比如改變消息數(shù)，顯示小紅點(diǎn)等。
這個我在Demo中有寫到，用的時候可以像我一樣再封裝一層MessageManager
MessageDelegate.swift文件
import Foundation
protocol MessageDelegate {
func getNewMessage(message: String)
}
MessageManager.swift文件
import Foundation
class MessageManager: NSObject {
public var delegate = MulticastDelegate<MessageDelegate>()
static let shareManager = MessageManager()
override init() {
super.init()
}
}
使用
import UIKit
class ViewController: UIViewController, MessageDelegate {
var btn = UIButton()
override func viewDidLoad() {
super.viewDidLoad()
MessageManager.shareManager.delegate.add(self)
btn.frame = CGRect(x: 100, y: 100, width: 100, height: 100)
btn.backgroundColor = UIColor.red
btn.addTarget(self, action: #selector(bottonClick(sender:)), for: UIControlEvents.touchUpInside)
self.view.addSubview(btn)
}
@objc func bottonClick(sender: UIButton) {
MessageManager.shareManager.delegate.invoke { (delegate) in
delegate.getNewMessage(message: "您收到一條新消息啦")
}
}
func getNewMessage(message: String) {
print("ViewController \(message)")
}
override func didReceiveMemoryWarning() {
super.didReceiveMemoryWarning()
}
}
import UIKit
class SecondViewController: UIViewController, MessageDelegate{
override func viewDidLoad() {
super.viewDidLoad()
MessageManager.shareManager.delegate.add(self)
}
override func didReceiveMemoryWarning() {
super.didReceiveMemoryWarning()
}
func getNewMessage(message: String) {
print("SecondViewController \(message)")
}
}
2.希望多個對象能夠偵聽來自UINavigationController的委托回調(diào)，這個比上面那個略微高級一點(diǎn)。
可以根據(jù)需要可以添加更多的代理方法，使用在demo中有這里就不放了，不然文章太臭長了
import UIKit
class MulticastUINavigationControllerDelegate: MulticastDelegate<UINavigationControllerDelegate>, UINavigationControllerDelegate {
public override init() {
super.init()
}
@objc open func navigationController(_ navigationController: UINavigationController, willShow viewController: UIViewController, animated: Bool) {
for delegate in self {
delegate.navigationController?(navigationController, willShow: viewController, animated: animated)
}
}
func navigationControllerSupportedInterfaceOrientations(_ navigationController: UINavigationController) -> UIInterfaceOrientationMask {
return reduce(.all, { (mask, delegate) in
guard let newMask = delegate.navigationControllerSupportedInterfaceOrientations?(navigationController) else {
return mask
}
return mask.intersection(newMask)
})
}
}
3.降低耦合
比如UITableView繼承于UIScrollView,我們希望他的delegate中UIScrollViewDelegate的方法由一個獨(dú)立的類來處理，以便實現(xiàn)一些效果，比如像下圖這樣的頭部圖片滾動拉伸效果，只需要實現(xiàn)UIScrolViewDelegate的scrollViewDidScroll方法，這樣做的好處是可以降低代碼耦合度，將實現(xiàn)不同功能的方法封裝在獨(dú)立的delegate中，便于復(fù)用和維護(hù)管理。
import UIKit
class ScrollDelegate: MulticastDelegate<UIScrollViewDelegate>,UIScrollViewDelegate{
override init() {
super.init()
}
func scrollViewDidScroll(_ scrollView: UIScrollView) {
for delegate in self {
delegate.scrollViewDidScroll?(scrollView)
}
}
}
class CustomeHeaderView: UIView,UIScrollViewDelegate {
override init(frame: CGRect) {
super.init(frame: frame)
}
func scrollViewDidScroll(_ scrollView: UIScrollView) {
let offSetY = scrollView.contentOffset.y+20
if scrollView.contentOffset.y>0&&scrollView.contentOffset.y<100{
self.alpha = offSetY/100
}
}
required init?(coder aDecoder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
}
參考文章
NSHashTable的特性和使用
iOS開發(fā)中利用消息轉(zhuǎn)發(fā)實現(xiàn)多重代理
利用OC的消息轉(zhuǎn)發(fā)機(jī)制實現(xiàn)多重代理
MulticastDelegate
Multicast on Swift 3 and MVVM-C
Notifying many delegates at once with Multicast Delegates
