如何优雅地关闭长度未知的 Go Channel？

在不知道通道长度的情况下确定何时关闭通道

在 Go 中使用通道时，确定关闭通道的适当时间至关重要。当通道长度未知时，这会带来挑战。

考虑以下场景：

package main

import (
    "fmt"
    "time"
)

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

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

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

在此示例中，一个 Goroutine 向通道发送 100 个值，目的是关闭它一旦所有值都已发送。然而，这种做法引起了人们的担忧。具体来说：

• 并发关闭：多个 goroutine 可能会尝试同时关闭通道，从而导致恐慌。
• 意外关闭：主 goroutine 可能会在收到所有值之前终止，从而导致某些值被

使用 WaitGroup 实现优雅关闭

为了解决这些问题，可以使用sync.WaitGroup 来同步通道的关闭和发送 goroutine 的完成。

package main

import (
    "fmt"
    "sync"
    "time"
)

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

    wg.Add(1) // Increment counter for sender goroutine
    go func() {
        defer wg.Done() // Decrement counter when goroutine completes
        for i := 0; i < 100; i++ {
            ch <- i
        }
        close(ch)
    }()

    go func() {
        wg.Wait() // Wait until the sender goroutine completes
        close(ch) // Close the channel after all values have been sent
    }()

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

实施详细信息

• WaitGroup：sync.WaitGroup 允许我们跟踪正在运行的 goroutine 的数量。它确保主 Goroutine 在所有发送者 Goroutine 完成之前不会终止。
• 单独关闭 Goroutine： 通道的关闭在专用 Goroutine 中执行。这保证了只有在通道中等待的所有数据都被读取后，主 Goroutine 才会退出。
• 多个发送者：这种方法也适用于多个 Goroutine 发送到同一通道的场景。 sync.WaitGroup 确保仅在所有发送者 goroutine 完成发送值后关闭通道。

以上是如何优雅地关闭长度未知的 Go Channel？的详细内容。更多信息请关注PHP中文网其他相关文章！