Kotlin 枚举和 when

image.png

前言

---

• 在接下来这章中，我们将会讨论 Kotlin 中的枚举和 when 结构。

枚举类

---

• 在 Java 中，声明一个枚举类只需要一个关键字 enum 就可以了，但是在 Kotlin 中如果声明一个枚举类，你需要 enum 和 class 两个关键字。接下来我们声明一个有关颜色的枚举。

    enum class Color {
        RED, ORANGE, YELLOW, GREEN, BLUE, INDIGO, VIOLET
    }
  

• 这里的 enum 是一个软关键字，它只有在 class 前才有意义。

• 和 Java 中一样，枚举并不仅仅是值的列表，也可以给枚举类声明属性和方法。

    enum class Color(val r: Int, val g: Int, val b: Int) {
        RED(255, 0, 0), ORANGE(255, 165, 0), YELLOW(255, 255, 0),
        GREEN(0, 255, 0), BLUE(0, 0, 255), INDIGO(75, 0, 130), 
        VIOLET(238, 130, 238); // 这里必须要有一个分号
  
        fun rgb() = (r * 256 + g) * 256 + b
    }
  

• 枚举常量声明构造方法和属性与常规类相同，当你声明每个枚举常量的时候，必须提供该常量的属性值。

这个例子中向你展现了 Kotlin 中唯一必须使用分号的地方：如果要在枚举类中定义方法，就要使用分号把枚举常量类别和方法声明分隔开。

when

---

• 如果你需要一个函数来告诉你每个颜色对应的字符串，在 Java 中可以用 switch 语句完成，而在 Kotlin 中对应的结构是 when。
• 和 if 类似，Kotlin 中的 when 也是一个有返回值的表达式，所以可以写一个直接返回表达式的表达式体函数。

fun getString(color: Color) = 
    when(color) {
        Color.RED -> "red"
        Color.ORANGE -> "orange"
        Color.YELLOW -> "yellow"
        Color.GREEN -> "green"
        Color.BLUE -> "blue"
        Color.INDIGO -> "indigo"
        Color.VIOLET -> "violet"
    }

• 和 Java 不同的是，你不需要在每个分支后面加上 break，如果匹配成功，只会执行对应的分支。
• 也可以把多个值合并到一个分支，只需要用逗号隔开即可。

fun getString(color: Color) = 
    when(color) {
        Color.RED, Color.ORANGE, Color.YELLOW -> "warm"
        Color.GREEN, Color.BLUE, Color.INDIGO, Color.VIOLET -> "cold"
    }

• Kotlin 中的 when 要比 Java 中的 switch 强大的多，switch 要求必须使用常量（枚举常量、字符串或数字字面值）作为分支条件，但是，when 可以使用任何对象作为分支。
• 比如写一个工具方法进行四舍五入运算。

fun <N: Number> round(num: N): Int { // 使用泛型，所有基本数值类型都是继承 Number
    return when(num) {
        is Float -> (this + 0.5F).toInt() // 是浮点类型，进行四舍五入计算
        is Double -> (this + 0.5).toInt()
        else -> num.toInt() // 上面的都不符合，执行 else 分支
    }
}

• Kotlin 的 Number 类中，提供一系列的 toInt()、toFloat() 等方法，用于类型间切换。
• 同时，when 表达式也可以不带参数。

fun <N: Number> round(num: N): Int {
    return when {
        is Float -> (this + 0.5F).toInt()
        is Double -> (this + 0.5).toInt()
        else -> num.toInt() 
    }
}

• 这样就要求分支条件是任意的布尔表达式。

智能转换

---

• 在上面的例子中，我们用到了 is 关键字，用来判断 num 是否是对应的类型， is 关键字和 Java 中的 instanceof 相似，但是 Kotlin 中的 is 功能更加强大。
• 这里我们用到了一个动画播放的例子。

/* Java */
Drawable drawable = iv.getDrawable();
if(drawable instanceof AnimationDrawable) { // 使用 instanceof 判断是否是动画类型
    AnimationDrawable anim = (AnimationDrawable) drawable; // 强制类型转换
    anim.start();
}

/* Kotlin */
val drawable  = iv.drawable
if(drawable is AnimationDrawable) { // 使用 is 判断是否是动画类型
    drawable.start() // 判断成功自动类型转换
}

• 在 Kotlin 中，如果你使用 is 检查过一个变量是不是某一类型，那么后面就不需要再进行类型转换，变量可以直接当作检查过的类型使用。事实上，编译器为你执行了类型转换，我们把这种行为称为智能转换。
• 智能转换只有在变量经过 is 检查后不再发生变化才有效，当你对一个属性进行智能转换的时候，这个属性必须是 val 属性，并且没有自定义的访问器。

在有需要的时候也可以使用 as 来进行显示转换
val drawable = iv.drawable as AnimationDrawable

最后

---

• 今天就讲到这里，如有疑问欢迎在评论区提问。
• 欢迎关注~