如何优雅地等待 Go 例程完成？

如何优雅地等待 Go 例程完成

在 Go 中，并发是通过 goroutine 实现的，goroutines 是与以下进程同时运行的轻量级线程主要例程。为了确保程序不会在所有 goroutine 完成其任务之前退出，正确等待它们完成至关重要。

使用 Channel 来发出完成信号

等待 goroutine 的一种常见方法是使用布尔通道。当 goroutine 完成其任务时，它会向通道发送一个 true 值，主例程通过从通道接收该值来等待该值。

func do_stuff(done chan bool) {
    fmt.Println("Doing stuff")
    done <- true
}

func main() {
    fmt.Println("Main")
    done := make(chan bool)
    go do_stuff(done)
    <-done
}

为什么

如果我们取消最后一行注释会发生什么？

如果我们取消最后一行的注释，我们会收到死锁错误，因为通道是空的并且没有其他 goroutine 向其发送值。因此，主例程将无限等待接收值，从而导致死锁。

使用同步包的替代方法

对于更复杂的并发场景，sync包提供了方便的同步机制。例如，如果您需要并行化一系列长时间运行的函数，以下代码片段演示了如何使用sync.WaitGroup类型：

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go func() {
            longOp()
            wg.Done()
        }()
    }

    // will wait until wg.Done is called 10 times
    // since we made wg.Add(1) call 10 times
    wg.Wait()
}

func longOp() {
    time.Sleep(time.Second * 2)
    fmt.Println("long op done")
}

在此示例中，WaitGroup确保主例程在所有 goroutine 调用 wg.Done 之前不会退出。

以上是如何优雅地等待 Go 例程完成？的详细内容。更多信息请关注PHP中文网其他相关文章！