首页 > Golang > Go教程

Golang多任务同步等待技巧

时间：2025-10-09 12:59:29 491浏览收藏

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《Golang多任务同步等待实现方法》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

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

在Golang中实现多任务同步等待，最常用且推荐的方式是使用 sync.WaitGroup。它能有效协调多个goroutine的执行，确保主流程等待所有子任务完成后再继续。

使用 sync.WaitGroup 实现同步等待

WaitGroup 是 Go 标准库中用于等待一组并发任务完成的同步原语。它通过计数器控制，当计数器归零时，阻塞的 Wait 调用才会返回。

基本使用步骤：

创建一个 sync.WaitGroup 变量
每启动一个goroutine前调用 Add(1) 增加计数
在每个goroutine结束时调用 Done() 减少计数
在主协程中调用 Wait() 阻塞，直到计数归零

示例代码：

func main() {
  var wg sync.WaitGroup

  for i := 0; i < 5; 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 < 5; i++ {
    wg.Add(1)
    go func(id int) {
      defer wg.Done()
      select {
      case <-time.After(2 * time.Second):
        fmt.Printf("任务 %d 成功完成\n", id)
      case <-ctx.Done():
        fmt.Printf("任务 %d 被取消\n", id)
      }
    } (i)
  }

  ch := make(chan struct{})
  go func() {
    wg.Wait()
    close(ch)
  }()

  select {
  case <-ch:
    fmt.Println("全部任务正常完成")
  case <-ctx.Done():
    fmt.Println("等待超时，部分任务未完成")
  }
}

常见注意事项

使用 WaitGroup 时需注意以下几点，避免出现死锁或 panic：

确保每次 Add(n) 调用都对应 n 次 Done()，否则 Wait 可能永不返回
不要在 goroutine 外部调用 Done()，应由每个任务自己负责通知完成
避免在 Add 前启动 goroutine，防止竞争条件
通常将 defer wg.Done() 放在 goroutine 开头，确保无论函数如何退出都能触发

基本上就这些。sync.WaitGroup 简单高效，适合大多数多任务同步场景。配合 context 使用，还能增强程序的健壮性和可控性。不复杂但容易忽略细节。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《Golang多任务同步等待技巧》文章吧，也可关注golang学习网公众号了解相关技术文章。