golang协程同步的几种方式

10xjzheng · · 1311 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

1. 最不靠谱的sleep

package main
 import (
   "fmt"
  "time"
 )
 func main() {
   go func() {
    fmt.Println("Goroutine 1")
  }()
  go func() {
    fmt.Println("Goroutine 2")
  }()
  time.Sleep(time.Second * 1) // 睡眠1秒，等待上面两个协程结束

2.goroutine / channel 最佳组合拳

package main
 import (
   "fmt"
 )
 func main() {
   ch := make(chan struct{})
  count := 2 // count 表示活动的协程个数
  go func() {
    fmt.Println("Goroutine 1")
    ch <- struct{}{} // 协程结束，发出信号
  }()
  go func() {
    fmt.Println("Goroutine 2")
    ch <- struct{}{} // 协程结束，发出信号
  }()
  for range ch {
    // 每次从ch中接收数据，表明一个活动的协程结束
    count--
    // 当所有活动的协程都结束时，关闭管道
    if count == 0 {
      close(ch)
    }
  }
}

3.最简单的方式 sync.WaitGroup

package main
 import (
   "fmt"
  "sync"
 )
 func main() {
  var wg sync.WaitGroup
  wg.Add(2) // 因为有两个动作，所以增加2个计数
  go func() {
    fmt.Println("Goroutine 1")
    wg.Done() // 操作完成，减少一个计数
  }()
  go func() {
    fmt.Println("Goroutine 2")
    wg.Done() // 操作完成，减少一个计数
  }()
  wg.Wait() // 等待，直到计数为0
}

有疑问加站长微信联系（非本文作者）