Go：使用一个通道从多个 goroutines 接收结果

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-loop只会在频道关闭后终止。您在 for-loop 终止后关闭通道，因此出现死锁。

您可以通过将 for-loop 放入 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 可以处理这个异常

Go：使用一个通道从多个 goroutines 接收结果

Go: using one channels to receive results from multiple goroutines

go

goroutine