一、前言
官方提供了供Activity和Fragment使用的ViewModel,但是并没有提供全局的ViewModel。所以如果我们在进行Activity跳转的时候,并不能保持数据使用同一份。所以这里需要对系统的ViewModel进行拓展。这里结合网上的一些资料和部分源码。形成了如下方便使用的代码。
二、代码定义
该代码定义在任何一个kt文件的顶层文件中
@MainThread
public inline fun <reified VM : ViewModel> ComponentActivity.applicationViewModels(
noinline factoryProducer: (() -> ViewModelProvider.Factory)? = null
): Lazy<VM> {
val factoryPromise = factoryProducer ?: {
defaultViewModelProviderFactory
}
return ViewModelLazy(VM::class, {viewModelStore }, factoryPromise)
}
val viewModelStore: ViewModelStore by lazy {
ViewModelStore()
}
使用方式如下:
private val viewModel1: TestAndroidViewModel by applicationViewModels()
private val viewModel2: TestAndroidViewModel by applicationViewModels {
ViewModelProvider.AndroidViewModelFactory.getInstance(application)
}
这份代码使用避免了对ViewModel和AndroidViewModel的类型限制,可以做到灵活使用,也没有对Factory限制,可以做到自定义任何一种Factory。不过处于对ViewModelStore存储规则的考虑,尽量不要存储同一种类型,参数不一样的ViewModel
三、参考源码位置
-
ComponentActivity.viewModels:该文件位于Activity-ktx拓展库中 -
FragmentManager: 第2559行 -
FragmentManagerViewModel:全部 -
SavedStateViewModelFactory:第96行 -
ViewModelStoreOwner: 全部
以上源码参考以下依赖版本:
implementation 'androidx.core:core-ktx:1.7.0'
implementation 'androidx.appcompat:appcompat:1.4.0'
implementation 'androidx.activity:activity-ktx:1.4.0'
implementation 'androidx.fragment:fragment-ktx:1.4.0'
四、参考链接
-
【Kotlin】就几行代码?! 用SharedFlow写个FlowEventBus_移动端开发干货分享-CSDN博客 -
[Android — ViewModel 工厂和实例化](Android — ViewModel factory and instantiation | Mahendran)
|