UiState.copy lambda未返回UiState对象致ViewModel编译错误

问题场景 #

在ViewModel中更新StateFlow的UiState时，调用UiState.copy()本应返回修改后的状态对象，但代码编译失败，提示lambda返回类型不匹配：需要CalendarUiState，实际返回Unit。

原代码 #

class CalendarViewModel: ViewModel() {
    private val _calendarUiState = MutableStateFlow(CalendarUiState())
    val calendarUiState: StateFlow<CalendarUiState> = _calendarUiState.asStateFlow()

    fun test() {
        _calendarUiState.update(
            { calendarUiState -> calendarUiState.copy(currentDate = "new_date") }
        )
    }
}

错误提示 #

Type mismatch.
Required:
CalendarUiState
Found:
Unit

问题原因 #

你给update方法传参时多套了一层括号，Kotlin会把括号内的lambda解析成语句块，而语句块默认返回Unit，完全不符合update要求的(T) -> T类型（接收当前状态，返回新状态）。

解决方案 #

去掉lambda外层多余的括号，或者简化lambda写法：

修正后的代码

class CalendarViewModel: ViewModel() {
    private val _calendarUiState = MutableStateFlow(CalendarUiState())
    val calendarUiState: StateFlow<CalendarUiState> = _calendarUiState.asStateFlow()

    fun test() {
        // 写法1：移除外层括号，直接传递lambda
        _calendarUiState.update { calendarUiState -> 
            calendarUiState.copy(currentDate = "new_date") 
        }

        // 写法2：省略参数名，用it指代当前状态（更简洁）
        _calendarUiState.update { 
            it.copy(currentDate = "new_date") 
        }
    }
}

说明 #

MutableStateFlow.update函数要求传入一个能生成新状态的lambda，类型为(当前状态) -> 新状态。修正后，lambda会直接返回copy生成的新CalendarUiState对象，类型匹配，编译通过且能正常更新状态。

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