sync.WaitGroup 在 Channels 上的优势是什么？

对于您的简单示例（表示作业完成），这WaitGroup是显而易见的选择。Go 编译器非常友好，不会怪你使用通道来发送简单的完成任务信号，但一些代码审查者会这样做。"WaitGroup 等待一组 goroutines 完成。主 goroutine 调用Add(n)设置要等待的 goroutines 的数量。然后每个 goroutines 运行并Done()在完成时调用。同时，Wait 可用于阻塞直到所有goroutine 已经完成了。”words := []string{"foo", "bar", "baz"}var wg sync.WaitGroupfor _, word := range words {    wg.Add(1)    go func(word string) {        defer wg.Done()        time.Sleep(100 * time.Millisecond) // a job        fmt.Println(word)    }(word)}wg.Wait()可能性仅受您的想象力的限制：通道可以被缓冲：words := []string{"foo", "bar", "baz"}done := make(chan struct{}, len(words))for _, word := range words {    go func(word string) {        time.Sleep(100 * time.Millisecond) // a job        fmt.Println(word)        done <- struct{}{} // not blocking    }(word)}for range words {    <-done}通道可以是无缓冲的，你可以只使用一个信号通道（例如chan struct{}）：words := []string{"foo", "bar", "baz"}done := make(chan struct{})for _, word := range words {    go func(word string) {        time.Sleep(100 * time.Millisecond) // a job        fmt.Println(word)        done <- struct{}{} // blocking    }(word)}for range words {    <-done}您可以限制具有缓冲通道容量的并发作业数：t0 := time.Now()var wg sync.WaitGroupwords := []string{"foo", "bar", "baz"}done := make(chan struct{}, 1) // set the number of concurrent job herefor _, word := range words {    wg.Add(1)    go func(word string) {        done <- struct{}{}        time.Sleep(100 * time.Millisecond) // job        fmt.Println(word, time.Since(t0))        <-done        wg.Done()    }(word)}wg.Wait()您可以使用频道发送消息：done := make(chan string)go func() {    for _, word := range []string{"foo", "bar", "baz"} {        done <- word    }    close(done)}()for word := range done {    fmt.Println(word)}基准：    go test -benchmem -bench . -args -n 0# BenchmarkEvenWaitgroup-8  1827517   652 ns/op    0 B/op  0 allocs/op# BenchmarkEvenChannel-8    1000000  2373 ns/op  520 B/op  1 allocs/op    go test -benchmem -bench .# BenchmarkEvenWaitgroup-8  1770260   678 ns/op    0 B/op  0 allocs/op# BenchmarkEvenChannel-8    1560124  1249 ns/op  158 B/op  0 allocs/op代码( main_test.go):package mainimport (    "flag"    "fmt"    "os"    "sync"    "testing")func BenchmarkEvenWaitgroup(b *testing.B) {    evenWaitgroup(b.N)}func BenchmarkEvenChannel(b *testing.B) {    evenChannel(b.N)}func evenWaitgroup(n int) {    if n%2 == 1 { // make it even:        n++    }    for i := 0; i < n; i++ {        wg.Add(1)        go func(n int) {            select {            case ch <- n: // tx if channel is empty            case i := <-ch: // rx if channel is not empty                // fmt.Println(n, i)                _ = i            }            wg.Done()        }(i)    }    wg.Wait()}func evenChannel(n int) {    if n%2 == 1 { // make it even:        n++    }    for i := 0; i < n; i++ {        go func(n int) {            select {            case ch <- n: // tx if channel is empty            case i := <-ch: // rx if channel is not empty                // fmt.Println(n, i)                _ = i            }            done <- struct{}{}        }(i)    }    for i := 0; i < n; i++ {        <-done    }}func TestMain(m *testing.M) {    var n int // We use TestMain to set up the done channel.    flag.IntVar(&n, "n", 1_000_000, "chan cap")    flag.Parse()    done = make(chan struct{}, n)    fmt.Println("n=", n)    os.Exit(m.Run())}var (    done chan struct{}    ch   = make(chan int)    wg   sync.WaitGroup)

sync.WaitGroup 在 Channels 上的优势是什么？

3回答