RxSwift 實現替換delegate

目標

最近寫項目 ，寫到須要爲本身寫的一個控件添加rx訂閱方式的案例。
目前有一個代理：

// 代理方式獲取結果
@objc public protocol ZZPhotoPickerControllerDelegate : NSObjectProtocol {
    @objc optional func photoPickerController(_ photoPickerController: ZZPhotoPickerController, didSelect assets: [Any])
}

須要寫一個可以實現下邊這種方式的擴展

photoPickerController.rx.assetsSelected.subscribe(onNext: { assets in
  // do something
}

思路

剛開始徹底摸不着頭腦。後來想到Rx寫了對UICollectionViewDelegate的擴展：

collectionView.rx.itemSelected.subscribe(onNext: { indexPath in
  // do something
}

跟個人需求是同樣的。
因而就去看itemSelected的源代碼：

/// Reactive wrapper for `delegate` message `collectionView(_:didSelectItemAtIndexPath:)`.
    public var itemSelected: ControlEvent<IndexPath> {
        let source = delegate.methodInvoked(#selector(UICollectionViewDelegate.collectionView(_:didSelectItemAt:)))
            .map { a in
                return try castOrThrow(IndexPath.self, a[1])
            }
        
        return ControlEvent(events: source)
    }

souce是一個Observable，由delegate.methodInvoked產生。delegate是什麼delegate？爲何會有methodInvoked方法？因而繼續點進去。

extension Reactive where Base: UIScrollView {
         /// ...這部分代碼省略不用看

        /// Reactive wrapper for `delegate`.
        ///
        /// For more information take a look at `DelegateProxyType` protocol documentation.
        public var delegate: DelegateProxy<UIScrollView, UIScrollViewDelegate> {
            return RxScrollViewDelegateProxy.proxy(for: base)
        }
        
        /// ...後面的代碼暫時也不用看
}

能夠看到delegate是一個DelegateProxy<UIScrollView, UIScrollViewDelegate>類型，根據字面是理解就是代理的代理。而後還看到這裏的rx是擴展自UIScrollView的，UICollectionView是繼承自UIScrollView，能夠知道這裏的delegate也是繼承過來的使用的。還能夠看到RxScrollViewDelegateProxy這個東西，能夠想到若是咱們要仿寫的話，本身也應該寫這樣一個代理的代理類。先點進去看看：

open class RxScrollViewDelegateProxy
    : DelegateProxy<UIScrollView, UIScrollViewDelegate>
    , DelegateProxyType 
    , UIScrollViewDelegate {

    /// Typed parent object.
    public weak private(set) var scrollView: UIScrollView?

    /// - parameter scrollView: Parent object for delegate proxy.
    public init(scrollView: ParentObject) {
        self.scrollView = scrollView
        super.init(parentObject: scrollView, delegateProxy: RxScrollViewDelegateProxy.self)
    }

    // Register known implementations
    public static func registerKnownImplementations() {
        self.register { RxScrollViewDelegateProxy(scrollView: $0) }
        self.register { RxTableViewDelegateProxy(tableView: $0) }
        self.register { RxCollectionViewDelegateProxy(collectionView: $0) }
        self.register { RxTextViewDelegateProxy(textView: $0) }
    }

    /// ...後面的感受沒什麼關係，先不看
}

能夠看到它實際上是一個DelegateProxy<UIScrollView, UIScrollViewDelegate>，而且遵照了DelegateProxyType和UIScrollViewDelegate協議，能夠感受出它是一個連接rx和delegate的紐帶。有一個實例變量scrollView，有一個init方法，有一個registerKnownImplementations靜態方法。
如今腦海中大概有一個模糊的思路：咱們要先建立一個紐帶delegateProxy對象，而後在目標類的rx擴展中建立一個delegateProxy實例，最後在咱們的assetsSelected事件流中用這個delegateProxy的methodInvoked截獲delegate中的目標方法，並生成可訂閱的Observable返回給controlEvent，這樣連接打通。

開動

首先建立一個RxPhotoPickerControllerDelegateProxy

class RxPhotoPickerControllerDelegateProxy: DelegateProxy<ZZPhotoPickerController, ZZPhotoPickerControllerDelegate>, DelegateProxyType, ZZPhotoPickerControllerDelegate {
    
    /// Typed parent object.
    public weak private(set) var photoPickerController: ZZPhotoPickerController?
    
    /// - parameter scrollView: Parent object for delegate proxy.
    public init(photoPickerController: ParentObject) {
        self.photoPickerController = photoPickerController
        super.init(parentObject: photoPickerController, delegateProxy: RxPhotoPickerControllerDelegateProxy.self)
    }
    
    static func registerKnownImplementations() {
        self.register { RxPhotoPickerControllerDelegateProxy(photoPickerController: $0) }
    }
    
    // 把上面的寫好後，編輯器會提示你須要實現一下兩個方法，一個是獲取，一個是設置，因此很好理解該在方法裏實現什麼。
    static func currentDelegate(for object: ZZPhotoPickerController) -> ZZPhotoPickerControllerDelegate? {
        return object.zzDelegate
    }
    
    static func setCurrentDelegate(_ delegate: ZZPhotoPickerControllerDelegate?, to object: ZZPhotoPickerController) {
        object.zzDelegate = delegate
    }
    
}

而後給目標的rx擴展寫一個delegateProxy實例：

extension Reactive where Base: ZZPhotoPickerController {
    
    public var zzDelegate: DelegateProxy<ZZPhotoPickerController, ZZPhotoPickerControllerDelegate> {
        return RxPhotoPickerControllerDelegateProxy.proxy(for: base)
    }
    
}

最後寫咱們的assetsSelected：

extension Reactive where Base: ZZPhotoPickerController {
    
    var assetsSelected: ControlEvent<[Any]> {
        let source: Observable<[Any]> = self.zzDelegate.methodInvoked(#selector(ZZPhotoPickerControllerDelegate.photoPickerController(_:didSelect:))).map { a in
            return a[1] as! [Any]
        }
        return ControlEvent.init(events: source)
    }
    
}

要注意裏面有個方法castOrThrow，這個方法rx並無開放出來，是個內部方法，若是照着寫報錯。能夠研究出該方法只是一個類型推斷而已，因此能夠簡單寫。

完成

而後就能夠愉快的去對assetsSelected進行訂閱了。

vc.rx.assetsSelected.subscribe(onNext: { (assets) in
                // do something
            }).disposed(by: self.disposeBag)