关闭频道的正确方法

Question

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

我需要你的帮助:

告诉我应该在代码中的什么地方关闭频道。
告诉我这段代码的整体逻辑是否符合惯用语。

我的代码

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

Answer 1

当频道上只有一个发件人时，该发件人通常会在必要时负责关闭频道。请记住，虽然清理不需要关闭频道，但只有在您需要发出频道已关闭的信号时才会这样做。

当有多个发件人时，您需要协调所有发件人完成，您可以使用 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

关闭频道的正确方法

Proper way to close a channel

go

goroutine

我的代码

输出