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

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

如何控制 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)
}

登录后复制

以上就是如何控制 Golang 协程的生命周期？的详细内容，更多请关注叮当号网其它相关文章！