Go:使用一个通道接收来自多个goroutine的结果

Question

我有一个包含数千条记录的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，但没有成功。

Answer 1

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

Answer 2

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

goroutines睡着了-死锁！因为它是同时写的。使用Mutex可以处理此异常