You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

508 lines
23 KiB
Markdown

This file contains ambiguous Unicode characters!

This file contains ambiguous Unicode characters that may be confused with others in your current locale. If your use case is intentional and legitimate, you can safely ignore this warning. Use the Escape button to highlight these characters.

# 20控制结构Go中的switch语句有哪些变化
你好我是Tony Bai。
经过前两节课的学习,我们已经掌握了控制结构中的分支结构以及循环结构。前面我们也提到过,在计算机世界中,再复杂的算法都可以通过顺序、分支和循环这三种基本的控制结构构造出来。所以,理论上讲,我们现在已经具备了实现任何算法的能力了。
不过理论归理论我们还是要回到现实中来继续学习Go语言中的控制结构现在我们还差一种分支控制结构没讲。除了if语句之外Go语言还提供了一种更适合多路分支执行的分支控制结构也就是**switch语句**。
今天这一节课我们就来系统学习一下switch语句。Go语言中的switch语句继承自它的先祖C语言所以我们这一讲的重点是Go switch语句相较于C语言的switch有哪些重要的改进与创新。
在讲解改进与创新之前我们先来认识一下switch语句。
## 认识switch语句
我们先通过一个例子来直观地感受一下switch语句的优点。在一些执行分支较多的场景下使用switch分支控制语句可以让代码更简洁可读性更好。
比如下面例子中的readByExt函数会根据传入的文件扩展名输出不同的日志它使用了if语句进行分支控制
```plain
func readByExt(ext string) {
if ext == "json" {
println("read json file")
} else if ext == "jpg" || ext == "jpeg" || ext == "png" || ext == "gif" {
println("read image file")
} else if ext == "txt" || ext == "md" {
println("read text file")
} else if ext == "yml" || ext == "yaml" {
println("read yaml file")
} else if ext == "ini" {
println("read ini file")
} else {
println("unsupported file extension:", ext)
}
}
```
如果用switch改写上述例子代码我们可以这样来写
```plain
func readByExtBySwitch(ext string) {
switch ext {
case "json":
println("read json file")
case "jpg", "jpeg", "png", "gif":
println("read image file")
case "txt", "md":
println("read text file")
case "yml", "yaml":
println("read yaml file")
case "ini":
println("read ini file")
default:
println("unsupported file extension:", ext)
}
}
```
从代码呈现的角度来看针对这个例子使用switch语句的实现要比if语句的实现更加简洁紧凑。并且即便你这个时候还没有系统学过switch语句相信你也能大致读懂上面readByExtBySwitch的执行逻辑。
简单来说readByExtBySwitch函数就是将输入参数ext与每个case语句后面的表达式做比较如果相等就执行这个case语句后面的分支然后函数返回。这里具体的执行逻辑我们在后面再分析现在你有个大概认识就好了。
接下来我们就来进入正题来看看Go语言中switch语句的一般形式
```plain
switch initStmt; expr {
case expr1:
// 执行分支1
case expr2:
// 执行分支2
case expr3_1, expr3_2, expr3_3:
// 执行分支3
case expr4:
// 执行分支4
... ...
case exprN:
// 执行分支N
default:
// 执行默认分支
}
```
我们按语句顺序来分析一下。首先看这个switch语句一般形式中的第一行这一行由switch关键字开始它的后面通常接着一个表达式expr这句中的initStmt是一个可选的组成部分。和if、for语句一样我们可以在initStmt中通过短变量声明定义一些在switch语句中使用的临时变量。
接下来switch后面的大括号内是一个个代码执行分支每个分支以case关键字开始每个case后面是一个表达式或是一个逗号分隔的表达式列表。这里还有一个以default关键字开始的特殊分支被称为**默认分支**。
最后我们再来看switch语句的执行流程。其实也很简单switch语句会用expr的求值结果与各个case中的表达式结果进行比较如果发现匹配的case也就是case后面的表达式或者表达式列表中任意一个表达式的求值结果与expr的求值结果相同那么就会执行该case对应的代码分支分支执行后switch语句也就结束了。如果所有case表达式都无法与expr匹配那么程序就会执行default默认分支并且结束switch语句。
那么问题就来了在有多个case执行分支的switch语句中**Go是按什么次序对各个case表达式进行求值并且与switch表达式expr进行比较的**
我们通过一段示例代码来回答这个问题。这是一个一般形式的switch语句为了能呈现switch语句的执行次序我以多个输出特定日志的函数作为switch表达式以及各个case表达式
```plain
func case1() int {
println("eval case1 expr")
return 1
}
func case2_1() int {
println("eval case2_1 expr")
return 0
}
func case2_2() int {
println("eval case2_2 expr")
return 2
}
func case3() int {
println("eval case3 expr")
return 3
}
func switchexpr() int {
println("eval switch expr")
return 2
}
func main() {
switch switchexpr() {
case case1():
println("exec case1")
case case2_1(), case2_2():
println("exec case2")
case case3():
println("exec case3")
default:
println("exec default")
}
}
```
执行一下这个示例程序,我们得到如下结果:
```plain
eval switch expr
eval case1 expr
eval case2_1 expr
eval case2_2 expr
exec case2
```
从输出结果中我们看到Go先对switch expr表达式进行求值然后再按case语句的出现顺序从上到下进行逐一求值。在带有表达式列表的case语句中Go会从左到右对列表中的表达式进行求值比如示例中的case2\_1函数就执行于case2\_2函数之前。
如果switch表达式匹配到了某个case表达式那么程序就会执行这个case对应的代码分支比如示例中的“exec case2”。这个分支后面的case表达式将不会再得到求值机会比如示例不会执行case3函数。这里要注意一点即便后面的case表达式求值后也能与switch表达式匹配上Go也不会继续去对这些表达式进行求值了。
除了这一点外你还要注意default分支。**无论default分支出现在什么位置它都只会在所有case都没有匹配上的情况下才会被执行的。**
不知道你有没有发现这里其实有一个优化小技巧考虑到switch语句是按照case出现的先后顺序对case表达式进行求值的那么如果我们将匹配成功概率高的case表达式排在前面就会有助于提升switch语句执行效率。这点对于case后面是表达式列表的语句同样有效我们可以将匹配概率最高的表达式放在表达式列表的最左侧。
到这里我们已经了解了switch语句的一般形式以及执行次序。有了这个基础后接下来我们就来看看这节课重点Go语言的switch语句和它的“先祖”C语言中的Switch语句相比都有哪些优化与创新
## switch语句的灵活性
为方便对比我们先来简单了解一下C语言中的switch语句。C语言中的switch语句对表达式类型有限制每个case语句只可以有一个表达式。而且除非你显式使用break跳出程序默认总是执行下一个case语句。这些特性开发人员带来了使用上的心智负担。
相较于C语言中switch语句的“死板”Go的switch语句表现出极大的灵活性主要表现在如下几方面
**首先switch语句各表达式的求值结果可以为各种类型值只要它的类型支持比较操作就可以了。**
C语言中switch语句中使用的所有表达式的求值结果只能是int或枚举类型其他类型都会被C编译器拒绝。
Go语言就宽容得多了只要类型支持比较操作都可以作为switch语句中的表达式类型。比如整型、布尔类型、字符串类型、复数类型、元素类型都是可比较类型的数组类型甚至字段类型都是可比较类型的结构体类型也可以。下面就是一个使用自定义结构体类型作为switch表达式类型的例子
```plain
type person struct {
name string
age int
}
func main() {
p := person{"tom", 13}
switch p {
case person{"tony", 33}:
println("match tony")
case person{"tom", 13}:
println("match tom")
case person{"lucy", 23}:
println("match lucy")
default:
println("no match")
}
}
```
不过实际开发过程中以结构体类型为switch表达式类型的情况并不常见这里举这个例子仅是为了说明Go switch语句对各种类型支持的广泛性。
而且当switch表达式的类型为布尔类型时如果求值结果始终为true那么我们甚至可以省略switch后面的表达式比如下面例子
```plain
// 带有initStmt语句的switch语句
switch initStmt; {
case bool_expr1:
case bool_expr2:
... ...
}
// 没有initStmt语句的switch语句
switch {
case bool_expr1:
case bool_expr2:
... ...
}
```
不过这里要注意在带有initStmt的情况下如果我们省略switch表达式那么initStmt后面的分号不能省略因为initStmt是一个语句。
**第二点switch语句支持声明临时变量。**
在前面介绍switch语句的一般形式中我们看到和if、for等控制结构语句一样switch语句的initStmt可用来声明只在这个switch隐式代码块中使用的变量这种就近声明的变量最大程度地缩小了变量的作用域。
**第三点case语句支持表达式列表。**
在C语言中如果要让多个case分支的执行相同的代码逻辑我们只能通过下面的方式实现
```plain
void check_work_day(int a) {
switch(a) {
case 1:
case 2:
case 3:
case 4:
case 5:
printf("it is a work day\n");
break;
case 6:
case 7:
printf("it is a weekend day\n");
break;
default:
printf("do you live on earth?\n");
}
}
```
在上面这段C代码中case 1~case 5匹配成功后执行的都是case 5中的代码逻辑case 6~case 7匹配成功后执行的都是case 7中的代码逻辑。
之所以可以实现这样的逻辑是因为当C语言中的switch语句匹配到某个case后如果这个case对应的代码逻辑中没有break语句那么代码将继续执行下一个case。比如当a = 3时case 3后面的代码为空逻辑并且没有break语句那么C会继续向下执行case4、case5直到在case 5中调用了break代码执行流才离开switch语句。
这样看虽然C也能实现多case语句执行同一逻辑的功能但在case分支较多的情况下代码会显得十分冗长。
Go语言中的处理要好得多。Go语言中switch语句在case中支持表达式列表。我们可以用表达式列表实现与上面的示例相同的处理逻辑
```plain
func checkWorkday(a int) {
switch a {
case 1, 2, 3, 4, 5:
println("it is a work day")
case 6, 7:
println("it is a weekend day")
default:
println("are you live on earth")
}
}
```
根据前面我们讲过的switch语句的执行次序理解上面这个例子应该不难。和C语言实现相比使用case表达式列表的Go实现简单、清晰、易懂。
**第四点取消了默认执行下一个case代码逻辑的语义。**
在前面的描述和check\_work\_day这个C代码示例中你都能感受到在C语言中如果匹配到的case对应的代码分支中没有显式调用break语句那么代码将继续执行下一个case的代码分支这种“隐式语义”并不符合日常算法的常规逻辑这也经常被诟病为C语言的一个缺陷。要修复这个缺陷我们只能在每个case执行语句中都显式调用break。
Go语言中的Swith语句就修复了C语言的这个缺陷取消了默认执行下一个case代码逻辑的“非常规”语义每个case对应的分支代码执行完后就结束switch语句。
如果在少数场景下你需要执行下一个case的代码逻辑你可以显式使用Go提供的关键字fallthrough来实现这也是Go“显式”设计哲学的一个体现。下面就是一个使用fallthrough的switch语句的例子我们简单来看一下
```plain
func case1() int {
println("eval case1 expr")
return 1
}
func case2() int {
println("eval case2 expr")
return 2
}
func switchexpr() int {
println("eval switch expr")
return 1
}
func main() {
switch switchexpr() {
case case1():
println("exec case1")
fallthrough
case case2():
println("exec case2")
fallthrough
default:
println("exec default")
}
}
```
执行一下这个示例程序,我们得到这样的结果:
```plain
eval switch expr
eval case1 expr
exec case1
exec case2
exec default
```
我们看到switch expr的求值结果与case1匹配成功Go执行了case1对应的代码分支。而且由于case1代码分支中显式使用了fallthrough执行完case1后代码执行流并没有离开switch语句而是继续执行下一个case也就是case2的代码分支。
这里有一个注意点由于fallthrough的存在Go不会对case2的表达式做求值操作而会直接执行case2对应的代码分支。而且在这里case2中的代码分支也显式使用了fallthrough于是最后一个代码分支也就是default分支对应的代码也被执行了。
另外还有一点要注意的是如果某个case语句已经是switch语句中的最后一个case了并且它的后面也没有default分支了那么这个case中就不能再使用fallthrough否则编译器就会报错。
到这里我们看到Go的switch语句不仅修复了C语言switch的缺陷还为Go开发人员提供了更大的灵活性我们可以使用更多类型表达式作为switch表达式类型也可以使用case表达式列表简化实现逻辑还可以自行根据需要确定是否使用fallthrough关键字继续向下执行下一个case的代码分支。
除了这些之外Go语言的switch语句还支持求值结果为类型信息的表达式也就是type switch语句接下来我们就详细分析一下。
## type switch
“type switch”这是一种特殊的switch语句用法我们通过一个例子来看一下它具体的使用形式
```plain
func main() {
var x interface{} = 13
switch x.(type) {
case nil:
println("x is nil")
case int:
println("the type of x is int")
case string:
println("the type of x is string")
case bool:
println("the type of x is string")
default:
println("don't support the type")
}
}
```
我们看到这个例子中switch语句的形式与前面是一致的不同的是switch与case两个关键字后面跟着的表达式。
switch关键字后面跟着的表达式为`x.(type)`这种表达式形式是switch语句专有的而且也只能在switch语句中使用。这个表达式中的**x必须是一个接口类型变量**,表达式的求值结果是这个接口类型变量对应的动态类型。
什么是一个接口类型的动态类型呢?我们简单解释一下。以上面的代码`var x interface{} = 13`为例x是一个接口类型变量它的静态类型为`interface{}`如果我们将整型值13赋值给xx这个接口变量的动态类型就为int。关于接口类型变量的动态类型我们后面还会详细讲这里先简单了解一下就可以了。
接着case关键字后面接的就不是普通意义上的表达式了而是一个个具体的类型。这样Go就能使用变量x的动态类型与各个case中的类型进行匹配之后的逻辑就都是一样的了。
现在我们运行上面示例程序输出了x的动态变量类型
```plain
the type of x is int
```
不过,通过`x.(type)`我们除了可以获得变量x的动态类型信息之外也能获得其动态类型对应的值信息现在我们把上面的例子改造一下
```plain
func main() {
var x interface{} = 13
switch v := x.(type) {
case nil:
println("v is nil")
case int:
println("the type of v is int, v =", v)
case string:
println("the type of v is string, v =", v)
case bool:
println("the type of v is bool, v =", v)
default:
println("don't support the type")
}
}
```
这里我们将switch后面的表达式由`x.(type)`换成了`v := x.(type)`。对于后者你千万不要认为变量v存储的是类型信息其实**v存储的是变量x的动态类型对应的值信息**这样我们在接下来的case执行路径中就可以使用变量v中的值信息了。
然后我们运行上面示例可以得到v的动态类型和值
```plain
the type of v is int, v = 13
```
另外你可以发现在前面的type switch演示示例中我们一直使用interface{}这种接口类型的变量Go中所有类型都实现了interface{}类型所以case后面可以是任意类型信息。
但如果在switch后面使用了某个特定的接口类型I那么case后面就只能使用实现了接口类型I的类型了否则Go编译器会报错。你可以看看这个例子
```plain
type I interface {
M()
}
type T struct {
}
func (T) M() {
}
func main() {
var t T
var i I = t
switch i.(type) {
case T:
println("it is type T")
case int:
println("it is type int")
case string:
println("it is type string")
}
}
```
在这个例子中我们在type switch中使用了自定义的接口类型I。那么理论上所有case后面的类型都只能是实现了接口I的类型。但在这段代码中只有类型T实现了接口类型IGo原生类型int与string都没有实现接口I于是在编译上述代码时编译器会报出如下错误信息
```plain
19:2: impossible type switch case: i (type I) cannot have dynamic type int (missing M method)
21:2: impossible type switch case: i (type I) cannot have dynamic type string (missing M method)
```
好了到这里关于switch语句语法层面的知识就都学习完了。Go对switch语句的优化与增强使得我们在日常使用switch时很少遇到坑但这也并不意味着没有最后我们就来看在Go编码过程中我们可能遇到的一个与switch使用有关的问题跳不出循环的break。
## 跳不出循环的break
在上一节课讲解break语句的时候我们曾举了一个找出整型切片中第一个偶数的例子当时我们是把for与if语句结合起来实现的。现在我们把那个例子中if分支结构换成这节课学习的switch分支结构试试看。我们这里直接看改造后的代码
```plain
func main() {
var sl = []int{5, 19, 6, 3, 8, 12}
var firstEven int = -1
// find first even number of the interger slice
for i := 0; i < len(sl); i++ {
switch sl[i] % 2 {
case 0:
firstEven = sl[i]
break
case 1:
// do nothing
}
}
println(firstEven)
}
```
我们运行一下这个修改后的程序得到结果为12。
奇怪这个输出的值与我们的预期的好像不太一样。这段代码中切片中的第一个偶数是6而输出的结果却成了切片的最后一个偶数12。为什么会出现这种结果呢
这就是Go中 break语句与switch分支结合使用会出现一个“小坑”。和我们习惯的C家族语言中的break不同Go语言规范中明确规定**不带label的break语句中断执行并跳出的是同一函数内break语句所在的最内层的for、switch或select**。所以上面这个例子的break语句实际上只跳出了switch语句并没有跳出外层的for循环这也就是程序未按我们预期执行的原因。
要修正这一问题我们可以利用上节课学到的带label的break语句试试。这里我们也直接看看改进后的代码:
```plain
func main() {
var sl = []int{5, 19, 6, 3, 8, 12}
var firstEven int = -1
// find first even number of the interger slice
loop:
for i := 0; i < len(sl); i++ {
switch sl[i] % 2 {
case 0:
firstEven = sl[i]
break loop
case 1:
// do nothing
}
}
println(firstEven) // 6
}
```
在改进后的例子中我们定义了一个labelloop这个label附在for循环的外面指代for循环的执行。当代码执行到“break loop”时程序将停止label loop所指代的for循环的执行。关于带有label的break语句你可以再回顾一下第19讲这里就不多说了。
和switch语句一样能阻拦break跳出的还有一个语句那就是select我们后面讲解并发程序设计的时候再来详细分析。
## 小结
好了,今天的课讲到这里就结束了,现在我们一起来回顾一下吧。
在这一讲中我们讲解了Go语言提供的另一种分支控制结构switch语句。和if分支语句相比在一些执行分支较多的场景下使用switch分支控制语句可以让代码更简洁、可读性更好。
Go语言的switch语句继承自C语言但“青出于蓝而胜于蓝”Go不但修正了C语言中switch语句默认执行下一个case的“坑”还对switch语句进行了改进与创新包括支持更多类型、支持表达式列表等让switch的表达力得到进一步提升。
除了使用常规表达式作为switch表达式和case表达式之外Go switch语句又创新性地支持type switch也就是用类型信息作为分支条件判断的操作数。在Go中这种使用方式也是switch所独有的。这里我们要注意的是只有接口类型变量才能使用type switch并且所有case语句中的类型必须实现switch关键字后面变量的接口类型。
最后还需要你记住的是switch会阻拦break语句跳出for循环就像我们这节课最后那个例子中那样对于初学者来说这是一个很容易掉下去的坑你一定不要走弯路。
## 思考题
为了验证在多分支下基于switch语句实现的分支控制更为简洁你可以尝试将这节课中的那些稍复杂一点的例子改写为基于if条件分支的实现然后再对比一下两种实现的复杂性直观体会一下switch语句的优点。
欢迎你把这节课分享给更多对Go语言中的switch语句感兴趣的朋友。我是Tony Bai我们下节课见。