如何解决使用ViewModelProvider的自定义View布局预览失效问题？

问题根源 #

布局预览模式下，自定义View无法获取有效的ViewModelStoreOwner，代码里用!!强制非空触发了NullPointerException。

修复方案 #

• 方案1：给ViewModel获取逻辑添加空值判断
  修改lazy初始化代码，判断ViewModelStoreOwner是否存在，不存在时返回ViewModel的空实现或测试实例：

private val mainViewModel by lazy {
    val owner = findViewTreeViewModelStoreOwner()
    if (owner != null) {
        ViewModelProvider(owner).get<MainViewModel>()
    } else {
        // 预览时返回占位实例，需确保MainViewModel有无参构造或适配的构造方法
        MainViewModel()
    }
}

• 方案2：用isInEditMode()跳过预览时的ViewModel初始化
  利用View自带的isInEditMode()判断是否处于预览模式，预览时直接返回占位实例：

private val mainViewModel by lazy {
    if (isInEditMode) {
        MainViewModel() // 预览专用占位实例
    } else {
        ViewModelProvider(findViewTreeViewModelStoreOwner()!!).get<MainViewModel>()
    }
}

• 方案3：解耦自定义View与ViewModel（推荐）
  让自定义View只负责UI渲染，ViewModel的获取和业务逻辑交给上层的Activity/Fragment，通过方法传递数据或回调：

1. 在自定义View中定义数据设置方法：

fun setBarInfo(title: String, onBackClicked: () -> Unit) {
    // 这里写更新UI的逻辑，比如设置标题、绑定返回点击事件
}

2. 在Activity/Fragment中获取ViewModel后，调用该方法传递数据：

// Activity中示例
val mainViewModel = ViewModelProvider(this).get<MainViewModel>()
topBar.setBarInfo(
    title = mainViewModel.title.value.orEmpty(),
    onBackClicked = { mainViewModel.handleBack() }
)

内容的提问来源于stack exchange，提问作者Greelings