<think>
我们需要
自定义一个ViewModelFactory来创建带有SavedStateHandle的
ViewModel。这通常在需要保存和恢复状态时使用，例如在配置更改后恢复数据。
通常，我们会继承`AbstractSavedState
ViewModelFactory`或者实现`
ViewModelProvider.Factory`接口，并处理SavedStateHandle的创建。
下面是
一个示例，展示如何
自定义一个ViewModelFactory来创建带SavedStateHandle的
ViewModel。
步骤：
1. 创建
一个自定义的工厂类，继承自`AbstractSavedState
ViewModelFactory`（推荐，因为它简化了SavedStateHandle的管理）或者实现`
ViewModelProvider.Factory`。
2. 在工厂中，我们需要重
写`create`方法，在这个方法中利用SavedStateHandle来创建
ViewModel实例。
这里我们展示两种方式：
方式一：继承`AbstractSavedState
ViewModelFactory`（更简单，推荐）
```kotlin
import androidx.lifecycle.AbstractSavedState
ViewModelFactory
import androidx.lifecycle.SavedStateHandle
import androidx.lifecycle.
ViewModel
import androidx.savedstate.SavedStateRegistryOwner
// 假设我们有
一个需要SavedStateHandle的
ViewModel
class My
ViewModel(savedStateHandle: SavedStateHandle, dependency: MyDependency) :
ViewModel() {
//
ViewModel的逻辑
}
class My
ViewModelFactory(
owner: SavedStateRegistryOwner, // 通常是Activity或Fragment
private val dependency: MyDependency // 可能的依赖
) : AbstractSavedState
ViewModelFactory(owner, null) { // 第二个参数是默认的Bundle，可以传入null
@Suppress("UNCHECKED_CAST")
override fun <T :
ViewModel> create(
key: String,
modelClass: Class<T>,
handle: SavedStateHandle
): T {
// 根据modelClass来创建不同的
ViewModel
return if (modelClass.isAssignableFrom(My
ViewModel::class.java)) {
My
ViewModel(handle, dependency) as T
} else {
throw IllegalArgumentException("Unknown
ViewModel class")
}
}
}
```
方式二：实现`
ViewModelProvider.Factory`接口（需要自己管理SavedStateHandle）
```kotlin
import androidx.lifecycle.SavedStateHandle
import androidx.lifecycle.
ViewModel
import androidx.lifecycle.
ViewModelProvider
import androidx.lifecycle.
viewmodel.CreationExtras
class My
ViewModelFactory(
private val dependency: MyDependency
) :
ViewModelProvider.Factory {
@Suppress("UNCHECKED_CAST")
override fun <T :
ViewModel> create(modelClass: Class<T>, extras: CreationExtras): T {
// 从extras中获取SavedStateHandle
val savedStateHandle = extras.get(SavedStateHandle.CREATOR_KEY)?.createSavedStateHandle()
?: throw IllegalStateException("SavedStateHandle not available")
if (modelClass.isAssignableFrom(My
ViewModel::class.java)) {
return My
ViewModel(savedStateHandle, dependency) as T
}
throw IllegalArgumentException("Unknown
ViewModel class")
}
}
```
在Activity或Fragment中使用
自定义工厂：
```kotlin
class MyActivity : AppCompatActivity() {
private lateinit var myDependency: MyDependency
private lateinit var
viewModel: My
ViewModel
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
myDependency = ... // 初始化依赖
// 创建工厂
val factory = My
ViewModelFactory(this, myDependency)
// 获取
ViewModel
viewModel =
ViewModelProvider(this, factory).get(My
ViewModel::class.java)
// 或者使用委托方式（需要添加依赖：androidx.fragment:fragment-ktx）
// val
viewModel: My
ViewModel by
viewModels { factory }
}
}
```
注意：在Fragment中，可以使用`activity
ViewModels`来共享Activity作用域的
ViewModel。
这种方式可以避免为每个
ViewModel手动创建工厂，特别是当有多个
ViewModel需要创建时，可以在工厂中根据传入的class进行判断创建相应的
ViewModel。
§§相关问题§§
1. 如何在
自定义ViewModel工厂中处理多个不同的
ViewModel类型？
2. 使用SavedStateHandle时，如何确保在进程死亡后恢复数据？
3. 在
自定义工厂中如何传递多个依赖项？
4. 继承AbstractSavedState
ViewModelFactory和直接实现
ViewModelProvider.Factory接口有何主要区别？
5. 在Fragment中使用
自定义工厂创建
ViewModel时需要注意哪些作用域问题？
</think>
###
自定义 ViewModelFactory 创建带 SavedStateHandle 的
ViewModel
#### 核心实现步骤
```kotlin
// 1. 定义带SavedStateHandle的
ViewModel
class Profile
ViewModel(
private val savedStateHandle: SavedStateHandle, // 状态句柄
private val userRepo: UserRepository // 其他依赖
) :
ViewModel() {
// 状态属性
val userName = savedStateHandle.getLiveData<String>("username")
// 方法示例
fun updateName(name: String) {
savedStateHandle["username"] = name
}
}
// 2. 创建
自定义Factory
class Profile
ViewModelFactory(
private val owner: SavedStateRegistryOwner, // Activity/Fragment
private val defaultArgs: Bundle? = null, // 初始参数
private val userRepo: UserRepository // 依赖注入
) : AbstractSavedState
ViewModelFactory(owner, defaultArgs) {
@Suppress("UNCHECKED_CAST")
override fun <T :
ViewModel> create(
key: String,
modelClass: Class<T>,
handle: SavedStateHandle
): T {
// 验证
ViewModel类型
require(modelClass == Profile
ViewModel::class.java) {
"Unknown
ViewModel class: ${modelClass.name}"
}
// 创建
ViewModel实例
return Profile
ViewModel(handle, userRepo) as T
}
}
```
#### 在 Activity/Fragment 中使用
```kotlin
// Activity中使用
class ProfileActivity : AppCompatActivity() {
private lateinit var
viewModel: Profile
ViewModel
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
// 获取依赖
val userRepo = (application as MyApp).userRepository
// 创建工厂
val factory = Profile
ViewModelFactory(
owner = this,
defaultArgs = intent.extras, // 传递Intent参数
userRepo = userRepo
)
// 获取
ViewModel
viewModel =
ViewModelProvider(this, factory)[Profile
ViewModel::class.java]
// 使用委托简化(KTX扩展)
// private val
viewModel: Profile
ViewModel by
viewModels { factory }
}
}
```
#### 高级技巧：通用工厂模板
```kotlin
// 通用
ViewModel工厂
abstract class Generic
ViewModelFactory<VM :
ViewModel>(
private val owner: SavedStateRegistryOwner,
private val defaultArgs: Bundle? = null
) : AbstractSavedState
ViewModelFactory(owner, defaultArgs) {
@Suppress("UNCHECKED_CAST")
override fun <T :
ViewModel> create(
key: String,
modelClass: Class<T>,
handle: SavedStateHandle
): T {
return create
ViewModel(handle) as T
}
// 抽象方法由子类实现
abstract fun create
ViewModel(handle: SavedStateHandle): VM
}
// 具体实现
class ProfileFactory(
owner: SavedStateRegistryOwner,
private val repo: UserRepository
) : Generic
ViewModelFactory<Profile
ViewModel>(owner) {
override fun create
ViewModel(handle: SavedStateHandle): Profile
ViewModel {
return Profile
ViewModel(handle, repo)
}
}
```
#### 状态管理最佳实践
1. **状态读
写方法**
```kotlin
// 最佳状态访问方式
val isLoading: StateFlow<Boolean> = savedStateHandle.getStateFlow("loading", false)
fun setLoading(value: Boolean) {
savedStateHandle["loading"] = value
}
```
2. **参数验证**
```kotlin
// 在Factory中添加参数验证
init {
requireNotNull(userRepo) { "UserRepository dependency missing" }
}
```
3. **测试方案**
```kotlin
@Test
fun test
ViewModelCreation() {
// 准备测试环境
val testHandle = SavedStateHandle()
val mockRepo = mock<UserRepository>()
// 创建工厂
val factory = Profile
ViewModelFactory(
owner = mockOwner(),
userRepo = mockRepo
)
// 验证创建
val vm = factory.create("key", Profile
ViewModel::class.java, testHandle)
assertNotNull(vm)
verify(mockRepo).getUserData()
}
```
本文探讨了如何将自定义的分页数据Model与Entity Framework生成的Model整合到同一个ViewModel中，并实现数据的有效传递。
扫一扫
4054
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
