GitHub Demo链接
前言：最近换了新公司，开始了RxSwift+MVVM的爬坑之旅，相比之前臃肿的MVC，使用MVVM架构之后，愈发感慨这些架构设计者的强大😂，只要你想，几乎可以把所有的业务逻辑从Controller抽离到ViewModel里，让你的Controller无比清爽！当然这离不开RxSwift的响应式编程方式，它让MVVM的实现变得简单，我们今天来讲一下ViewModel的不同实现方式。
PS：本篇文章参考了[译]RxSwift + MVVM: 怎样搞定 ViewModels，主要思路皆来源于文中的内容，原文没有Demo，我做了一个Demo做补充，想要详细了解的可以去读一下这篇。
关于ViewModel
ViewModel的使用，更像是一个黑盒，我们在Controller里赋予它输入inputs，ViewModel会将逻辑内部处理，然后输出outputs到view上。
-
第一种ViewModel：外部初始化Input
首先我们来看一下它的结构
protocol TransformViewModelType {
associatedtype Input
associatedtype Output
func transform(input: Input) -> Output
}
需要外部传入初始化好的input，然后调用ViewModel 的transform方法输出output
我们用实际需求来实现一个这种ViewModel，用户可以在最上部的TextField输入内容，然后点击Send的Button，数据会在下面的Label显示出来
Simulator Screen Shot - iPhone Xʀ - 2019-04-11 at 13.52.36.png
这里抽离一下要实现的业务，我们需要的输入有两个，一个是textfield的文字，一个是send按钮的点击事件，需要的输出就是文字，具体实现如下：
final class TransformViewModel: TransformViewModelType {
struct Input {
let message: Observable<String>
let send: Observable<Void>
}
struct Output {
let result: Driver<String>
}
func transform(input: TransformViewModel.Input) -> TransformViewModel.Output {
let result = input.send
.withLatestFrom(input.message)
.map { (message) -> String in
return "viewModel1: \(message)"
}
.startWith("result1")
.asDriver(onErrorJustReturn: "")
return Output.init(result: result)
}
}
接下来是output与View的绑定：
func bindModel1() {
let input = TransformViewModel.Input.init(message: textField.rx.text.orEmpty.asObservable(),
send: sendButton.rx.tap.asObservable())
let output = viewModel1.transform(input: input)
output.result
.drive(label_1.rx.text)
.disposed(by: rx.disposeBag)
}
PS：rx.disposeBag来源于NSObject-Rx框架的扩展，是一个非常实用的RxSwift的语法糖。
-
第二种ViewModel：内部初始化Input
这种类型我们使用私有的Subject在内部做初始化，用户在创建ViewModel的时候，已经完成了相关逻辑处理的绑定，留下了需要的数据接口给外部，可以随时接入input或者渲染output给View，更加灵活方便一些，我们来看结构：
protocol SubjectViewModelType {
associatedtype Input
associatedtype Output
var input: Input {get}
var output: Output {get}
}
不再需要transform方法转化输出，而是两个get属性，还是上面👆的需求，我们来看一下这种方式的具体实现：
final class SubjectViewModel: SubjectViewModelType {
var input: SubjectViewModel.Input
var output: SubjectViewModel.Output
struct Input {
let message: AnyObserver<String>
let send: AnyObserver<Void>
}
struct Output {
let result: Driver<String>
}
private let messageSubject = PublishSubject<String>()
private let sendSubject = PublishSubject<Void>()
init() {
let result = sendSubject
.withLatestFrom(messageSubject)
.map { (message) -> String in
return "viewModel2: \(message)"
}
.startWith("result2")
.asDriver(onErrorJustReturn: "")
input = Input.init(message: messageSubject.asObserver(), send: sendSubject.asObserver())
output = Output.init(result: result)
}
}
我们使用ViewModel内部私有的Subject，在整个ViewModel的init时候，就完成了逻辑处理的绑定，所以不再需要另外的类似transform的逻辑转化，下面👇是这种方式与View的数据绑定：
func bindModel2() {
textField.rx.text.orEmpty
.subscribe(viewModel2.input.message)
.disposed(by: rx.disposeBag)
sendButton.rx.tap
.subscribe(viewModel2.input.send)
.disposed(by: rx.disposeBag)
viewModel2.output.result
.drive(label_2.rx.text)
.disposed(by: rx.disposeBag)
}
省去了在外部创建input和获取output的麻烦，看起来是不是更优雅一些呢😊
对比分析
在最开始提到的参考文章里，作者说第二种类型的ViewModel适用的范围更广一些，但是在我的实际使用中，发现我们要想在View Model中做一些数据的中继，处理更多的逻辑，需要用到Input之外的变量的话，第二种方法明显有一些无法实现的情况，因为它是在Init方法里做内部逻辑的绑定，所以self本身并未初始化，无法读取自身其他未初始化的数据；而第一种方式则没有关系，因为它的逻辑绑定发生在transform方法里，这时的ViewModel已经初始化完成了。
当然这只是少部分的情况，不可否认，第二种方式的ViewModel使用起来更加的灵活优雅，大家看具体的情况来选择合适的类型即可。
欢迎更正错误和交流，回复评论和私信皆可 🎉🎉🎉
