首页 > 文章列表 > 如何控制 Golang 协程的生命周期?

如何控制 Golang 协程的生命周期?

协程 生命周期
111 2024-05-23

控制 Go 协程的生命周期可以通过以下方式:创建协程:使用 go 关键字启动新任务。终止协程:等待所有协程完成,使用 sync.WaitGroup。使用通道关闭信号。使用上下文 context.Context。

如何控制 Golang 协程的生命周期?

如何控制 Go 协程的生命周期?

Go 语言中,协程(又称 Goroutine)是轻量的并发执行体,用于执行特定的任务,而不会阻塞主线程。管理协程的生命周期对于编写健壮和可维护的并行代码至关重要。

创建协程

使用 go 关键字创建协程。它在后台启动一个新的任务,使主线程可以继续执行。

go func() {
    // 执行任务
}

终止协程

无法直接终止协程,但有以下方法可以间接达到此目的:

等待所有协程完成

使用 sync.WaitGroup 同步任务完成:

import "sync"

var wg sync.WaitGroup

func main() {
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go func(i int) {
            // 执行任务
            wg.Done()
        }(i)
    }
    wg.Wait()
}

使用通道关闭信号

使用通道向协程发送"关闭"信号:

func main() {
    done := make(chan struct{})
    for i := 0; i < 5; i++ {
        go func(i int) {
            for {
                select {
                case <-done:
                    return // 协程停止
                default:
                    // 执行任务
                }
            }
        }(i)
    }
    close(done) // 向所有协程发送"关闭"信号
}

使用上下文

使用 context.Context 管理协程的执行,当上下文被取消时,协程也会终止:

import "context"

func main() {
    ctx, cancel := context.WithCancel(context.Background())
    for i := 0; i < 5; i++ {
        go func(i int) {
            for {
                select {
                case <-ctx.Done():
                    return // 协程停止
                default:
                    // 执行任务
                }
            }
        }(i)
    }
    cancel() // 取消上下文
}

实战案例

以下是一个使用通道关闭信号的实战案例:

package main

import "fmt"
import "time"

func main() {
    // 使用信道告诉协程何时退出
    stop := make(chan struct{})

    // 创建 5 个协程
    for i := 0; i < 5; i++ {
        go func(i int) {
            for {
                // 检查是否已经收到退出信号
                select {
                case <-stop:
                    fmt.Printf("协程 %d 已退出n", i)
                    return
                default:
                    fmt.Printf("协程 %d 正在运行n", i)
                    time.Sleep(time.Second)
                }
            }
        }(i)
    }

    // 运行 5 秒,然后发送退出信号
    time.Sleep(time.Second * 5)
    close(stop)

    // 等待所有协程退出
    time.Sleep(time.Second * 1)
}