在选择中支持一种通信（chan）

Question

我知道，如果在select 语句中可以进行多个“通信”，则随机选择一个。我正在尝试寻找一种替代方法，它可以更喜欢一种“交流”而不是另一种。

背景是我在一个使用上下文杀死的通道上的 go-routine 中发送值。当我杀死它时，我希望立即关闭通道，但目前代码有时会在关闭之前在通道上发送最终值。

下面是简化版的代码：

   ctx, cancel := context.WithCancel(context.Background())
   ch := make(chan int)

   go func() {
      defer close(ch)
      for i := 1; ; i++ {
         select {
         case <-ctx.Done():
            return
         case ch <- i:
         }
      }
   }()

   print(<-ch)
   print(<-ch)
   cancel()
   print(<-ch)
   print(<-ch)

这有时会打印 1200，但通常会打印 1230。Try it on the playground

对于如何重组代码以支持第一种情况有什么想法吗？ （即总是打印 1200。）

Answer 1

请注意，这是一个更新的答案，因为原始答案存在问题。

正如其他人指出的那样，如果没有额外的同步，您将无法避免竞争条件。您可以使用 Mutex，但 sync.Cond 似乎很合适。 在下面的代码中，接收 goroutine 表示它已从 chan 接收到值。它在发出信号之前取消上下文（使用Cond.Signal），发送goroutine 等待信号。这避免了竞争条件，因为上下文状态在检查之前已更新。

ctx, cancel := context.WithCancel(context.Background())
ch := make(chan int)
cond := sync.NewCond(&sync.Mutex{}) // *** new

go func() {
    defer close(ch)
    cond.L.Lock()                   // *** new
    defer cond.L.Unlock()           // *** new
    for i := 1; ; i++ {
        ch <- i                     // *** moved
        cond.Wait()                 // *** new
        if ctx.Err() != nil {       // *** changed
            return
        }
    }
}()

print(<-ch)
cond.Signal()                      // *** new
print(<-ch)
cond.Signal()                      // *** new
print(<-ch)
cancel()
cond.Signal()                      // *** new
print(<-ch)
cond.Signal()                      // *** new

这是我能看到的最简单的方法，接收 goroutine 在取消上下文后将不会在通道上接收任何值。

在Playground上试试

Answer 2

这似乎是不可能的，因为cancel() 不是主 goroutine 中的阻塞操作。正因为如此，当select 解除阻塞时，可能会有多种情况可用，并且没有办法让一个通道优于另一个通道。任何类型的 check-channel-then-write 方案都将是活泼的，因为在检查后可以取消上下文。

使用done 通道并写入它而不是取消上下文将起作用，因为写入done 通道将是主 goroutine 的阻塞操作，并且 select 将始终有一个活动案例。