如何在Golang中实现多任务同步等待-Golang-PHP中文网

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

P粉602998670

发布： 2025-10-11 14:39:01

原创

263人浏览过

如何在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 < 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 包实现超时控制。

ViiTor实时翻译

AI实时多语言翻译专家！强大的语音识别、AR翻译功能。

116

查看详情

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