带你领略 Kotlin 中的 “when”魔法

Posted by Dorck on May 21, 2019

提到 when,大家都会联想到 Java 中的 switch,然而在 kotlin 中,when 显然比 Java 中的 switch 要强大得多。首先,我们先来看看 when 的特点:

  • 它可以作为表达式使用
  • 使用更加安全
  • 强大灵活的分支结构
  • 可以不带参数

接下来,我来带大家逐步领略这些特点。以下面这段 Java 功能代码为例:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
switch(animal) {
    case EAGLE:
        System.out.println("鸟类");
        break;
    case DOLPHIN:
        System.out.println("兽类");
        break;
    case LOCUST:
        System.out.println("昆虫类");
        break;
    case CARP:
        System.out.println("鱼类");
        break;
    case TIGER:
        System.out.println("兽类");
        break;
    case DUCK:
        System.out.println("鸟类");
        break;
    default:
        System.out.println("未知动物");
        break;
}

从以上代码可以看出,我们这里实现的功能是:通过用户输入一个动物名称来得到其对应的种类信息。乍一看,你可能会觉得上面的代码再正常不过,但是倘若我们的输入情形很多的话,就会增加密密麻麻的限制条件,这肯定是我们不想看到的。那么,让我们来看看通过 Kotlin 的 when 表达式如何实现相同功能:

1
2
3
4
5
6
7
when(animal) {
    EAGLE, DUCK -> println("鸟类")
    DOLPHIN, TIGER -> println("兽类")
    CARP -> println("鱼类")
    LOCUST -> println("昆虫类")
    else -> println("未知动物")
}

OK,以上就是该功能代码的 kotlin 实现方式,代码不仅简化了很多,也省去了大量的 break 语句,避免了 Java 中因遗漏 break 而导致的 bug,增强了安全性。如果匹配成功,对应的分支便会执行,同时也可以把多个情况合并到同一个分支,只需要通过逗号隔开,并没有额外的代码。

PS:每个条件分支的处理通过在 -> 之后来进行。

作为表达式使用

首先,我们需要重温一下表达式语句的区别。表达式有值,并能作为另一个表达式的一部分来使用;而语句没有返回值。Java 中的控制结构皆为语句。而在 Kotlin 中,除了循环体结构外,大多数控制结构都是表达式。举个栗子😄,还是上面的例子,我们可以将其优化为以下代码:

1
2
3
4
5
6
7
8
9
var result = when(animal) {
    EAGLE, DUCK -> "鸟类"
    DOLPHIN, TIGER -> "兽类"
    CARP -> "鱼类"
    LOCUST -> "昆虫类"
    else -> "未知动物"
}
// 输入:DUCK,返回:当前动物的种类为:鸟类
println("当前动物的种类为: $result")

此外,我们也可以直接通过表达式函数来直接得到最终的结果:

1
2
3
4
5
6
7
8
fun displayAnimalType(animal: Animal) =
            when(animal) {
                Animal.EAGLE, Animal.DUCK -> "鸟类"
                Animal.DOLPHIN, Animal.TIGER -> "兽类"
                Animal.CARP -> "鱼类"
                Animal.LOCUST -> "昆虫类"
                else -> "未知动物"
            }

强大灵活的分支

Kotlin 中的 when 远比 Java 当中的 switch 要强大得多。switch 只能以常量作为分支条件,而 when 允许使用任意对象。emmm~举个简单的例子吧:

1
2
3
4
5
6
7
fun handleResult(score: Int) =
            when(score) {
                in 0..39 -> "fail to go up to the next grade."
                in 40 until 60 -> "not pass, should test again."
                97,98,99,100 -> "learning outstanding results."
                else -> "pass the test."
            }

Java 当中,当我们的分支情况比较多或者每种分支可能会重复多次时,势必会通过 if-else 来各种判断,而在 Kotlin 当中,一个 when 就能帮我们完成这些操作。

此外,如果你想判断一个未知变量的类型,而其可能类型有很多种可能性,那么也可以通过 when 来实现,例如:

1
2
3
4
5
6
7
fun judgeAnimalKind(animal: Any) =
            when(animal) {
                is Bird -> "这是鸟类"
                is Fish -> "这是鱼类"
                judgeIfInsect(animal) -> "这是昆虫"
                else -> "我不知道这是什么动物"
            }

当然,我们也可以将代码块作为我们的分支体,这时候,代码块中最后一个表达式或者变量就是该分支体的返回结果,如:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
fun judgeAnimalKind(animal: Any) =
            when(animal) {
                is Bird -> "这是鸟类"
                is Fish -> "这是鱼类"
                judgeIfInsect(animal) -> "这是昆虫"
                else -> {
                    val kind = if (judgeDolphin(animal) || judgeTiger(animal)) {
                        "这是兽类"
                    }else {
                        "我不知道这是什么动物"
                    }
                    kind
                }
            }

上述代码中,kind 即为我们的 else 分支块中的最终返回结果。

无参的情况

特别地,when 中的参数可能并非满足我们的需求,我们可以选择省略该参数,例如这样:

1
2
3
4
when {
    phoneNumber?.length != 11 -> toast("illegal phone number.")
    password.isNullOrEmpty() -> toast("please input password")
}

总结

通过以上的介绍,我们不难发现:when 的使用场景要比 Java 的 switch 灵活、强大的多,同时,我们也可以借助 when 来重构和优化复杂的 if-else 结构,以简化我们的代码,提高代码的可读性。


许可协议: 署名-非商业性使用-禁止演绎 4.0 国际 转载请保留原文链接及作者。