使用select在不同通道上接收多个并发协程：逻辑与性能是否有差异？

本文探讨了在 Go 语言中使用 “select” 在不同通道上接收多个并发协程的两种方案：receive() 和 receive2()。作者质疑 receive2() 是否过于复杂，并询问为什么使用 “select” 的方案被认为更惯用。本文还讨论了 “扇入” 方案的实现，即将多个通道的结果收集到一个通道中。

选择与在不同通道上接收的多个并发协程：逻辑或性能有区别吗？

我的问题更普遍的是关于 go 中“扇入”方案的实现。在我看来，使用“select”的方案在任意大量通道（大量通道）的情况下不起作用。

请参阅下面示例中的 receive() 和 receive2()。

receive2() 函数是否过于复杂？矫枉过正？

为什么选择的表述被认为更惯用？

package main

import (
    "fmt"
    "time"
)

func main() {

    var ch1 = make(chan int)
    var ch2 = make(chan int)

    send(ch1, ch2)
    //receive(ch1, ch2)
    receive2(ch1, ch2)

    time.Sleep(3 * time.Second)

}

func send(ch1 chan int, ch2 chan int) {
    go func() {
        for i := 0; i < 10; i++ {
            ch1 <- i
        }
        close(ch1)
    }()
    go func() {
        for i := 0; i < 10; i++ {
            ch2 <- i
        }
        close(ch2)
    }()
}

func receive(ch1 chan int, ch2 chan int) {
    go func() {
        for item := range ch1 {
            fmt.Printf("1: %d\n", item)
        }
    }()

    go func() {
        for item := range ch2 {
            fmt.Printf("2: %d\n", item)
        }
    }()
}

func receive2(ch1 chan int, ch2 chan int) {
    for {
        select {
        case x, ok := <-ch1:
            fmt.Println("ch1", x, ok)
            if !ok {
                ch1 = nil
            }
        case x, ok := <-ch2:
            fmt.Println("ch2", x, ok)
            if !ok {
                ch2 = nil
            }
        }

        if ch1 == nil && ch2 == nil {
            break
        }
    }
}

正确答案

您的实现不是扇入。扇入将多个通道的结果收集到一个通道中。像这样，您可以在单个 out 通道上接收所有值。

func fanin(in ...chan int) chan int {
    out := make(chan int)
    for _, c := range in {
        go func(i chan int) {
            for v := range i {
                out <- v
            }

        }(c)
    }
    return out
}

在您的实现中，第一个 receive() 只是从 n 个通道同时接收，彼此独立。它可以被重写以接受可变参数，并且看起来更像是一个实际的扇入：

func receive(cs ...chan int) {
    for i, cN := range cs {
        go func(i int, c chan int) {
            for item := range c {
                fmt.Printf("%d: %d\n", i, item)
            }
        }(i, cN)
    }
}

receive2() 本质上是连续的并且不会扩展，因为您必须为每个通道编写一个 case，并且 && 将它们全部写在一起才能知道何时打破循环。 && 可以用 sync.waitgroup 重写，但每次迭代时您仍将仅处理一项（在准备接收的情况下随机处理）。

理论要掌握，实操不能落！以上关于《使用select在不同通道上接收多个并发协程：逻辑与性能是否有差异？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！