如何在 Go 中管理 Goroutine 的生命周期？

Go 中管理 Goroutine 生命周期的核心方法如下：使用 context.Context：通过取消信号和截止日期控制 Goroutine 生命周期。使用 sync.WaitGroup：等待 Goroutine 完成任务，以便主 Goroutine 继续执行。使用通道：通过信号通信协调多个 Goroutine，等待所有 Goroutine 完成后进行后续处理。

在 Go 中管理 Goroutine 的生命周期

Goroutine 是 Go 中支持并发性的轻量级线程。它们的管理至关重要，可以防止资源泄漏和程序崩溃。本文将探讨管理 Goroutine 生命周期的各种方法，包括：

1. 使用 context.Context

context.Context 提供了一种在 Goroutine 内传播取消和截止日期信号的机制。要使用它来管理 Goroutine 的生命周期，可以使用以下步骤：

package main

import (
    "context"
    "fmt"
    "time"
)

func main() {
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()

    go func() {
        // Goroutine 的代码
    }()

    // 等待 Goroutine 完成，或超时。
    select {
    case <-ctx.Done():
        fmt.Println("Goroutine 已取消或超时。")
    }
}

2. 使用 sync.WaitGroup

sync.WaitGroup 允许 Goroutine 相互等待，直到它们完成各自的任务。使用它来管理 Goroutine 的生命周期，可以在主 Goroutine 中等待所有子 Goroutine 都完成。

package main

import (
    "fmt"
    "runtime"
    "sync"
)

func main() {
    wg := sync.WaitGroup{}

    // 创建 5 个子 Goroutine
    for i := 0; i < 5; i++ {
        wg.Add(1) // 递增WaitGroup计数，表明正在等待另一个Goroutine完成

        go func(i int) {
            fmt.Printf("Goroutine %d 结束。\n", i)
            wg.Done() // 递减WaitGroup计数，表明Goroutine已完成
        }(i)
    }

    // 等待所有子 Goroutine 完成
    wg.Wait()

    fmt.Println("所有子 Goroutine 都已完成。")
}

3. 使用通道

通道可以在 Goroutine 之间进行通信，也可以用于管理它们的生命周期。通过将一个 Goroutine 标记为已完成的信号发送到一个通道中，主 Goroutine 可以等待所有子 Goroutine 都完成。

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    signals := make(chan struct{})

    // 创建 5 个子 Goroutine
    for i := 0; i < 5; i++ {
        wg.Add(1)

        go func(i int) {
            defer wg.Done() // Goroutine退出时递减WaitGroup计数

            // 等待其他Goroutine完成
            <-signals
            fmt.Printf("Goroutine %d 已完成。\n", i)
        }(i)
    }

    // 等待所有子 Goroutine完成
    wg.Wait()

    // 发送信号标记所有子Goroutine都已完成
    close(signals)

    fmt.Println("所有子 Goroutine 都已完成。")
}