文章目录

• 一、前言
• 二、代码定义
• 三、参考源码位置
• 四、参考链接

一、前言

官方提供了供Activity和Fragment使用的ViewModel，但是并没有提供全局的ViewModel。所以如果我们在进行Activity跳转的时候，并不能保持数据使用同一份。所以这里需要对系统的ViewModel进行拓展。这里结合网上的一些资料和部分源码。形成了如下方便使用的代码。

二、代码定义

该代码定义在任何一个kt文件的顶层文件中

//一个全局的ViewModel
@MainThread
public inline fun <reified VM : ViewModel> ComponentActivity.applicationViewModels(
    noinline factoryProducer: (() -> ViewModelProvider.Factory)? = null
): Lazy<VM> {
    val factoryPromise = factoryProducer ?: {
        defaultViewModelProviderFactory
    }
    return ViewModelLazy(VM::class, { application.viewModelStore }, factoryPromise)
}

val Application.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

三、参考源码位置

1. ComponentActivity.viewModels：该文件位于Activity-ktx拓展库中

2. FragmentManager: 第2559行

3. FragmentManagerViewModel:全部

4. SavedStateViewModelFactory:第96行

5. 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'

四、参考链接

1. 【Kotlin】就几行代码?! 用SharedFlow写个FlowEventBus_移动端开发干货分享-CSDN博客

2. [Android — ViewModel 工厂和实例化](Android — ViewModel factory and instantiation | Mahendran)