带通道的简单并行示例，导致死锁

Question

我正在尝试使用通道在 golang 中实现一个简单的并行化示例。该代码尝试实现并行化的映射函数。也使通道缓冲以缓解对通道阻塞性质的限制。但是代码仍然会导致死锁。func pmmap(inp []int, f func(int) int, p int) {    var wg sync.WaitGroup    var output []chan int    tot := len(inp)    out := make(chan int, tot)    slice := tot / p    for i := 0; i < p; i++ {        temp := make(chan int, slice)        output = append(output, temp)        start_ind := slice * i        end_ind := start_ind + slice        fmt.Println(start_ind, end_ind)        wg.Add(1)        go func(si, ei int, out chan int, wg *sync.WaitGroup) {            fmt.Println("goroutine started with ", si, ei)            for ind := si; ind < ei; ind++ {                out <- f(inp[ind])            }            wg.Done()        }(start_ind, end_ind, output[i], &wg)    }    wg.Add(1)    go func(wg *sync.WaitGroup) {        for i := 0; i < p; i++ {            for val := range output[i] {                out <- val            }            close(output[i])        }        wg.Done()    }(&wg)    wg.Add(1)    go func(wg *sync.WaitGroup) {        for i := range out {            fmt.Println(i)        }        wg.Done()    }(&wg)    time.Sleep(time.Second * 6)    wg.Wait()    close(out)}func add4(i int) int {    return i + 4}func main() {    temp := []int{}    for i := 1; i <= 20; i++ {        temp = append(temp, i)    }    pmmap(temp, add4, 2)}从上面代码的输出中，我得到死锁是因为通道输出 [1] 永远不会被读取。但我不知道为什么0 1010 20goroutine started with  0 10goroutine started with  10 20567891011121314fatal error: all goroutines are asleep - deadlock!

Answer 1

问题是通过一个通道不断尝试从通道接收，直到通道关闭，但完成发送后range您没有通道。close

在您的代码中添加close(out)之前wg.Done将修复它。

游乐场： https: //play.golang.org/p/NbKTx6Lke7X

编辑：修复了关闭封闭频道的错误。