
在Golang中实现多任务同步等待,最常用且推荐的方式是使用 sync.WaitGroup。它能有效协调多个goroutine的执行,确保主流程等待所有子任务完成后再继续。
使用 sync.WaitGroup 实现同步等待
WaitGroup 是 Go 标准库中用于等待一组并发任务完成的同步原语。它通过计数器控制,当计数器归零时,阻塞的 Wait 调用才会返回。
基本使用步骤:
创建一个 sync.WaitGroup 变量每启动一个goroutine前调用 Add(1) 增加计数在每个goroutine结束时调用 Done() 减少计数在主协程中调用 Wait() 阻塞,直到计数归零
示例代码:
立即学习“go语言免费学习笔记(深入)”;
func main() {
var wg sync.WaitGroup
for i := 0; i wg.Add(1)
go func(id int) {
defer wg.Done()
fmt.Printf(“任务 %d 开始n”, id)
time.Sleep(time.Second)
fmt.Printf(“任务 %d 完成n”, id)
} (i)
}
wg.Wait()
fmt.Println(“所有任务已完成”)
}
结合 context 控制超时和取消
在实际开发中,任务可能因异常长时间运行而需要限制等待时间。此时可结合 context 包实现超时控制。
通过 context.WithTimeout 或 context.WithCancel 创建上下文,在 WaitGroup 等待的同时监听 context 的关闭信号。
示例:带超时的等待
func main() {
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
var wg sync.WaitGroup
for i := 0; i wg.Add(1)
go func(id int) {
defer wg.Done()
select {
case fmt.Printf(“任务 %d 成功完成n”, id)
case fmt.Printf(“任务 %d 被取消n”, id)
}
} (i)
}
ch := make(chan struct{})
go func() {
wg.Wait()
close(ch)
}()
select {
case fmt.Println(“全部任务正常完成”)
case fmt.Println(“等待超时,部分任务未完成”)
}
}
常见注意事项
使用 WaitGroup 时需注意以下几点,避免出现死锁或 panic:
确保每次 Add(n) 调用都对应 n 次 Done(),否则 Wait 可能永不返回不要在 goroutine 外部调用 Done(),应由每个任务自己负责通知完成避免在 Add 前启动 goroutine,防止竞争条件通常将 defer wg.Done() 放在 goroutine 开头,确保无论函数如何退出都能触发
基本上就这些。sync.WaitGroup 简单高效,适合大多数多任务同步场景。配合 context 使用,还能增强程序的健壮性和可控性。不复杂但容易忽略细节。
以上就是如何在Golang中实现多任务同步等待的详细内容,更多请关注创想鸟其它相关文章!
版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容, 请发送邮件至 chuangxiangniao@163.com 举报,一经查实,本站将立刻删除。
发布者:程序猿,转转请注明出处:https://www.chuangxiangniao.com/p/1412663.html
微信扫一扫
支付宝扫一扫