控制 Go 协程的生命周期可以通过以下方式:创建协程:使用 go 关键字启动新任务。终止协程:等待所有协程完成,使用 sync.WaitGroup。使用通道关闭信号。使用上下文 context.Context。
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 协程的生命周期?的详细内容。更多信息请关注PHP中文网其他相关文章!