RxSwift + MJRefresh 打造自动处理刷新控件状态

本文是基于 iOS - RxSwift 项目实战记录 所述，若是你还未阅读过，建议你最好还先阅读一遍，并下载Demo熟悉一下 : )

前言

MVVM的模式中，多出了ViewModel这个角色，将逻辑处理、网络请求等繁杂操做中ViewController中抽离出来，ViewController得以瘦身。 结合RxSwift架构，咱们通常就会在ViewModel中定义一个input收集繁杂操做所需的信息，经过一个transform方法将input做为参数传入，进而获得一个output供controller使用。

在使用RxSwift开发时会大量的使用到这种形式，其中就包括咱们的网络请求。 结合 iOS - RxSwift 项目实战记录 中所述的“MJRefresh在RxSwift中的使用”，在output中定义了一个变量

let refreshStatus = Variable<LXFRefreshStatus>(.none)
复制代码

controller经过output将其进行监听，从而当值发生变化时，controller就能实时获取当前应所处的刷新状态

vmOutput.refreshStatus.asObservable().subscribe(onNext: {[weak self] status in
    switch status {
    case .beingHeaderRefresh:
        self?.tableView.mj_header.beginRefreshing()
    case .endHeaderRefresh:
        self?.tableView.mj_header.endRefreshing()
    case .beingFooterRefresh:
        self?.tableView.mj_footer.beginRefreshing()
    case .endFooterRefresh:
        self?.tableView.mj_footer.endRefreshing()
    case .noMoreData:
        self?.tableView.mj_footer.endRefreshingWithNoMoreData()
    default:
        break
    }
}).addDisposableTo(rx_disposeBag)
复制代码

若是在一个项目多处使用到了这种方式，咱们就能够看到弊端——重复代码，过于冗余。

难道咱们每次都要在controller中进行如此操做吗？

面向协议

关于协议的内容能够看下我以前的这两篇文章 iOS - Swift 面向协议编程（一） iOS - Swift 面向协议编程（二）

总结协议的两大做用：一、规范 二、定制能力

定义协议 Refreshable

/* ============================ Refreshable ================================ */
// 须要使用 MJExtension 的控制器使用

protocol Refreshable {
    
}
extension Refreshable where Self : UIViewController {
    func initRefreshHeader(_ scrollView: UIScrollView, _ action: @escaping () -> Void) -> MJRefreshHeader {
        scrollView.mj_header = MJRefreshNormalHeader(refreshingBlock: { action() })
        return scrollView.mj_header
    }
    
    func initRefreshFooter(_ scrollView: UIScrollView, _ action: @escaping () -> Void) -> MJRefreshFooter {
        scrollView.mj_footer = MJRefreshAutoNormalFooter(refreshingBlock: { action() })
        return scrollView.mj_footer
    }
}
复制代码

在controller中遵循 Refreshable 协议，经过initRefreshHeader方法或者initRefreshFooter方法给tableView或者collectionView赋予头部或尾部刷新的能力，而且书写下拉刷新时须要执行的代码

// 如下拉刷新为例
let refreshHeader = initRefreshHeader(liveCollectionView) { [weak self] in
    // 下拉后须要执行的操做 
    self?.vmOutput?.requestCommand.onNext(())
}
复制代码

接下来再讲讲output，只要有网络请求的地方，就会须要须要监听请求状态，既然这样，那么能够为output定义一个协议OutputRefreshProtocol，专门用来规范必需声明的属性

/* ============================ OutputRefreshProtocol ================================ */
// viewModel 中 output使用

protocol OutputRefreshProtocol {
    // 告诉外界的tableView当前的刷新状态
    var refreshStatus : Variable<LXFRefreshStatus> {get}
}
复制代码

接着让output去遵循该协议，并进行初始化刷新状态的值为.none

struct LXFLiveOutput: OutputRefreshProtocol {
    var refreshStatus: Variable<LXFRefreshStatus>
    
    let sections: Driver<[LXFLiveSection]>
    init(sections: Driver<[LXFLiveSection]>) {
        self.sections = sections
        refreshStatus = Variable<LXFRefreshStatus>(.none)
    }
}
复制代码

到此为止，其实跟以前没啥两样，只是使controller更方便初始化刷新控件而已。接下来才是本文的重点。

重点

刷新的状态无非也就那么几种，下拉重载数据，上拉加载更多，请求完成时结束下拉或上拉等等。。。那咱们何须要在每一个controller中再去管理这等杂事？？ 而至此，刷新控件的状态是由变量 refreshStatus 来决定，此时 refreshStatus 又声明在 OutputRefreshProtocol 协议中，咱们何再也不定义一个方法，将刷新控件的状态交给refreshStatus本身来帮咱们处理呢～

extension OutputRefreshProtocol {
    func autoSetRefreshHeaderStatus(header: MJRefreshHeader?, footer: MJRefreshFooter?) -> Disposable {
        return refreshStatus.asObservable().subscribe(onNext: { (status) in
            switch status {
            case .beingHeaderRefresh:
                header?.beginRefreshing()
            case .endHeaderRefresh:
                header?.endRefreshing()
            case .beingFooterRefresh:
                footer?.beginRefreshing()
            case .endFooterRefresh:
                footer?.endRefreshing()
            case .noMoreData:
                footer?.endRefreshingWithNoMoreData()
            default:
                break
            }
        })
    }
}
复制代码

这时须要咱们将刷新控件的对象 header / footer 传入到方法中，实现自动控制刷新控件状态。

总结使用

1、output中遵照协议 OutputRefreshProtocol， 并初始化 refreshStatus 的值为 none

struct LXFLiveOutput: OutputRefreshProtocol {
    var refreshStatus: Variable<LXFRefreshStatus>
    
    let sections: Driver<[LXFLiveSection]>
    init(sections: Driver<[LXFLiveSection]>) {
        self.sections = sections
        refreshStatus = Variable<LXFRefreshStatus>(.none)
    }
}
复制代码

2、controller 遵照协议 Refreshable，经过协议中的方法初始化刷新控件及对应的操做，并将刷新控件对象做为参数传入到自动处理状态方法中

extension LXFLiveViewController: Refreshable 复制代码

let refreshHeader = initRefreshHeader(liveCollectionView) { [weak self] in
    self?.vmOutput?.requestCommand.onNext(())
}
vmOutput?.autoSetRefreshHeaderStatus(header: refreshHeader, footer: nil).disposed(by: rx.disposeBag)
复制代码

3、viewModel中根据实际状况实时更新 refreshStatus 的刷新状态

开源库

LXFProtocolTool

推荐使用 LXFProtocolTool/Refreshable ，会不断更新，功能更增强大

pod 'LXFProtocolTool/Refreshable'
复制代码

案例

协议：Refreshable.swift

ViewModel：LXFLiveViewModel

Controller：LXFLiveViewController

LXFBiliBili