正确关闭频道的方法

时间:2018-12-18 18:24:04

标签: go goroutine

我想要一堆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

1 个答案:

答案 0 :(得分:5)

当一个频道上有一个发送者时,通常在必要时该发送者将负责关闭该频道。请记住,尽管清理并不需要关闭通道,但这仅在您需要表明该通道已关闭的情况下。

有多个发件人时,您需要与所有发件人进行协调,您可以使用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