在 Go 中访问通道长度的正确方法

Question

我使用 Go 已经有一段时间了，而且每天都在进步，但我本身并不是专家。目前我正在处理并发和 goroutines，因为我认为这是我的 Go 工具带中的最后一个未知数。我想我已经掌握了它的窍门，但仍然绝对是初学者。

我遇到的问题对我来说似乎很基本，但我尝试过的任何工作都没有。我想找出一种计算通道长度的方法。

根据我收集到的信息，len() 仅适用于缓冲通道，因此在这种情况下对我没有帮助。我正在做的是分批从数据库中读取值。我有一个生成器函数，就像

func gen() chan Result {
  out := make(chan Result)

  go func() {
    ... query db
    for rows.Next() {
      out <- row
    }
     close(out)
   }()

  return out
}

那么我就这样使用它

c := gen()

...

// do other stuff

我想用输出通道返回计数，或者将所有计数包装在一个结构类型中并返回它。

像这样：

c, len := gen()

或：

a := gen()

fmt.Println(a.c)
fmt.Println(a.len)

我相信除了使用 atomic 之外，我已经尝试了所有方法，我认为这实际上会起作用，但我阅读了周围，显然使用 atomic 不是正确的事情。我还有哪些其他选项不会让我留下 0 或无限阻塞

谢谢！

Answer 1

len 内置函数将返回通道的“长度”：

func len(v Type) int

len 内置函数根据类型返回 v 的长度：

Array: the number of elements in v.
Pointer to array: the number of elements in *v (even if v is nil).
Slice, or map: the number of elements in v; if v is nil, len(v) is zero.
String: the number of bytes in v.
Channel: the number of elements queued (unread) in the channel buffer;

如果 v 为零，则 len(v) 为零。

但我认为这对你没有帮助。

您真正需要的是一种解决问题的新方法：计算通道中队列中的项目并不是处理“批量”任务的合适方法。

你需要这个长度做什么？

Answer 2

您使用的是非缓冲通道。谢谢你???? 无缓冲通道不使用内存。因此从不包含任何东西！ 无缓冲通道的唯一目的是通过将元素从一个传递到另一个来实现 goroutine 之间的同步。而已 ！

    go func(){
        c:=make(chan struct{})
        c<-struct{}{} // Definitely locked
    }()

又一个死锁

    go func(){
        c:=make(chan struct{})
        <-c // Definitely locked
        c<-struct{}{} // Never get there
    }()

使用另一个 goroutine 读取通道

    go func(){
        c:=make(chan struct{})
        go func(){<-c}()
        c<-struct{}{}
    }()

在您的情况下，您有一个生成器，这意味着您必须读取通道，直到生产者 goroutine 将其关闭。这是一个很好的设计，可以确保你的 goroutine 没有悬空。

// Read the channel until the injecter goroutine finishes and closes it.
for r := range gen() {
...
}
// goroutine inner of gen() as finished

Answer 3

我假设，根据您的后续回答，您实际上想知道工作人员池和通道上的缓冲区的“好”值，以保持一切“最佳”工作。

这非常困难，并且取决于工人正在做什么，但首先我会在运行时查看缓冲通道和工人池的最小值。GOMAXPROCS(0)。如果你有很多资源，那么你可以去“无限”工人。