为什么这个 OrDone 通道实现从 Done 通道接收两次？

本篇文章给大家分享《为什么这个 OrDone 通道实现从 Done 通道接收两次？》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

一直在阅读 go 中的 concurrency，它引入了一个方便的“or-done”通道。

tldr;当您使用无法控制的通道（可能来自系统的其他部分）时，代码可能会变得有点难看。

// quite nice to read
for v := range mychan {
  ... do stuff
}

// not so nice
loop:
for {
  select {
  case <- done:
    break loop
  case maybeval, ok := <-mychan:
    if !ok {
      return
    }
    // do something with maybeval
  }
}

本书提供了一种使用 ordone 通道来简化此操作的方法。定义如下。我不明白的是，为什么在嵌套选择中，我们需要再次从 <-done 接收数据。

ordone := func(done, c <-chan interface{}) <-chan interface{} {
  valstream := make(chan interface{})
  go func() {
    defer close(valstream)
    for {
      select {
      case <- done:
        return
      case v, ok := <-c:
        if !ok {
          return
        }
        select {
        case valstream <- v:
        case <-done: // why do we also need to receive on done here?
        }
      }
    }
  }
  return valstream
}

这允许您返回到原始的 for 循环，从而增强可读性 - 就像这样：

for val := range orDone(done, myChan) {
  // Once again, do something
}

解决方案

实际上只是增加了 peter 答案的可见性。

这是因为，如果接收 valstream 的人失去兴趣，valstream 本身可能会阻止发送。

在您从通道 c 收到值的情况下，您可以输入嵌套的 select{}。此时如果我们取出第二个 <-done 你就会得到这个；

select {
case valStream: <- v:
}

即使 done 通道关闭，这也会无限期地阻塞，直到从通道 v 接收到值为止。通过添加嵌套检查，我们允许自己在任一点退出 select{}。

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《为什么这个 OrDone 通道实现从 Done 通道接收两次？》文章吧，也可关注golang学习网公众号了解相关技术文章。