重构代码以按惯用方式使用单个通道

Question

我有以下代码：

package main

import (
    "fmt"
    "time"
)

type Response struct {
    Data   string
    Status int
}

func main() {
    var rc [10]chan Response
    for i := 0; i < 10; i++ {
        rc[i] = make(chan Response)
    }
    var responses []Response

    for i := 0; i < 10; i++ {
        go func(c chan<- Response, n int) {
            c <- GetData(n)
            close(c)
        }(rc[i], i)
    }

    for _, resp := range rc {
        responses = append(responses, <-resp)
    }

    for _, item := range responses {
        fmt.Printf("%+v\n", item)
    }
}

func GetData(n int) Response {
    time.Sleep(time.Second * 5)
    return Response{
        Data:   "adfdafcssdf4343t43gf3jn4jknon239nwcwuincs",
        Status: n,
    }
}

您能告诉我哪种方法是实现相同目标但使用单一渠道的正确方法吗？

Answer 1

由于您可以同时编写不同的数组和切片元素，因此您不需要任何通道。详情请参见能否同时写入不同的切片元素。

只需启动您的 goroutine，并让它们写入适当的数组（或切片）元素。使用 async.WaitGroup等待所有完成：

wg := &sync.WaitGroup{}
var responses [10]Response
for i := range responses {
    wg.Add(1)
    go func(n int) {
        defer wg.Done()
        responses[n] = GetData(n)
    }(i)
}

wg.Wait()
for _, item := range responses {
    fmt.Printf("%+v\n", item)
}

这输出与您的代码相同。在Go Playground上试一试。

另请参阅相关：如何从以特定顺序执行的 N 个 goroutine 中收集值？