使用Kotlin构建MVVM应用程序—提高篇：ViewModel

写在前面

大家好，这里是使用Kotlin构建MVVM应用程序—提高篇：ViewModel。

本篇文章将介绍google推荐的架构组件ViewModel的使用方法及实现原理。

为什么要有ViewModel?

为什么？看到ViewModel这个名字相信都会联系到MVVM架构中的VM。

但是在我看来，这两者并非是一个意思。如果你想实现MVVM架构的APP，按照

使用Kotlin构建MVVM应用程序基础篇的内容就已经足够了。

而我推测google把它称为ViewModel的原因可能有两点：

1. ViewModel架构组件是为VM层服务的。
2. 容易联想到MVVM架构，代表着google更推荐Android工程师们应用MVVM架构，而并非冗杂繁复的MVP。

当然这些是题外话。既然不使用ViewModel也能构建MVVM应用，那么ViewModel是来做什么的呢？

The ViewModel class is designed to store and manage UI-related data in a lifecycle conscious way. The ViewModel class allows data to survive configuration changes such as screen rotations.

简单说来，ViewModel是用来存储和管理UI相关的数据，将一个Activity或Fragment组件相关的数据逻辑抽象出来，并能适配组件的生命周期，如当屏幕旋转Activity重建后，ViewModel中的数据依然有效。它还可以帮助开发者轻易实现 Fragment 与 Fragment 之间, Activity 与 Fragment 之间的通讯以及共享数据。

注‘Android技术交流群878873098，欢迎大家加入交流，畅谈！本群有免费学习资料视频’并且免费分享源码解析视频

其中最具有吸引力的功能就是屏幕旋转Activity重建后，ViewModel中的数据依然有效。遥想当年，屏幕旋转当真是开发者不得不迈过的槛。而现在很多应用都只要求竖屏或者强制竖屏，不得横屏，所以对于这样的应用，我的建议是可以不用ViewModel组件，按照普通的VM开发就可以了。当然适当了解一下还是可以的。

ViewModel快速开始

首先我们需要在app/build.gradle加入相应的依赖

1. //ViewModel
2. implementation "android.arch.lifecycle:extensions:1.1.1"
3. implementation "android.arch.lifecycle:viewmodel:1.1.1"
4. kapt "android.arch.lifecycle:compiler:1.1.1"

然后，让你的VM层继承ViewModel组件提供的ViewModel类，如果需要用到Application的context，那么就继承AndroidViewModel类。

1. class PaoViewModel @Inject constructor(private val repo: PaoRepo) :ViewModel(){
2. //...
3. }

在View层的PaoActivity之中，以前的mViewModel是通过Dagger2注入的，而现在需要进行一下修改

1. class PaoActivity : RxAppCompatActivity() {
2.  
3. lateinit var mBinding : PaoActivityBinding
4.  
5. lateinit var mViewModel : PaoViewModel
6.  
7. @Inject
8. lateinit var factory: ViewModelProvider.Factory
9.  
10. override fun onCreate(savedInstanceState: Bundle?) {
11. //....
12. mViewModel=ViewModelProviders.of(this,factory).get(PaoViewModel::class.java)
13. //...
14. }
15.  
16. }

这里有两个概念ViewModelProvider.Factory和ViewModelProviders。再继续之前，我们需要先了解如何打造自己的ViewModelProvider.Factory。

ViewModelProvider.Factory

看到这个名字，自然就联想到工厂模式，这里提供我们需要的mViewModel实例。当然ViewModel没有Dagger2那么神奇，不会帮我们自动生成，所以需要我们自己来实现需要的ViewModelProvider.Factory。

1. class PaoViewModelFactory : ViewModelProvider.Factory{
2. //需要实现create方法，返回具体的viewmodel
3. override fun <T : ViewModel?> create(modelClass: Class<T>): T {
4. //return T
5. }
6. }

那怎么创建具体的实例呢？可以在这里一个个new出相应的依赖，但既然已经有Dagger2这么棒的依赖管理工具，当然是使用Dagger2了。

注‘Android技术交流群878873098，欢迎大家加入交流，畅谈！本群有免费学习资料视频’并且免费分享源码解析视频

1. @Singleton
2. class PaoViewModelFactory @Inject constructor(private val viewModel:PaoViewModel): ViewModelProvider.Factory{
3. override fun <T : ViewModel?> create(modelClass: Class<T>): T {
4. if (modelClass.isInstance(viewModel)){
5. return viewModel as T
6. }else{
7. throw IllegalArgumentException("unknown model class $modelClass")
8. }
9. }
10.  
11. }

到此，当然还没有结束。这样的写法只针对单个ViewModel，如果有多个呢？我们自然不希望出现如下的代码

1. @Singleton
2. class PaoViewModelFactory @Inject constructor(private val viewModel:PaoViewModel,private val viewModelOne:OneViewModel,private val viewModelOther:OtherViewModel): ViewModelProvider.Factory{
3. override fun <T : ViewModel?> create(modelClass: Class<T>): T {
4. return when{
5. modelClass.isInstance(viewModel) -> viewModel as T
6. modelClass.isInstance(viewModelOne) -> viewModelOne as T
7. modelClass.isInstance(viewModelOther) -> viewModelOther as T
8. //...
9. else -> throw IllegalArgumentException("unknown model class $modelClass")
10. }
11. }
12.  
13. }

这样费力不太好并且没有效率的事当然不是有(想)效(偷)率(懒)的程序员喜欢做的事。

简单的做法是注入一个map，value是我们需要的ViewModel实例，key值为相应的Class。

幸运的是通过Dagger2的@IntoMap可以为我们自动构造所需的map对象，是不是对Dagger2开始爱不释手了^_^

1. @Module
2. abstract class ViewModelModule{
3.  
4. @Binds
5. @IntoMap
6. @ViewModelKey(PaoViewModel::class)//自定义的mapKey
7. abstract fun bindPaoViewModel(viewModel: PaoViewModel):ViewModel
8.  
9. // @Binds
10. // @IntoMap
11. // @ViewModelKey(OtherViewModel::class)
12. // abstract fun bindOtherViewModel(viewModel: OtherViewModel):ViewModel
13. // ...
14.  
15. @Binds
16. abstract fun bindViewModelFactory(factory:PaoViewModelFactory):ViewModelProvider.Factory
17. }

然后它放到AppComponent中

1. @Singleton
2. @Component(modules = arrayOf(
3. AndroidInjectionModule::class,
4. AppModule::class,
5. ViewModelModule::class,
6. ActivityModule::class)
7. )
8. interface AppComponent {}

最后修改PaoViewModelFactory，代码参考自android-architecture-components/GitHubBrowserSample

1. @Singleton
2. class PaoViewModelFactory @Inject constructor(private val creators:Map<Class<out ViewModel>,Provider<ViewModel>>): ViewModelProvider.Factory{
3. override fun <T : ViewModel?> create(modelClass: Class<T>): T {
4. val creator = creators[modelClass]?:creators.entries.firstOrNull{
5. modelClass.isAssignableFrom(it.key)
6. }?.value?:throw IllegalArgumentException("unknown model class $modelClass")
7. try {
8. @Suppress("UNCHECKED_CAST")
9. return creator.get() as T
10. } catch (e: Exception) {
11. throw RuntimeException(e)
12. }
13. }
14.  
15. }

大功告成，现在我们的ViewModel已经具备了屏幕旋转Activity重建后，ViewModel中的数据依然有效的能力。

写在最后

和Dagger-Android一样，这里只是简单介绍了ViewModel的使用方法，详细的原理分析放在了小专栏里，希望深入了解ViewModel的欢迎订阅一哈> --.>

注‘Android技术交流群878873098，欢迎大家加入交流，畅谈！本群有免费学习资料视频’并且免费分享源码解析视频