使用 channel 或 sync.Cond 等待条件

Question

我正在尝试等待一个特定的条件，我想知道如何最好地做到这一点。我有一个看起来像这样的结构（简化）：

type view struct {
    timeFrameReached bool
    Rows []*sitRow
}

在 goroutine 中，我正在更新一个文件，该文件被读入 view 变量。行数增加，timeFrameReached 最终将是true。

在其他地方，我想等待以下条件成立：

view.timeFrameReached == true || len(view.Rows) >= numRows

我正在尝试学习通道以及 Go 的条件变量如何工作，我想知道这里最好的解决方案是什么。理论上，我可以做一些像这样微不足道的事情：

for {
    view = getView()
    if view.timeFrameReached == true || len(view.Rows) >= numRows {
        break
    }
}

但这显然是一个幼稚的解决方案。 numRows 的值来自 HTTP 请求，因此条件方法似乎具有挑战性。 goroutine 不知道何时广播条件，因为它不知道它正在寻找的行数。

Answer 1

我想我会用条件变量来做到这一点。这个概念并不是说Signal 应该只在服务员想要检查的条件为真时才执行，而是在它可能为真时（即，被检查的东西已经改变）。

这样做的正常模式是：

mutex.Lock()
for {
    view = getView()
    if view.timeFrameReached == true || len(view.Rows) >= numRows {
        break
    }
    cond.Wait(&mutex)
}
// Do stuff with view
mutex.Unlock()

然后，在 view 被更改的代码中：

mutex.Lock()
// Change view
cond.Signal() // or cond.Broadcast()
mutex.Unlock()

显然，我在不知道您的程序如何工作的情况下编写了此代码，因此您可能需要进行一些更改。

您可以通过在通道上发送信号并尝试从通道接收等待来对通道执行类似的操作，但这对我来说似乎更复杂。 （另外，如果您有多个 goroutine 等待，您可以使用 cond.Broadcast 通知所有这些 goroutine 唤醒。）

Answer 2

我的一个想法涉及通过通道传达所需的行数，构建视图的 goroutine 将执行非阻塞接收，以查看主线程是否正在请求一定数量的行。如果是，它会发回一条消息，表明条件满足。

这里的main 函数请求行数：

if numRows > len(viewFile.View.Rows) && !viewFile.View.TimeFrameReached {
    // Send the required number of rows
    rows <- numRows
    // Wait for the prefetch loop to signal that the view file is ready
    <-rows // Discard the response value and move on
    view = getView()
}

这里 goroutine 检查是否需要一定数量的行。如果是这样，它会在准备好时以肯定的信号进行响应。该信号的值无关紧要。

select {
    case numRows := <-rows:
        if len(viewFile.View.Rows) >= numRows || viewFile.View.TimeFrameReached {
            rows <- 1
        }
    default:
}