使用等待组的 Golang 工作者

Question

我是 Golang 的新手，并试图了解 Golang 中的 WaitGroups 和并发是如何工作的。在此示例中，创建了 5 个工作人员，并使用一个通道将工作传递给每个工作人员。工作人员被要求休眠 1 秒以模拟繁重的计算。一切顺利，但程序并没有优雅地退出。而是打印此错误消息。请帮助理解为什么会发生这种情况。

fatal error: all goroutines are asleep - deadlock!

这是代码，

import (

    "fmt"

    "sync"

    "time"

)

func worker(wg *sync.WaitGroup, messageChannel <-chan string) {

    defer wg.Done()

    for i := range messageChannel {

        time.Sleep(time.Second)

        fmt.Println("done processing - ", i)

    }

}

func stop(wg *sync.WaitGroup) {

    fmt.Println("waiting on the main thread")

    wg.Wait()

}

func main() {

    wg := new(sync.WaitGroup)

    messageChannel := make(chan string, 50)

    // create workers

    for i := 0; i < 5; i++ {

        wg.Add(1)

        go worker(wg, messageChannel)

    }

    // input some messages

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

        messageChannel <- fmt.Sprint(i)

    }

    stop(wg)

    close(messageChannel)

}

提前致谢！

Answer 1

为了扩展@Peter 的评论，这里是您编写的代码的执行流程：初始化后，你启动你的workergoroutines。每个workergoroutine 都会超过messageChannel，延迟 1 秒会打印出一条消息。接下来，您messageChannel通过 for 循环插入一些消息。每个可用worker的 goroutine 都会收到一条消息，直到所有消息都被处理并打印出来。之后，workergoroutines 正在等待来自messageChannel.在用于在中插入消息的 for 循环messageChannel完成后，您执行该stop函数，该函数阻塞wg.Wait()并等待所有wg.Done()调用在所有workergoroutine 中执行。然而，由于messageChannel没有关闭，没有一个workergoroutine 可以完成执行，也没有一个wg.Done()调用被执行。workergoroutines 被卡住是因为它messageChannel永远不会关闭，maingoroutines 因为函数wg.Wait()内部的调用而被卡住stop，你最终会陷入一个所有 goroutines 都处于休眠状态的死锁。根据建议，您只需要交换地点stop和close电话//rest of the codeclose(messageChannel)stop(wg)这样，当所有消息都插入到 中时messageChannel，您调用close(messageChannel)然后stop(wg)，这会阻塞wg.Wait调用。该close(messageChannel)调用确保，一旦从 中读取所有消息messageChannel，goroutine内部的for-range循环将退出，并且所有调用都将被执行。一旦发生这种情况，将解除阻塞，程序将正确完成执行。messageChannelworkerdefer wg.Done()wg.Wait()