所有的 goroutine 都在休眠 - 死锁

Question

我正在使用 Go 构建工作系统的骨架，但我收到“致命错误：所有 goroutine 都处于睡眠状态 - 死锁！”。

我使用两个渠道进行协调，一个用于创建工作，第二个用于发送结果。创建作业后，我关闭输入通道。

我的问题是如何关闭输出通道以便程序可以正确退出。 代码是：

package main

import (
    "bufio"
    "flag"
    "fmt"
    "log"
    "math/rand"
    "os"
    "time"
)

type Work struct {
    id int
    ts time.Duration
}

const (
    NumWorkers = 5000
    NumJobs    = 100000
)

func worker(in <-chan *Work, out chan<- *Work) {
    for w := range in {
        st := time.Now()
        time.Sleep(time.Duration(rand.Int63n(int64(200 * time.Millisecond))))
        w.ts = time.Since(st)
        out <- w
    }
}

func main() {
    wait := flag.Bool("w", false, "wait for <enter> before starting")
    flag.Parse()

    if *wait {
        fmt.Printf("I'm <%d>, press <enter> to continue", os.Getpid())
        reader := bufio.NewReader(os.Stdin)
        reader.ReadString('\n')
    }

    Run()
}

func Run() {
    in, out := make(chan *Work, 100), make(chan *Work, 100)
    for i := 0; i < NumWorkers; i++ {
        go worker(in, out)
    }
    go createJobs(in)
    receiveResults(out)
}

func createJobs(queue chan<- *Work) {
    for i := 0; i < NumJobs; i++ {
        work := &Work{i, 0}
        queue <- work
    }
    close(queue)
}

func receiveResults(completed <-chan *Work) {
    for w := range completed {
        log.Printf("job %d completed in %s", w.id, w.ts)
    }
}

任何帮助表示赞赏:)

Answer 1

我错过了关于您知道原始答案中死锁原因的部分。

• 您提到了 WaitGroup，它基本上只是一个信号量
• 您可以使用另一个“控制”通道，工作人员在完成工作后会使用该通道

-

func worker(ctrl chan<- bool, in <-chan *Work, out chan<- *Work) {
    for w := range in {
        st := time.Now()
        time.Sleep(time.Duration(rand.Int63n(int64(200 * time.Millisecond))))
        w.ts = time.Since(st)
        out <- w
    }
    ctrl <- true
}

func control(ctrl <-chan bool, numWorkers int, out chan<- *Work) {
    for i=0; i<numWorkers; i++ {
        <-ctrl
    }
    close(out)
}

原答案：

您在完成时执行range：

for w := range completed {
    log.Printf("job %d completed in %s", w.id, w.ts)
}

但该频道从未关闭