ViewModel的创建
ViewModel的创建
ViewModel本身只是ViewModel
这个类的子类:
class MainViewModel: ViewModel() {
}
在屏幕旋转UI重建的时候, 它是如何拥有保持数据的能力的呢? 它又是何时被清理的呢?
答案全跟它是如何创建, 保存的有关系.
本文回顾一下创建ViewModel的几种常见写法.
注: 本文中的图并不是严格意义的时序图(也不符合规范), 只是为了简略表示一下代码中的调用关系.
原生手动创建ViewModel
当ViewModel没有构造参数
当ViewModel没有参数的时候很简单:
class MainViewModel: ViewModel() {}
class MainActivity : ComponentActivity() {
private lateinit var viewModel: MainViewModel
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
viewModel = ViewModelProvider(this).get(MainViewModel::class.java)
}
}
当然不是随便就new
出来的啦, 得通过ViewModelProvider
来get.
注意: 这句不能在Activity的onCreate()
之前调用, 也意味着你不能声明字段直接赋值.
否则你就会得到这个报错:
Caused by: java.lang.IllegalStateException: Your activity is not yet attached to the Application instance. You can't request ViewModel before onCreate call.
此时的方法调用大概是这样:
从这里我们看到ViewModelProvider
这个工具人要借助其他两个东西来提供ViewModel:
ViewModelStore
: 负责存储ViewModel.Factory
: 负责实例化具体的ViewModel类型.
请记住这两个知识点, 后面要考.
在上面这个最简单的例子中:
- Activity是
ViewModelStoreOwner
, 它可以getViewModelStore()
当ViewModelStoreOwner
(比如Activity)因为configuration changes重建, 新的owner仍然会get这个旧的ViewModelStore
实例. - 我们没有传
Factory
, 所以最终用的是没有参数的NewInstanceFactory
.
当ViewModel有构造参数
但是通常, ViewModel会需要一些依赖, 我们就需要从构造传入一些参数
class MainViewModel(
private val repository: MainRepository,
) : ViewModel()
此时我们的工厂就需要自己实现了:
我们需要把依赖对象传给工厂, 好让它构造ViewModel的时候能用上:
class MyViewModelFactory constructor(private val repository: MainRepository) :
ViewModelProvider.Factory {
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
if (modelClass.isAssignableFrom(MainViewModel::class.java)) {
return MainViewModel(repository) as T
}
throw IllegalArgumentException("Unknown ViewModel class")
}
}
接着把我们的工厂传给ViewModelProvider
:
class MainActivity : ComponentActivity() {
private lateinit var viewModel: MainViewModel
private val repository: MainRepository = MainRepository()
private val viewModelFactory: MyViewModelFactory = MyViewModelFactory(repository)
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
viewModel = ViewModelProvider(this, viewModelFactory).get(MainViewModel::class.java)
}
}
此时的过程大概是这样:
AndroidX来帮忙
感谢AndroidXactivity-ktx
(fragment-ktx
里也有Fragment版本的)包里的by viewModels()
属性代理,
我们上述的代码可以简化成这样:
class MyViewModelFactory constructor(private val repository: MainRepository) :
ViewModelProvider.Factory {
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
if (modelClass.isAssignableFrom(MainViewModel::class.java)) {
return MainViewModel(repository) as T
}
throw IllegalArgumentException("Unknown ViewModel class")
}
}
class MainActivity : ComponentActivity() {
private val viewModel: MainViewModel by viewModels {
viewModelFactory
}
private val repository: MainRepository = MainRepository()
private val viewModelFactory: MyViewModelFactory = MyViewModelFactory(repository)
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
// use viewModel
}
}
工厂依然自己写, 简化了provider get的部分:
- 扩展方法, 隐含activity对象.
- lazy规避了生命周期的问题, 只要使用ViewModel的地方不在
onCreate
之前就行.
ViewModel没有参数的时候更简单:
private val viewModel: MainViewModel by viewModels()
在Fragment里还有by activityViewModels()
, 可以获取scope和Activity绑定的ViewModel.
Dagger时代
有了dagger, 我们可以在构造上标记@Inject
告诉dagger帮我们创建repository和viewModelFactory:
class MainRepository @Inject constructor(){}
@Singleton
class MyViewModelFactory @Inject constructor(private val repository: MainRepository) :
ViewModelProvider.Factory {
override fun <T : ViewModel?> create(modelClass: Class<T>): T {
if (modelClass.isAssignableFrom(MainViewModel::class.java)) {
return MainViewModel(repository) as T
}
throw IllegalArgumentException("Unknown ViewModel class")
}
}
但是我们却不能对ViewModel这样做, 因为ViewModel应该被存储在ViewModelStore里, 而它是由activity提供的.
我们会注入viewModelFactory, 然后用它来创建ViewModel:
class MainActivity : ComponentActivity() {
private val viewModel: MainViewModel by viewModels {
viewModelFactory
}
@Inject
lateinit var viewModelFactory: MyViewModelFactory
}
因为你用的是dagger, 你会需要在onCreate()
里写类似这样的东西:
(applicationContext as MyApplication).appComponent.inject(this)
这里只是用DI框架简化了依赖和工厂的构造.
Hilt时代
在ViewModel上标记: @HiltViewModel
, 构造标记:@Inject
:
@HiltViewModel
class MainViewModel @Inject constructor(
private val repository: MainRepository,
) : ViewModel() {
}
依赖们也可以构造标记@Inject
:
class MainRepository @Inject constructor(){}
在Activity加上注解@AndroidEntryPoint
,
也用了by viewModels()
:
@AndroidEntryPoint
class MainActivity : ComponentActivity() {
private val viewModel: MainViewModel by viewModels()
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
// use viewModel
}
}
请注意, 和dagger不同的是, 这里并不需要注入工厂(也不需要写工厂), Hilt用的是自己的工厂HiltViewModelFactory
.
这个过程:
Compose的ViewModel()方法
在Compose中获取一个ViewModel非常简单, 可以只用一个viewModel()
方法.
(这个方法在androidx.lifecycle:lifecycle-viewmodel-compose
依赖包里).
@Composable
fun Greeting(name: String) {
val viewModel: MainViewModel = viewModel()
// use viewModel
}
Activity的onCreate()中setContent是Compose, 于是在这里就设置好各种owners,
与ViewModel相关的就是这个LocalViewModelStoreOwner
, 之后包在里面的内容就可以随时获取owner, 得到ViewModelStore和ViewModel了.
Compose的hiltViewModel()方法
上面的viewModel()
方法不管在哪里获取, ViewModel的scope都是和当前的Activity或Fragment绑定.
假如我们有多个composable的界面呢?
使用hiltViewModel()
这个composable方法我们可以获取到一个scope到某个导航目的地的ViewModel.
(这个方法在androidx.hilt:hilt-navigation-compose
依赖包里).
NavHost(navController = navController, startDestination = "friendslist") {
composable("friendslist") {
val viewModel = hiltViewModel<MainViewModel>()
FriendsList(viewModel = viewModel, navHostController = navController)
}
...
}
这种方式获取的ViewModel, 它的生命周期是和这个导航目的地相关的, 当退出这个界面, ViewModel就被clear, 下次再进就又是新的对象.
此时的owner终于不再是Activity或Fragment, 而是NavBackStackEntry
.
导航这个话题就先不展开这里讲了.
总结
ViewModel的创建很关键, 关系到它的生命周期.
手动创建比较麻烦, 很多样板代码.
本文总结了几种常见的创建方式, 希望读者看完后能有更清晰的理解, 每种方式都是怎么回事, 那些方便的工具替我们做了什么.