如何在Dagger2 android中为ViewModel使用构造函数注入?

时间:2020-08-17 02:45:00

标签: android-studio mvvm dependency-injection dagger android-viewmodel

我想在Dagger2中使用ViewModel,但是我不知道如何在ViewModelProviders或ViewModelProvider.Factory中使用构造函数注入

如何在构造函数注入中使用ViewModel?我应该使用ViewModelProvider.Factory还是ViewModelProviders?

更新: 您可以在this link

中找到详细的解决方案

1 个答案:

答案 0 :(得分:0)

您可以创建自己的自定义类,扩展ViewModelProvider.Factory

class ViewModelFactory @Inject constructor(
private val providers: @JvmSuppressWildcards Map<Class<out ViewModel>, Provider<ViewModel>>
) : ViewModelProvider.Factory {

override fun <T : ViewModel> create(modelClass: Class<T>): T {
    var provider = providers[modelClass]
    if (provider == null) {
        for ((key, value) in providers) {
            if (modelClass.isAssignableFrom(key)) {
                provider = value
                break
            }
        }
    }
    if (provider == null) {
        throw IllegalStateException("unknown model class $modelClass")
    }

    return provider.get() as T
  }
}

为此类创建模块,并将该模块插入您的ApplicationComponent类中

@Module
abstract class ViewModelFactoryModule {
   @Binds
   abstract fun bindViewModelFactory(viewModelFactory: ViewModelFactory): ViewModelProvider.Factory
}

然后在片段内部注入该工厂

@Inject
lateinit var factory: ViewModelProvider.Factory

并像这样初始化您的viewModel

val vm: YourViewModel by viewModels { factory }