时间戳

首页 > 解决方案 > 在选择中支持一种通信（chan）

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

问题描述

我知道，如果有多个“通信”可以在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。 在操场上尝试一下

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

标签： goselectchannel

解决方案

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

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

推荐阅读