Go 语言条件语句 switch 结构

到目前为止,我们看到的 Go 程序都是从 main() 函数开始执行,然后按顺序执行该函数体中的代码。但我们经常会需要只有在满足一些特定情况时才执行某些代码,也就是说在代码里进行条件判断。针对这种需求,Go 提供了下面这些条件结构和分支结构:

switch 结构

相比较 C 和 Java 等其它语言而言,Go 语言中的 switch 结构使用上更加灵活。它接受任意形式的表达式:

  1. switch var1 {
  2. case val1:
  3. ...
  4. case val2:
  5. ...
  6. default:
  7. ...
  8. }

变量 var1 可以是任何类型,而 val1val2 则可以是同类型的任意值。类型不被局限于常量或整数,但必须是相同的类型;或者最终结果为相同类型的表达式。前花括号 { 必须和 switch 关键字在同一行

您可以同时判断多个可能符合条件的值,使用逗号分割它们,例如:case val1, val2, val3

每一个 case 分支都是唯一的,从上至下逐一判断,直到匹配为止。Go 语言使用快速的查找算法来测试 switch 条件与 case 分支的匹配情况,直到算法匹配到某个 case 或者进入 default 条件为止。

一旦成功地匹配到某个分支,在执行完相应代码后就会退出整个 switch 代码块,也就是说您不需要特别使用 break 语句来表示结束。

因此,程序也不会自动地去执行下一个分支的代码。如果在执行完每个分支的代码后,还希望继续执行后续分支的代码,可以使用 fallthrough 关键字来达到目的。

因此:

  1. switch i {
  2. case 0: // 空分支,只有当 i == 0 时才会进入分支
  3. case 1:
  4. f() // 当 i == 0 时函数不会被调用
  5. }

并且:

  1. switch i {
  2. case 0: fallthrough
  3. case 1:
  4. f() // 当 i == 0 时函数会被调用
  5. }

case ...:语句之后,您不需要使用花括号将多行语句括起来,但您可以在分支中进行任意形式的编码。当代码块只有一行时,可以直接放置在 case 语句之后。

您同样可以使用 return 语句来提前结束代码块的执行。当您在 switch 语句块中使用 return 语句,并且您的函数是有返回值的,您还需要在 switch 之后添加相应的 return 语句以确保函数始终会返回。

可选的 default 分支可以出现在任何顺序,但最好将它放在最后。它的作用类似与 if-else 语句中的 else,表示不符合任何已给出条件时,执行相关语句。

示例:

package main

import “fmt”

  1. func main() {
  2. var num1 int = 100
  3. switch num1 {
  4. case 98, 99:
  5. fmt.Println("It's equal to 98")
  6. case 100:
  7. fmt.Println("It's equal to 100")
  8. default:
  9. fmt.Println("It's not equal to 98 or 100")
  10. }
  11. }

输出:

  1. It's equal to 100

switch 语句的第二种形式是不提供任何被判断的值(实际上默认为判断是否为 true),然后在每个 case 分支中进行判断不同的条件。当任意分支的判断结果为 true 时,该分支的代码会被执行。这看起来非常像链式的 if-else 语句,但是在判断条件非常多的情况下,提供了可读性更好的书写方式。

  1. switch {
  2. case condition1:
  3. ...
  4. case condition2:
  5. ...
  6. default:
  7. ...
  8. }

例如:

  1. switch {
  2. case i < 0:
  3. f1()
  4. case i == 0:
  5. f2()
  6. case i > 0:
  7. f3()
  8. }

任何支持进行相等判断的类型都可以作为测试表达式的条件,包括 int、string、指针等。

示例:

  1. package main
  2. import "fmt"
  3. func main() {
  4. var num1 int = 7
  5. switch {
  6. case num1 < 0:
  7. fmt.Println("Number is negative")
  8. case num1 > 0 && num1 < 10:
  9. fmt.Println("Number is between 0 and 10")
  10. default:
  11. fmt.Println("Number is 10 or greater")
  12. }
  13. }

输出:

  1. Number is between 0 and 10

switch 语句的第三种形式是包含一个初始化语句:

  1. switch initialization {
  2. case val1:
  3. ...
  4. case val2:
  5. ...
  6. default:
  7. ...
  8. }

这种形式可以非常优雅地进行条件判断:

  1. switch result := calculate(); {
  2. case result < 0:
  3. ...
  4. case result > 0:
  5. ...
  6. default:
  7. // 0
  8. }

在下面这个代码片段中,变量 a 和 b 被平行初始化,然后作为判断条件:

  1. switch a, b := x[i], y[j]; {
  2. case a < b: t = -1
  3. case a == b: t = 0
  4. case a > b: t = 1
  5. }

练习:请说出下面代码片段输出的结果,并自行验证结果准确性:

  1. k := 6
  2. switch k {
  3. case 4:
  4. fmt.Println("was <= 4")
  5. fallthrough
  6. case 5:
  7. fmt.Println("was <= 5")
  8. fallthrough
  9. case 6:
  10. fmt.Println("was <= 6")
  11. fallthrough
  12. case 7:
  13. fmt.Println("was <= 7")
  14. fallthrough
  15. case 8:
  16. fmt.Println("was <= 8")
  17. fallthrough
  18. default:
  19. fmt.Println("default case")
  20. }

分类导航