深入理解 Go 语言中的 select 语句

2024-01-24 09:28:48

Go 语言中的 select 语句

select 语句是一种并发原语，用于在多个通道上进行非阻塞式通信。它允许 goroutine 在多个通道之间进行选择，并在有数据可读或可写时执行相应的代码块。select 语句的语法如下：

select {
    case <case1>:
        // case1 的代码块
    case <case2>:
        // case2 的代码块
    ...
    default:
        // 默认代码块
}

每个 case 语句都与一个通道相关联，如果与该通道相关联的条件为真，则会执行相应的代码块。default 代码块在其他 case 语句都不满足条件时执行。

for 循环中的 select

当 select 语句放在 for 循环中时，它会在循环的每次迭代中执行。如果通道未关闭，则 select 语句将继续执行，直到有数据可读或可写。但是，如果与 case 语句相关联的通道已关闭，则每次都会执行到该 case。这是因为 select 语句会不断尝试从已关闭的通道中读取或写入数据，直到发生错误。

以下示例演示了在 for 循环中使用 select 语句时通道关闭的情况：

package main

import (
    "fmt"
    "time"
)

func main() {
    c := make(chan int, 0)

    go func() {
        time.Sleep(1 * time.Second)
        c <- 10
        close(c)
    }()

    for {
        select {
        case v := <-c:
            fmt.Println(v)
        }
    }
}

在这个示例中，我们在一个协程中启动了通道写入操作，然后关闭了通道。在主 goroutine 中，我们使用 for 循环中的 select 语句从通道中读取数据。由于通道在写入值后立即关闭，因此 select 语句将不断执行到从已关闭通道读取数据的 case，导致死循环。

只有一个 case 的 select

如果 select 语句中只有一个 case，而这个 case 被关闭了，则会出现死循环。这是因为 select 语句会不断尝试从已关闭的通道中读取或写入数据，直到发生错误。

以下示例演示了只有一个 case 的 select 语句中的死循环：

package main

import (
    "fmt"
    "time"
)

func main() {
    c := make(chan int, 0)

    close(c)

    for {
        select {
        case <-c:
            fmt.Println("This line will never be reached.")
        }
    }
}