golang优化-正确姿势使用 goroutine

Goroutines 是轻量级的，但它们不是免费的： 至少，它们会为堆栈和 CPU 的调度消耗内存。 虽然这些成本对于 Goroutines 的使用来说很小，但当它们在没有受控生命周期的情况下大量生成时会导致严重的性能问题。 具有非托管生命周期的 Goroutines 也可能导致其他问题，例如防止未使用的对象被垃圾回收并保留不再使用的资源。

因此，不要在代码中泄漏 goroutine。 使用 go.uber.org/goleak 来测试可能产生 goroutine 的包内的 goroutine 泄漏。

一般来说，每个 goroutine:

• 必须有一个可预测的停止运行时间； 或者
• 必须有一种方法可以向 goroutine 发出信号它应该停止
  在这两种情况下，都必须有一种方式代码来阻塞并等待 goroutine 完成。

For example:

Bad Code (大家平时写的的代码)

go func() {
  for {
    doSomething()
    time.Sleep(delay)
  }
}()

上面代码循环等待delay时间，执行函数，但没有办法阻止这个 goroutine。这将一直运行到应用程序退出。

Good Code(优化后的代码）

var (
  stop = make(chan struct{}) // 告诉 goroutine 停止
  done = make(chan struct{}) // 告诉我们 goroutine 退出了
)
go func() {
  defer close(done) //函数执行完关闭chan
  ticker := time.NewTicker(delay) //同bad code的 time.Sleep(delay)
  defer ticker.Stop()   
  for {
    select {
    case <-tick.C:
      doSomething()
    case <-stop: //读到stop后函数退出
      return
    }
  }
}()
// 其它...
close(stop)  // 指示 goroutine 停止
<-done       // close了函数那么也就close了done，也就能读到done了

这个 goroutine 可以通过 close(stop), 我们可以等待它退出 <-done.

等待 goroutines 退出

给定一个由系统生成的 goroutine， 必须有一种方案能等待 goroutine 的退出。 有两种常用的方法可以做到这一点：

• 使用同步等待组 sync.WaitGroup. 如果您要等待多个 goroutine，请执行此操作

var wg sync.WaitGroup
for i := 0; i < N; i++ {
  wg.Add(1)
  go func() {
    defer wg.Done()
    // ...
  }()
}

// To wait for all to finish:
wg.Wait()

• 添加另一个 chan struct{}，goroutine 完成后会关闭它。 如果只有一个 goroutine，请执行此操作。

done := make(chan struct{})
go func() {
  defer close(done)
  // ...
}()

// To wait for the goroutine to finish:
<-done