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

最新推荐文章于 2025-10-23 11:21:50 发布

原创最新推荐文章于 2025-10-23 11:21:50 发布 · 964 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#Android

写在前面

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

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

为什么要有ViewModel?

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

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

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

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

ViewModel架构组件是为VM层服务的。
容易联想到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加入相应的依赖

//ViewModel
implementation "android.arch.lifecycle:extensions:1.1.1"
implementation "android.arch.lifecycle:viewmodel:1.1.1"
kapt "android.arch.lifecycle:compiler:1.1.1"

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

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

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

class PaoActivity : RxAppCompatActivity() {
lateinit var mBinding : PaoActivityBinding
lateinit var mViewModel : PaoViewModel
@Inject
lateinit var factory: ViewModelProvider.Factory
override fun onCreate(savedInstanceState: Bundle?) {
//....
mViewModel=ViewModelProviders.of(this,factory).get(PaoViewModel::class.java)
//...
}
}

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

ViewModelProvider.Factory

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

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

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

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

@Singleton
class PaoViewModelFactory @Inject constructor(private val viewModel:PaoViewModel): ViewModelProvider.Factory{
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
if (modelClass.isInstance(viewModel)){
return viewModel as T
}else{
throw IllegalArgumentException("unknown model class $modelClass")
}
}
}

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

@Singleton
class PaoViewModelFactory @Inject constructor(private val viewModel:PaoViewModel,private val viewModelOne:OneViewModel,private val viewModelOther:OtherViewModel): ViewModelProvider.Factory{
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
return when{
modelClass.isInstance(viewModel) -> viewModel as T
modelClass.isInstance(viewModelOne) -> viewModelOne as T
modelClass.isInstance(viewModelOther) -> viewModelOther as T
//...
else -> throw IllegalArgumentException("unknown model class $modelClass")
}
}
}

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

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

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

@Module
abstract class ViewModelModule{
@Binds
@IntoMap
@ViewModelKey(PaoViewModel::class)//自定义的mapKey
abstract fun bindPaoViewModel(viewModel: PaoViewModel):ViewModel
// @Binds
// @IntoMap
// @ViewModelKey(OtherViewModel::class)
// abstract fun bindOtherViewModel(viewModel: OtherViewModel):ViewModel
// ...
@Binds
abstract fun bindViewModelFactory(factory:PaoViewModelFactory):ViewModelProvider.Factory
}

然后它放到AppComponent中

@Singleton
@Component(modules = arrayOf(
AndroidInjectionModule::class,
AppModule::class,
ViewModelModule::class,
ActivityModule::class)
)
interface AppComponent {}

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

@Singleton
class PaoViewModelFactory @Inject constructor(private val creators:Map<Class<out ViewModel>,Provider<ViewModel>>): ViewModelProvider.Factory{
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
val creator = creators[modelClass]?:creators.entries.firstOrNull{
modelClass.isAssignableFrom(it.key)
}?.value?:throw IllegalArgumentException("unknown model class $modelClass")
try {
@Suppress("UNCHECKED_CAST")
return creator.get() as T
} catch (e: Exception) {
throw RuntimeException(e)
}
}
}