如何让Jetpack Compose的Abstract Compose View随状态变化触发重组？

问题 #

我正在开发一款需要从自定义ViewModel获取数据的输入法键盘。当前遇到的问题是，当任何类型的状态发生变化时，键盘对应的Abstract Compose View无法触发重组。请问是否有办法为Abstract Compose View实现状态变化时的重组/更新机制？

我的InputMethodService代码： #

class IMEService:
    InputMethodService(),
    LifecycleOwner,
    ViewModelStoreOwner,
    SavedStateRegistryOwner {
    private val _lifecycleRegistry: LifecycleRegistry by lazy { LifecycleRegistry(this) }
    private val _store by lazy { ViewModelStore() }
    override fun getLifecycle(): Lifecycle = _lifecycleRegistry
    override fun getViewModelStore(): ViewModelStore = _store
    override val savedStateRegistry: SavedStateRegistry = SavedStateRegistryController.create(this).savedStateRegistry
    private fun handleLifecycleEvent(event: Lifecycle.Event) =
        _lifecycleRegistry.handleLifecycleEvent(event)

    @CallSuper
    override fun onCreate() {
        super.onCreate()
        // You must call performAttach() before calling performRestore(Bundle)
        savedStateRegistry.performAttach(lifecycle)
        savedStateRegistry.performRestore(null)
        handleLifecycleEvent(Lifecycle.Event.ON_CREATE)
    }

    @CallSuper
    override fun onTaskRemoved(rootIntent: Intent?) {
        super.onTaskRemoved(rootIntent)
        stopSelf()
    }

    @CallSuper
    override fun onDestroy() {
        super.onDestroy()
        handleLifecycleEvent(Lifecycle.Event.ON_DESTROY)
    }

    lateinit var viewModel: RepoViewModel
    override fun onCreateInputView(): View {
        val app = (application as Nitroless)
        val view = ComposeKeyboardView(this)
        viewModel = ViewModelProvider(this, RepoViewModelFactory(app.repository, app.communityReposRepository)).get(RepoViewModel::class.java)

        this.attachToDecorView(
            window?.window?.decorView
        )
        return view
    }

    fun attachToDecorView(decorView: View?) {
        if (decorView == null) return

        ViewTreeLifecycleOwner.set(decorView, this)
        ViewTreeViewModelStoreOwner.set(decorView, this)
        decorView.setViewTreeSavedStateRegistryOwner(this)
    }
}

我已在onCreateInputView()方法中配置了ViewModelProvider，能够正常访问ViewModel，但无法实现ComposeKeyboardView的更新。

我的ComposeKeyboardView代码： #

class ComposeKeyboardView(context: Context): AbstractComposeView(context) {
    @Composable
    override fun Content() {
        val viewModel: RepoViewModel = (context as IMEService).viewModel

        KeyboardScreen()
    }
}

解决方案 #

1. 确保ViewModel使用可观察状态容器 #

ViewModel中的状态必须用MutableStateFlow或StateFlow封装，让Compose能感知状态变化：

class RepoViewModel(...) : ViewModel() {
    // 示例：用MutableStateFlow定义可观察状态
    private val _dataState = MutableStateFlow<YourDataType>(initialValue)
    val dataState = _dataState.asStateFlow()

    // 更新状态的方法
    fun updateState(newData: YourDataType) {
        _dataState.value = newData
    }
}

2. 在Compose中正确观察状态并绑定ViewModel #

修改ComposeKeyboardView的Content方法，通过Compose内置的viewModel()方法获取ViewModel（避免强转Context），并使用collectAsState()监听状态变化：

class ComposeKeyboardView(context: Context): AbstractComposeView(context) {
    @Composable
    override fun Content() {
        // 通过LocalViewModelStoreOwner获取ViewModel，保证生命周期绑定正确
        val viewModel: RepoViewModel = viewModel()
        // 观察状态，状态更新时自动触发重组
        val currentState = viewModel.dataState.collectAsState()

        KeyboardScreen(data = currentState.value)
    }
}

3. 调整ViewModel初始化时机 #

在onCreateInputView中先初始化ViewModel，再创建ComposeKeyboardView，避免Compose加载时ViewModel未完成初始化：

override fun onCreateInputView(): View {
    val app = (application as Nitroless)
    // 先初始化ViewModel
    viewModel = ViewModelProvider(this, RepoViewModelFactory(app.repository, app.communityReposRepository)).get(RepoViewModel::class.java)
    // 再创建ComposeKeyboardView
    val view = ComposeKeyboardView(this)

    this.attachToDecorView(
        window?.window?.decorView
    )
    return view
}

4. 完善InputMethodService的生命周期分发 #

补充onStartInputView和onFinishInputView的生命周期处理，让Compose能感知输入法的显示/隐藏状态：

@CallSuper
override fun onStartInputView(info: EditorInfo?, restarting: Boolean) {
    super.onStartInputView(info, restarting)
    handleLifecycleEvent(Lifecycle.Event.ON_START)
}

@CallSuper
override fun onFinishInputView(finishingInput: Boolean) {
    super.onFinishInputView(finishingInput)
    handleLifecycleEvent(Lifecycle.Event.ON_STOP)
}

完成以上调整后，ViewModel的状态更新会被Compose感知，自动触发ComposeKeyboardView的重组，实现UI的实时更新。

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