分支循环

在编写 Go 程序的时候，我们不仅会用前面学到的数据结构来存储数据，还会用到 if、switch、for 来进行条件判断和流程控制，今天我们就来一起学习下它们。

if

if 主要用于条件判断，语法为：

  1. if 条件 {
  2.   # 业务代码
  3. }

先看一个简单例子:

  1. package main
  3. import "fmt"
  5. func main() {
  6.     age := 7
  8.     if age > 6 {
  9.         fmt.Println("It's primary school")
  10.     }
  11. }

我们可以在条件中使用 & 或 || 来进行组合判断：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     age := 7
  8.     if age > 6 && age <= 12 {
  9.         fmt.Println("It's primary school")
  10.     }
  11. }

我们还可以使用 if..else if..else 来实现多分支的条件判断:

  1. package main
  3. import "fmt"
  5. func main() {
  6.     age := 13
  8.     if age > 6 && age <= 12 {
  9.         fmt.Println("It's primary school")
  10.     } else if age > 12 && age <= 15 {
  11.         fmt.Println("It's middle school")
  12.     } else {
  13.         fmt.Println("It's high school")
  14.     }
  15. }

switch

如果我们的条件分支太多，可以考虑使用 switch 替换 if, 例如：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     age := 10
  8.     switch age {
  9.     case 5:
  10.         fmt.Println("The age is 5")
  11.     case 7:
  12.         fmt.Println("The age is 7")
  13.     case 10:
  14.         fmt.Println("The age is 10")
  15.     default:
  16.         fmt.Println("The age is unkown")
  17.     }
  18. }

注意：在 Go 中 switch 只要匹配中了就会中止剩余的匹配项，这和 Java 很大不一样，它需要使用 break 来主动跳出。

switch 的 case 条件可以是多个值，例如：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     age := 7
  8.     switch age {
  9.     case 7, 8, 9, 10, 11, 12:
  10.         fmt.Println("It's primary school")
  11.     case 13, 14, 15:
  12.         fmt.Println("It's middle school")
  13.     case 16, 17, 18:
  14.         fmt.Println("It's high school")
  15.     default:
  16.         fmt.Println("The age is unkown")
  17.     }
  18. }

注意： 同一个 case 中的多值不能重复。

switch 还可以使用 if..else 作为 case 条件，例如：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     age := 7
  8.     switch {
  9.     case age >= 6 && age <= 12:
  10.         fmt.Println("It's primary school")
  11.     case age >= 13 && age <= 15:
  12.         fmt.Println("It's middle school")
  13.     case age >= 16 && age <= 18:
  14.         fmt.Println("It's high school")
  15.     default:
  16.         fmt.Println("The age is unkown")
  17.     }
  18. }

小技巧： 使用 switch 对 interface{} 进行断言，例如：

  1. package main
  3. import "fmt"
  5. func checkType(i interface{}) {
  6.     switch v := i.(type) {
  7.     case int:
  8.         fmt.Printf("%v is an in\n", v)
  9.     case string:
  10.         fmt.Printf("%v is a string\n", v)
  11.     default:
  12.         fmt.Printf("%v's type is unkown\n", v)
  13.     }
  14. }
  16. func main() {
  17.     checkType(8)
  18.     checkType("hello, world")
  19. }

for

使用 for 来进行循环操作，例如：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     for i := 0; i < 2; i++ {
  7.         fmt.Println("loop with index", i)
  8.     }
  9. }

使用 for..range 对数组、切片、map、 字符串等进行循环操作，例如：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     numbers := []int{1, 2, 3}
  8.     for i, v := range numbers {
  9.         fmt.Printf("numbers[%d] is %d\n", i, v)
  10.     }
  11. }

注意: 这里的 i、v 是切片元素的位置索引和值。

  1. package main
  3. import "fmt"
  5. func main() {
  6.     cityCodes := map[string]int{
  7.         "北京": 1,
  8.         "上海": 2,
  9.     }
  11.     for i, v := range cityCodes {
  12.         fmt.Printf("%s is %d\n", i, v)
  13.     }
  14. }

注意: 这里的 i、v 是 map 的 一组键值对的键和值。

使用 continue 和 break 对循环进行控制，例如：

  1. package main
  3. import "fmt"
  5. func main() {
  6.     numbers := []int{1, 2, 3, 4, 5}
  8.     for i, v := range numbers {
  9.         if v == 4 {
  10.             break
  11.         }
  13.         if v%2 == 0 {
  14.             continue
  15.         }
  17.         fmt.Printf("numbers[%d] is %d\n", i, v)
  18.     }
  19. }

注意：

  • break 会结束所有循环。
  • continue 会跳过当前循环直接进入下一次循环。