【Go】 探索 Go 语言的内建函数 close

---

山水间歌声回荡
回荡思念的滚烫
去年的家书两行
读来又热了眼眶
云水边静沐暖阳
烟波里久违的故乡
别来无恙
你在心上
                     🎵 张靓颖/张杰《燕归巢》

---

在 Go 语言中，close 是一个用于关闭通道（channel）的内建函数。通道是 Go 并发编程中的重要组成部分，它们用于在不同的 goroutine 之间传递数据。close 函数可以让我们显式地关闭一个通道，从而通知接收方所有的发送操作已经完成。本文将详细介绍 close 函数的使用场景和示例。

close 函数的基本语法

close 函数的基本语法如下：

close(channel)

• channel：要关闭的通道。

关闭通道的基本使用

当我们使用 close 函数关闭一个通道时，表示不会再向该通道发送任何数据。接收方可以通过检查通道是否关闭来确定是否结束接收操作。

基本示例

package main

import "fmt"

func main() {
    ch := make(chan int)

    go func() {
        for i := 0; i < 5; i++ {
            ch <- i
        }
        close(ch)
    }()

    for v := range ch {
        fmt.Println(v)
    }
}

在这个例子中，关闭通道后，range 循环能够检测到通道已经关闭并自动退出循环。

检查通道是否关闭

我们可以通过接收操作的第二个返回值来检查通道是否关闭。

package main

import "fmt"

func main() {
    ch := make(chan int)
    go func() {
        ch <- 1
        close(ch)
    }()

    v, ok := <-ch
    if ok {
        fmt.Println("接收到数据:", v)
    } else {
        fmt.Println("通道已关闭")
    }
}

在这个例子中，ok 的值为 false 表示通道已关闭。

多个发送方与关闭通道

只有发送方需要关闭通道，并且应当确保只有一个发送方执行 close 操作，否则会导致 panic。

package main

import (
    "fmt"
    "sync"
)

func main() {
    ch := make(chan int)
    var wg sync.WaitGroup

    for i := 0; i < 3; i++ {
        wg.Add(1)
        go func(id int) {
            defer wg.Done()
            ch <- id
        }(i)
    }

    go func() {
        wg.Wait()
        close(ch)
    }()

    for v := range ch {
        fmt.Println(v)
    }
}

在这个例子中，确保了所有 goroutine 完成发送后，由一个独立的 goroutine 关闭通道。

使用 close 函数的注意事项

• 只能关闭发送方向的通道：只有发送方才能关闭通道，接收方不能关闭通道。
• 关闭后的通道行为：向关闭的通道发送数据会导致 panic，从关闭的通道接收数据不会阻塞，会立即返回通道元素类型的零值。
• 通道的多次关闭：多次关闭同一个通道会导致 panic，因此必须确保只有一个发送方进行关闭操作。

示例代码

以下是一个综合示例，展示了 close 函数在不同情况下的使用：

package main

import "fmt"

func main() {
    ch := make(chan int, 3)

    go func() {
        for i := 0; i < 3; i++ {
            ch <- i
        }
        close(ch)
    }()

    for {
        v, ok := <-ch
        if !ok {
            fmt.Println("通道已关闭")
            break
        }
        fmt.Println("接收到数据:", v)
    }
}

总结

close 是 Go 语言中的一个重要函数，用于显式关闭通道，从而通知接收方发送操作已完成。理解和正确使用 close 函数，对于编写健壮的并发程序至关重要。希望这篇博客能帮助你更好地理解和使用 Go 语言中的 close 函数，让你的并发编程更加顺畅。