Kotlin中如何合并针对同一枚举的多个when语句？

问题场景 #

给定枚举类：

enum class As(val value: String) {
    A1("a1"),
    A2("a2"),
    A3("a3"),
}

当前需要根据As类型实例设置color和padding两个变量，实现方式为两个独立的when语句：

val enumAs = As.A1

val color = when(enumAs){
    A1 -> Color.White
    else -> Color.Black
}

val padding = when(enumAs){
    A1 -> 8.dp
    A2 -> 12.dp
    else -> 16.dp
}

希望找到更优雅的实现方式，避免重复编写when判断。

优化方案 #

1. 枚举类内置属性 #

将color和padding直接作为枚举的成员属性，把配置逻辑集中到枚举内部：

enum class As(val value: String, val color: Color, val padding: Dp) {
    A1("a1", Color.White, 8.dp),
    A2("a2", Color.Black, 12.dp),
    A3("a3", Color.Black, 16.dp),
}

// 使用示例
val enumAs = As.A1
val color = enumAs.color
val padding = enumAs.padding

优点：逻辑高度内聚，新增枚举值时必须同步配置对应属性，避免遗漏；使用时直接访问属性，代码简洁。

2. 数据类封装+单次when判断 #

定义数据类封装样式配置，通过一次when返回完整配置，再解构赋值：

data class StyleConfig(val color: Color, val padding: Dp)

val enumAs = As.A1
val (color, padding) = when(enumAs) {
    As.A1 -> StyleConfig(Color.White, 8.dp)
    As.A2 -> StyleConfig(Color.Black, 12.dp)
    As.A3 -> StyleConfig(Color.Black, 16.dp)
}

优点：仅需一次when判断，减少重复逻辑；配置逻辑集中，可读性更强。

3. 扩展函数封装逻辑 #

通过扩展函数将color和padding的获取逻辑封装到枚举类外部，无需修改原枚举：

fun As.getColor(): Color = when(this) {
    As.A1 -> Color.White
    else -> Color.Black
}

fun As.getPadding(): Dp = when(this) {
    As.A1 -> 8.dp
    As.A2 -> 12.dp
    else -> 16.dp
}

// 使用示例
val enumAs = As.A1
val color = enumAs.getColor()
val padding = enumAs.getPadding()

优点：无需修改原枚举类，实现逻辑与枚举解耦，适合无法改动原枚举的场景。

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