RxSwift實現替換delegate的方法示例

這篇文章主要給大家介紹了關於RxSwift實現替換delegate的相關資料，文中通過示例代碼介紹的非常詳細，對大家學習或者使用RxSwift具有一定的參考學習價值，需要的朋友們下面來一起學習學習吧

目標

最近寫項目 ，寫到需要爲自己寫的一個控件添加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)

總結

以上就是這篇文章的全部內容了，希望本文的內容對大家的學習或者工作具有一定的參考學習價值，謝謝大家對神馬文庫的支持。