如何通知另一个 goroutine 停止？ [复制]

Question

我有 2 个 goroutine，g 用于检测 f 应该停止的条件，f 在进行实际处理之前检查它是否应该在每次迭代中停止。在 Java 等其他语言中，我会使用线程安全的共享变量，如以下代码：

func g(stop *bool) {
  for {
    if check_condition() {
      *stop = true
      return
    }
  }
}

func f(stop *bool) {
  for {
    if *stop {
      return
    }
    do_something()
  }
}

func main() {
  var stop = false
  go g(&stop)
  go f(&stop)
  ...
}

我知道上面的代码不安全，但是如果我使用通道将停止从g 发送到f，f 在从通道读取时会被阻止，这是我想要避免的。在 Go 中执行此操作的安全且惯用的方法是什么？

Answer 1

使用 channel close 来通知其他 goroutine 一个条件。在检查条件时使用带有默认子句的 select 以避免阻塞。

func g(stop chan struct{}) {
    for {
        if check_condition() {
            close(stop)
            return
        }
    }
}

func f(stop chan struct{}) {
    for {
        select {
        case <-stop:
            return
        default:
            do_something()
        }
    }
}

func main() {
    var stop = make(chan struct{})
    go g(stop)
    go f(stop)
}

它也可以向容量大于零的通道发送值，但关闭通道会扩展到支持多个 goroutine。

Answer 2

方法是使用带有默认子句的 select 语句（请参阅this example）。

所以f 看起来像：

func f(stop chan bool) {
  select {
    case s := <- stop:
      if s {
        return
      }
    default:
      do_something()
  }
}