为什么这个 goroutine 不调用 wg.Done()？

Question

假设在任意时刻，registerChan 上最多有两个元素（工作地址）。然后由于某种原因，下面的代码没有在最后两个 goroutine 中调用 wg.Done() 。func schedule(jobName string, mapFiles []string, nReduce int, phase jobPhase, registerChan chan string) {    var ntasks int    var nOther int // number of inputs (for reduce) or outputs (for map)    switch phase {    case mapPhase:        ntasks = len(mapFiles)        nOther = nReduce    case reducePhase:        ntasks = nReduce        nOther = len(mapFiles)    }    fmt.Printf("Schedule: %v %v tasks (%d I/Os)\n", ntasks, phase, nOther)    const rpcname = "Worker.DoTask"    var wg sync.WaitGroup    for taskNumber := 0; taskNumber < ntasks; taskNumber++ {        file := mapFiles[taskNumber%len(mapFiles)]        taskArgs := DoTaskArgs{jobName, file, phase, taskNumber, nOther}        wg.Add(1)        go func(taskArgs DoTaskArgs) {            workerAddr := <-registerChan            print("hello\n")            // _ = call(workerAddr, rpcname, taskArgs, nil)            registerChan <- workerAddr            wg.Done()        }(taskArgs)    }    wg.Wait()    fmt.Printf("Schedule: %v done\n", phase)}如果我放在wg.Done()前面，registerChan <- workerAddr它就可以正常工作，但我不知道为什么。我也尝试过推迟 wg.Done() 但这似乎不起作用，尽管我期望它能起作用。我认为我对 go 例程和通道的工作方式存在一些误解，这导致了我的困惑。

Answer 1

因为它停在这里：

workerAddr := <-registerChan

对于缓冲通道：
要使其workerAddr := <-registerChan正常工作：通道registerChan 必须有一个值；否则，代码将在此停止等待通道。

---

我设法以这种方式运行你的代码（试试这个）：

package main

import (

    "fmt"

    "sync"

)

func main() {

    registerChan := make(chan int, 1)

    for i := 1; i <= 10; i++ {

        wg.Add(1)

        go fn(i, registerChan)

    }

    registerChan <- 0 // seed

    wg.Wait()

    fmt.Println(<-registerChan)

}

func fn(taskArgs int, registerChan chan int) {

    workerAddr := <-registerChan

    workerAddr += taskArgs

    registerChan <- workerAddr

    wg.Done()

}

var wg sync.WaitGroup

输出：

55

说明：

此代码使用通道和 10 个 goroutine 加上一个主 goroutine 将 1 与 10 相加。

我希望这有帮助。

Answer 2

当您运行此语句时registerChan <- workerAddr，如果通道容量已满，则无法添加它，并且它将阻塞。如果你有一个池，比如 10 个workerAddr，你可以在调用之前将它们全部添加到容量为 10 的缓冲通道中schedule。不要在调用后添加，以保证如果您从通道中获取值，之后有空间再次添加它。defer在 goroutine 的开头使用是很好的。