近半年有机会在项目中使用了Compose,整体使用下来感觉比较新颖。由于我们项目设计的是整个项目只有一个Activity,不同的界面使用Androidx的navigation切换布局即可,这就凸显了一个问题,切换了界面,但是不同的composable的viewmodel该怎么管理?Android原生的viewmodel都是以activity和fragment控制生命周期,在activity和fragment销毁时进行viewmodel的scope和任务的清理。这显然不符合我们项目的设计,所以自己就摸索了个能创建和composable组件生命周期相同的viewmodel的代码,供大家参考:
@Composable
fun <VM : ViewModel> rememberScopedViewModel(modelClass: Class<VM>, factory: ViewModelProvider.Factory? = null): VM {
val viewModelStore = remember { ViewModelStore() }
val viewModelStoreOwner = remember {
object : ViewModelStoreOwner {
override val viewModelStore: ViewModelStore
get() = viewModelStore
}
}
DisposableEffect(Unit) {
onDispose {
viewModelStore.clear()
}
}
return remember {
val finalFactory = factory ?: object : ViewModelProvider.Factory {
override fun <T : ViewModel> create(modelClass: Class<T>): T {
return modelClass.getDeclaredConstructor().newInstance() as T
}
}
ViewModelProvider(
viewModelStoreOwner,
finalFactory,
)[modelClass]
}
}