如何在Golang中实现多任务同步等待

如何在golang中实现多任务同步等待

在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

(0)
打赏 微信扫一扫 微信扫一扫 支付宝扫一扫 支付宝扫一扫
上一篇 2025年12月16日 06:32:12
下一篇 2025年12月16日 06:32:22

相关推荐

发表回复

登录后才能评论
关注微信