正确的关闭渠道方式

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《正确的关闭渠道方式》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

我想要一堆 goroutine 来从很多服务器获取一些信息。我正在简化下面的代码，使其更具可读性。它似乎工作得很好，但在所有任务完成后它会出现恐慌，因为我从未关闭频道。问题是我不确定在哪里我应该关闭它。

我需要你的帮助：

• 告诉我应该在代码中的哪个位置关闭通道。
• 告诉我这段代码的整体逻辑是否符合惯用方式。

我的代码

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


        for i:= 0; i < 10 ; i++ {
                go func(c chan <- string,t int){
                        time.sleep( time.duration(rand.intn(3000)) * time.millisecond )
                        c <- strconv.itoa(t) + " : done " + strconv.itoa(rand.intn(3000))
                }(ch,i)
        }
        for val := range ch {
                fmt.println(val)

        }
}

输出

$ go run test_channels.go
0 : Done 1694
6 : Done 511
3 : Done 162
2 : Done 89
8 : Done 2728
5 : Done 1274
1 : Done 2211
9 : Done 1445
4 : Done 2237
7 : Done 1106
fatal error: all goroutines are asleep - deadlock!

goroutine 1 [chan receive]:
main.main()
        /home/matias/projects/src/github.com/matias/test/test_channels.go:22 +0x138
exit status 2

解决方案

当通道上只有一个发送者时，该发送者通常负责在必要时关闭通道。请记住，虽然清理不需要关闭通道，但仅当您需要发出通道已关闭信号时才需要关闭通道。

当有多个发件人时，您需要协调所有发件人完成，您可以使用 sync.waitGroup 来完成

ch := make(chan string)
var wg sync.WaitGroup

for i := 0; i < 10; i++ {
    wg.Add(1)
    go func(c chan<- string, t int) {
        defer wg.Done()
        time.Sleep(time.Duration(rand.Intn(3000)) * time.Millisecond)
        c <- strconv.Itoa(t) + " : Done " + strconv.Itoa(rand.Intn(3000))
    }(ch, i)
}

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

https://play.golang.org/p/ViOtMHbi43C

终于介绍完啦！小伙伴们，这篇关于《正确的关闭渠道方式》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！