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

大家好，我们又见面了啊~本文《简单的通道并行示例，导致死锁》的内容中将会涉及到等等。如果你正在学习Golang相关知识，欢迎关注我，以后会给大家带来更多Golang相关文章，希望我们能一起进步！下面就开始本文的正式内容~

我正在尝试使用通道在 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 10
10 20
goroutine started with  0 10
goroutine started with  10 20
5
6
7
8
9
10
11
12
13
14
fatal error: all goroutines are asleep - deadlock!

解决方案

问题是通道上的 range 不断尝试从通道接收数据，直到通道关闭，但发送完成后您没有关闭 通道。</p> <p>在代码中的 wg.Done 之前添加 close(out) 将修复此问题。