如何使用 sync.WaitGroup 来执行所有的 goroutine

发布于04月27日

目前，我正在将所有值推送到通道中，并从通道中读取数据，然后对其进行平方运算. 我想避免使用time.Sleep(2000 * time.Millisecond)，因为它会阻塞执行2秒，相反，我希望每个Goroutine都处理并等待它的执行，然后退出程序.我刚刚和golang 失go 了联系，所以现在问这个基本的问题:有人能帮我这个忙吗？

package main

import (
    "fmt"
    "sync"
    "time"
)

func doSquare(num int) int {
    return num * num
}

var wg sync.WaitGroup

func main() {

    wg.Add(1)
    st := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
    quit := make(chan bool)
    ch := make(chan int)
    go func() {
        for i := range st {
            ch <- i
        }
    }()

    go func() {
        for {
            select {
            case x := <-ch:
                fmt.Println(doSquare(x))
                // return

            case <-quit:
                wg.Done()
            default:
                // fmt.Println("---")
                // do something
            }
        }
    }()

    quit <- true

    wg.Wait()
    time.Sleep(2000 * time.Millisecond)
}

推荐答案

只需移动quit <- true到第一个Goroutine的尽头

func main() {
    wg.Add(1)
    st := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
    quit := make(chan bool)
    ch := make(chan int)
    go func() {
        for i := range st {
            ch <- i
        }
        quit <- true
    }()

    go func() {
        for {
            select {
            case x := <-ch:
                fmt.Println(doSquare(x))
                // return

            case <-quit:
                wg.Done()
                return
            default:
                // fmt.Println("---")
                // do something
            }
        }
    }()

    wg.Wait()
}

这是另一种表示close(ch)年前没有更多数据的方式

func main() {
    st := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
    quit := make(chan bool)
    ch := make(chan int) // improve concurrency by `ch := make(chan int, 100)`
    go func() {
        for i := range st {
            ch <- i
        }
        close(ch)
    }()

    go func() {
        for x := range ch {
            fmt.Println(doSquare(x))
        }
        quit <- true
    }()

    <-quit
}