当前位置：首页 > 文章列表 > Golang > Go教程 > Golang多任务同步等待技巧

Golang多任务同步等待技巧

2025-10-09 12:59:29 0浏览收藏

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("等待超时，部分任务未完成")
  }
}