Go 示例: 关闭通道

关闭通道表示不会再向其发送任何值。这对于向通道的接收方传达完成信息很有用。

 
package main
 
import "fmt"

在本例中，我们将使用一个名为 jobs 的通道，用于将工作从 main() goroutine 传递到 worker goroutine。当我们没有更多工作要交给 worker 时，我们将 close jobs 通道。

 
func main() {
    jobs := make(chan int, 5)
    done := make(chan bool)

以下是 worker goroutine。它重复地从 jobs 中接收，使用 j, more := <-jobs。在这种特殊的 2 值接收形式中，如果 jobs 已被 close 并且通道中的所有值都已经接收，则 more 值将为 false。我们使用它来在完成所有工作时通知 done。

 
    go func() {
        for {
            j, more := <-jobs
            if more {
                fmt.Println("received job", j)
            } else {
                fmt.Println("received all jobs")
                done <- true
                return
            }
        }
    }()

这将通过 jobs 通道向 worker 发送 3 个工作，然后关闭它。

 
    for j := 1; j <= 3; j++ {
        jobs <- j
        fmt.Println("sent job", j)
    }
    close(jobs)
    fmt.Println("sent all jobs")

我们使用之前看到的 同步 方法来等待 worker。

 
    <-done

从已关闭的通道读取会立即成功，并返回底层类型的零值。可选的第二个返回值为 true，如果接收到的值是由对通道的成功发送操作传递的，或者为 false，如果它是由于通道已关闭且为空而生成的零值。

 
    _, ok := <-jobs
    fmt.Println("received more jobs:", ok)
}
 
$ go run closing-channels.go 
sent job 1
received job 1
sent job 2
received job 2
sent job 3
received job 3
sent all jobs
received all jobs
received more jobs: false

关闭通道的概念自然地引出了我们的下一个示例：通道上的 range。

下一个示例：通道上的 Range.

作者：Mark McGranaghan 和 Eli Bendersky | 源代码 | 许可证