目标

最近写项目 ，写到需要为自己写的一个控件添加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静态方法。