控制 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 协程的生命周期?的详细内容,更多请关注编程网其它相关文章!