likes
comments
collection
share

Go语言中常见100问题-#34 break使用误区与解决方法

作者站长头像
站长
· 阅读数 5

前言

break语句通常用来终止一个循环,当循环语句带有switch或select语句时,使用break语句要特别小心,否则会产生bug。

案例引入

下面通过一个具体的例子说明,这段程序在循环内部通过switch判断i的值,如果i的值为2,期望通过break终止循环。

for i := 0; i < 5; i++ {
    fmt.Printf("%d ", i)
    switch i {
        default:
        case 2:
            break
    }
}

这段代码有啥问题吗?咋一看没问题。但是,实际效果并不是我们预期的那样,break语句没有终止循环,终止的是switch语句。输出结果是0 1 2 3 4而不是我们预期的0 1 2.

解决方法

记住一个基本原则,break语句终止的是最内层的for、switch、select语句。在上面的程序中,它终止的是for循环内部的swith语句。那如果想终止外面的for循环,怎么处理呢?最常用的方法是使用标签,示例代码如下。

loop:
for i := 0; i < 5; i++ {
    fmt.Printf("%d ", i)
    switch i {
        default:
        case 2:
            break loop
    }
}

这段代码,将标签loop和for循环关联起来,break loop可以终止loop语句,即整个for循环。运行上述程序,输出结果为0 1 2,与我们预期一致。

标准库处理方法

break label 像 goto语句一样?一些开发者可能对break label是否是惯用做法有疑问,认为它像是一个花哨的goto语句。事实并非如此,在标准库中也可以看到这种使用方法。例如,在 net/http 包中,有下面的语句。

readlines:
for {
    line, err := rw.Body.ReadString('\n')
    switch {
        case err == io.EOF:
            break readlines
        case err != nil:
            t.Fatalf("unexpected error reading from CGI: %v", err)
}
// ...
}

上述程序使用 break readlines终止for循环,强调读完所有行,在Go语言中这是一种惯用方法,不要觉得惊讶。在for select组合代码块中,break语句并不是我们预期的那样终止for循环的执行。例如下面代码,我们想在上下文取消的时候调用break语句终止for循环。

for {
    select {
        case <-ch:
            // Do something
        case <-ctx.Done():
            break
    }
}

在for、switch和select语句中,上述代码最内层的是select语句,所以break语句终止的是select而不是外层的for循环。如果希望终止for循环,可以采用break 标签,代码如下。


loop:
for {
    select {
        case <-ch:
            // Do something
        case <-ctx.Done():
            break loop
        }
}

NOTE 我们可以使用continue 标签,程序会跳转到标签的位置执行下一轮操作。例如上述代码,对于上述代码执行 continue loop,会重新执行for循环。

思考总结

当我们在for循环中使用swith、select语句并使用break终止操作时要特别小心,牢记一点,不接标签(label)的break语句会跳出最内层的switch、select或for代码块。此外,break label在Go语言中是一种惯用的使用方法,明确跳出到某个位置。

转载自:https://juejin.cn/post/7349551248515956773
评论
请登录