如何在Kotlin ViewModel中编写可复用的多数值增减函数？

完全可以实现适配所有追踪器的通用增减函数，下面给你两种实用方案，能大幅减少代码重复，符合Jetpack最佳实践：

方案1：基于Lambda的通用更新函数（推荐，类型安全无反射） #

这种方案不需要依赖反射，通过传递修改逻辑的Lambda实现通用更新，既灵活又能保证编译期类型安全，扩展性极强。

1. ViewModel中实现通用函数 #

class TrackerViewModel(private val trackerRepository: TrackerRepository) : ViewModel() {
    // 用MutableStateFlow维护UI状态（假设你原本是这么实现的）
    private val _trackerUiState = MutableStateFlow(TrackerDetails())
    val trackerUiState = _trackerUiState.asStateFlow()

    // 通用更新函数：接收一个修改Tracker对象的Lambda
    fun updateTracker(modifier: (Tracker) -> Tracker) {
        viewModelScope.launch {
            val currentState = _trackerUiState.value
            val updatedTracker = modifier(currentState.tracker)
            
            // 更新数据库
            trackerRepository.updateTracker(updatedTracker)
            // 更新UI状态，触发Compose重组
            _trackerUiState.update { it.copy(tracker = updatedTracker) }
        }
    }
}

2. Compose中调用通用函数 #

@Composable
fun TrackerScreen(viewModel: TrackerViewModel) {
    val trackerUiState = viewModel.trackerUiState.collectAsState()
    val currentTracker = trackerUiState.value.tracker

    Column {
        // 数值1追踪器
        Column {
            Text(text = "数值1：${currentTracker.valueOne}")
            Row {
                Button(onClick = {
                    viewModel.updateTracker { it.copy(valueOne = it.valueOne + 1) }
                }) {
                    Text("+1")
                }
                Button(onClick = {
                    viewModel.updateTracker { it.copy(valueOne = it.valueOne - 1) }
                }) {
                    Text("-1")
                }
            }
        }

        // 数值2追踪器
        Column {
            Text(text = "数值2：${currentTracker.valueTwo}")
            Row {
                Button(onClick = {
                    viewModel.updateTracker { it.copy(valueTwo = it.valueTwo + 1) }
                }) {
                    Text("+1")
                }
                Button(onClick = {
                    viewModel.updateTracker { it.copy(valueTwo = it.valueTwo - 1) }
                }) {
                    Text("-1")
                }
            }
        }

        // 新增其他追踪器时，直接复制上面的结构修改属性名即可
    }
}

优势：

• ViewModel无需为每个数值单独写函数，代码量骤减
• 编译期检查修改逻辑，避免运行时错误
• 灵活性高，除了增减1，还能轻松实现重置、增减固定值等逻辑

方案2：基于属性引用的通用增减函数（调用更简洁） #

如果希望Compose端的调用代码更简洁，可以结合Kotlin属性引用，封装专门的增减函数，适合只需要增减固定值的场景。

1. ViewModel中实现通用增减函数 #

import kotlin.reflect.KProperty1
import kotlin.reflect.full.constructors
import kotlin.reflect.full.memberProperties

class TrackerViewModel(private val trackerRepository: TrackerRepository) : ViewModel() {
    private val _trackerUiState = MutableStateFlow(TrackerDetails())
    val trackerUiState = _trackerUiState.asStateFlow()

    // 通用增减函数：接收要修改的属性引用和增减量
    fun adjustTrackerValue(property: KProperty1<Tracker, Int>, delta: Int) {
        viewModelScope.launch {
            val currentState = _trackerUiState.value
            val currentTracker = currentState.tracker
            val currentValue = property.get(currentTracker)

            // 通过反射动态生成修改后的Tracker对象（无需手动枚举所有属性）
            val updatedProps = Tracker::class.memberProperties.associate { prop ->
                prop to if (prop == property) currentValue + delta else prop.get(currentTracker)
            }
            val updatedTracker = Tracker::class.constructors.first().callBy(updatedProps)

            trackerRepository.updateTracker(updatedTracker)
            _trackerUiState.update { it.copy(tracker = updatedTracker) }
        }
    }
}

2. Compose中调用 #

@Composable
fun TrackerScreen(viewModel: TrackerViewModel) {
    val trackerUiState = viewModel.trackerUiState.collectAsState()
    val currentTracker = trackerUiState.value.tracker

    Column {
        Column {
            Text(text = "数值1：${currentTracker.valueOne}")
            Row {
                Button(onClick = { viewModel.adjustTrackerValue(Tracker::valueOne, 1) }) {
                    Text("+1")
                }
                Button(onClick = { viewModel.adjustTrackerValue(Tracker::valueOne, -1) }) {
                    Text("-1")
                }
            }
        }

        Column {
            Text(text = "数值2：${currentTracker.valueTwo}")
            Row {
                Button(onClick = { viewModel.adjustTrackerValue(Tracker::valueTwo, 1) }) {
                    Text("+1")
                }
                Button(onClick = { viewModel.adjustTrackerValue(Tracker::valueTwo, -1) }) {
                    Text("-1")
                }
            }
        }
    }
}

优势：

• Compose端调用代码更简洁，只需传递属性引用和增减量
• 新增属性时无需修改ViewModel，直接在Compose中使用新的属性引用即可

注意事项 #

• 确保ViewModel中正确维护UI状态的更新，比如用MutableStateFlow的update方法，这样Compose才能感知到状态变化并自动重组UI
• 方案1无反射依赖，性能更优；方案2依赖Kotlin反射，适合对调用简洁性要求更高的场景，日常使用中两者都能满足需求

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