Go：使用一个通道接收多个 goroutine 的结果

小伙伴们有没有觉得学习Golang很有意思？有意思就对了！今天就给大家带来《Go：使用一个通道接收多个 goroutine 的结果》，以下内容将会涉及到，若是在学习中对其中部分知识点有疑问，或许看了本文就能帮到你！

问题内容

我有一个包含数千条记录的 csv 文件。 我处理 goroutine 中的每条记录，并且希望将处理记录的所有结果收集到一个切片中，以便将它们写在另一个文件中。 我的代码的简化版本是：

var wg sync.waitgroup

func main() {
    c := make(chan string, 1)
    csvfile, err := os.open("data.csv")
    reader := csv.newreader(bufio.newreader(csvfile))
    //...
    for {
        line, err := reader.read()
        // ...
        wg.add(1)
        go processrecord(line, c)
    }
    //...
    results := make([]string, 0)
    for r := range c { // error here: fatal error: all goroutines are asleep - deadlock!
        results = append(results , r)
    }
    // write results to file
    wg.wait()
    close(c)
}

func processrecord(data []string, c chan string) {
    defer wg.done()
    // ...
    c <- *result
}

在 for 循环中，我得到了所有结果，但在得到最后一个结果后循环继续，我收到此错误：

fatal error: all goroutines are asleep - deadlock!

这段代码有什么问题？我在 so 上尝试了不同的 qa，但没有成功。

正确答案

只有当通道关闭时，for 循环才会终止。您将在 for 循环终止后关闭通道，因此出现死锁。

您可以通过将 for 循环放入 goroutine 中来修复：

results := make([]string, 0)
done:=make(chan struct{})
go func() {
    for r := range c {
        results = append(results , r)
    }
    close(done)
}()
wg.wait()
close(c)
<-done
// read results

for r := range c { // ERROR HERE: fatal error: all goroutines are asleep - deadlock!
    mutex.Lock()
    results = append(results , r)
    mutex.Unlock()
}

goroutines 正在睡觉 - 僵局！因为是同时写的。使用mutex可以处理这个异常

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注编程网公众号，一起学习编程~