for循环中的goroutines问题

Question

我正在尝试在Exercism 上解决这个问题：

编写一个程序，使用并行计算计算文本中字母的频率。

基本上，我有一个FreqMap 类型：

type FreqMap map[rune]int

还有一个Frequency 函数：

func Frequency(s string) FreqMap {
    m := make(FreqMap)
    for _, v := range s {
        m[v]++
    }
    return m
}

Exercism 提供了一个使用递归实现并发版本的示例，但我想使用for 循环实现我自己的版本。我想出了以下解决方案，但不起作用：

func ConcurrentFrequency(l []string) FreqMap {
    c := make(chan FreqMap)
    for i := 0; i < len(l); i++ {
        go func(i int) {
            c <- Frequency(l[i])
        }(i)
    }
    return <- c
}

这似乎只在 1 次迭代后返回，c 似乎只包含 1 个 goroutine 的结果；如果我添加 sync.WaitGroup，我会得到相同的结果。

你能解释一下我在这里缺少什么吗？

提前感谢您的帮助！

Answer 1

您的代码似乎只进行了一次迭代，因为ConcurrentFrequency 从通道返回第一个值，仅此而已。我想你想要这样的东西：

func ConcurrentFrequency(l []string) chan FreqMap {
    c := make(chan FreqMap)
    go func() {
        var wg sync.WaitGroup
        wg.Add(len(l))
        for _, s := range l {
            go func(s string) {
                defer wg.Done()
                c <- Frequency(s)
            }(s)
        }
        wg.Wait()
        close(c)
    }()
    return c
}

现在它返回地图的通道，这些你可能想组合成一个地图：

func main() {
    m := make(FreqMap)
    for v := range ConcurrentFrequency([]string{"foo", "bar","zoo"}) {
        for k, v := range v {
            m[k] += v
        }
    }
    fmt.Println(m)
}

---

更长的解释，不适合评论：

在for _, s := range l循环中，所有goroutines都写入同一个通道，但是由于该通道没有缓冲，一旦第一个值写入其中，它就是“满的”，这意味着不能写入其他值进去。所以循环中只有一个 goroutine 可以完成，wg.Done 只被调用一次。因此，如果源数组有多个字符串，其余的 gorutine 将无法完成，直到某些东西开始消耗通道中的值。但在您的版本中，它会卡在wg.Wait，因为并非所有goroutines 都已完成，因此ConcurrentFrequency 无法将通道返回给消费者。 以我编写ConcurrentFrequency 的方式，可以将通道返回给消费者，这（从通道读取）使其他Frequency(s) 调用能够写入通道。