等待缓冲通道满

Question

我一定错过了什么，因为我还没有在网上找到这个非常基本的问题的答案。我正在使用能够保存三个int 值的缓冲通道。

然后我使用三个 goroutine 来填充它，并且我想在缓冲通道满后执行操作。

这是一个解释问题的sn-p：

func main() {
    // Initialization of the slice a and 0 < n < len(a) - 1.
    difs := make(chan int, 3)
    go routine(a[:n], difs)
    go routine(a[n + 1:], difs)
    go routine(a[n - 1:n + 1], difs)

    fmt.Println(<-difs) // Display the first result returned by one of the routine.
}

func routine(a []int, out chan<- int) {
    // Long computation.
    out <- result
}

我想更新我的代码，以便fmt.Println(<-difs) 在计算完所有值后显示int 的数组。我可以使用 3 次 <-difs 但我想知道 Go 是否提供了更清洁的东西来做到这一点。

Answer 1

函数len() 支持通道，返回通道内未读队列元素的数量。但是，您必须运行一个循环来定期检查它。

处理此问题的传统方法是使用sync.WaitGroup，如下所示：

func main() {
    // Initialization of the slice a and 0 < n < len(a) - 1.
    var wg sync.WaitGroup
    wg.Add(3)
    difs := make(chan int, 3)
    go routine(a[:n], difs, &wg)
    go routine(a[n + 1:], difs, &wg)
    go routine(n - 1:n + 1], difs, &wg)

    wg.Wait() // blocks until wg.Done() is called 3 times
    fmt.Println(<-difs) // Display the first result returned by one of the routine.
}

// NOTE: This MUST accept a POINTER to the waitgroup.  They are not copy-safe.
func routine(a []int, out chan<- int, wg *sync.WaitGroup) {
    defer wg.Done()
    // Long computation.
    out <- result
}

Answer 2

使用通道本身等待，就像这个工作示例代码：

package main

import "fmt"

func main() {
    a := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12} // Initialization of the slice a and 0 < n < len(a) - 1.
    difs := make(chan int, 3)

    go routine(a[0:4], difs)
    go routine(a[4:8], difs)
    go routine(a[8:12], difs)

    result := []int{<-difs, <-difs, <-difs}

    fmt.Println(result) // Display the first result returned by one of the routine.
}

func routine(a []int, out chan<- int) {
    result := 0 // Long computation.
    for _, v := range a {
        result += v
    }
    out <- result
}

输出：

[10 42 26]