RxSwift實現(xiàn)替換delegate的方法示例
目標(biāo)
最近寫項目 ,寫到需要為自己寫的一個控件添加rx訂閱方式的案例。
目前有一個代理:
// 代理方式獲取結(jié)果
@objc public protocol ZZPhotoPickerControllerDelegate : NSObjectProtocol {
@objc optional func photoPickerController(_ photoPickerController: ZZPhotoPickerController, didSelect assets: [Any])
}
需要寫一個能夠?qū)崿F(xiàn)下邊這種方式的擴(kuò)展
photoPickerController.rx.assetsSelected.subscribe(onNext: { assets in
// do something
}
思路
剛開始完全摸不著頭腦。后來想到Rx寫了對UICollectionViewDelegate的擴(kuò)展:
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產(chǎn)生。delegate是什么delegate?為什么會有methodInvoked方法?于是繼續(xù)點進(jìn)去。
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>類型,根據(jù)字面是理解就是代理的代理。然后還看到這里的rx是擴(kuò)展自UIScrollView的,UICollectionView是繼承自UIScrollView,可以知道這里的delegate也是繼承過來的使用的。還可以看到RxScrollViewDelegateProxy這個東西,可以想到如果我們要仿寫的話,自己也應(yīng)該寫這樣一個代理的代理類。先點進(jìn)去看看:
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) }
}
/// ...后面的感覺沒什么關(guān)系,先不看
}
可以看到它其實是一個DelegateProxy<UIScrollView, UIScrollViewDelegate>,并且遵守了DelegateProxyType和UIScrollViewDelegate協(xié)議,可以感覺出它是一個鏈接rx和delegate的紐帶。有一個實例變量scrollView,有一個init方法,有一個registerKnownImplementations靜態(tài)方法。
現(xiàn)在腦海中大概有一個模糊的思路:我們要先創(chuàng)建一個紐帶delegateProxy對象,然后在目標(biāo)類的rx擴(kuò)展中創(chuàng)建一個delegateProxy實例,最后在我們的assetsSelected事件流中用這個delegateProxy的methodInvoked截獲delegate中的目標(biāo)方法,并生成可訂閱的Observable返回給controlEvent,這樣鏈接打通。
開動
首先創(chuàng)建一個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) }
}
// 把上面的寫好后,編輯器會提示你需要實現(xiàn)一下兩個方法,一個是獲取,一個是設(shè)置,所以很好理解該在方法里實現(xiàn)什么。
static func currentDelegate(for object: ZZPhotoPickerController) -> ZZPhotoPickerControllerDelegate? {
return object.zzDelegate
}
static func setCurrentDelegate(_ delegate: ZZPhotoPickerControllerDelegate?, to object: ZZPhotoPickerController) {
object.zzDelegate = delegate
}
}
然后給目標(biāo)的rx擴(kuò)展寫一個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并沒有開放出來,是個內(nèi)部方法,如果照著寫報錯??梢匝芯砍鲈摲椒ㄖ皇且粋€類型推斷而已,所以可以簡單寫。
完成
然后就可以愉快的去對assetsSelected進(jìn)行訂閱了。
vc.rx.assetsSelected.subscribe(onNext: { (assets) in
// do something
}).disposed(by: self.disposeBag)
總結(jié)
以上就是這篇文章的全部內(nèi)容了,希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,謝謝大家對腳本之家的支持。
相關(guān)文章
利用swift實現(xiàn)卡片橫向滑動動畫效果的方法示例
卡片橫向滑動動畫效果相信對大家來說都不陌生,下面這篇文章主要給大家介紹了關(guān)于利用swift實現(xiàn)卡片橫向滑動動畫效果的方法示例,文中通過示例代碼介紹的非常詳細(xì),對大家具有一定的參考學(xué)習(xí)價值,需要的朋友們下面來一起看看吧。2017-07-07

