php小編蘋果為您介紹一種實用的方法來執行所有的goroutine – sync.WaitGroup。在Go語言中，goroutine是輕量級的線程，可以同時執行多個任務。然而，有時我們需要等待所有的goroutine執行完畢后再繼續下一步操作。這時，sync.WaitGroup就派上用場了。它提供了一種簡單而有效的方式來等待所有的goroutine完成，確保程序的正確執行順序。接下來，我們將詳細介紹如何使用sync.WaitGroup來實現這個功能。
問題內容
目前我正在將所有值推送到通道并從中讀取并計算其平方。
我想避免使用 time.sleep(2000 * time.millisecond) 因為它會阻止執行 2 秒，而是我希望每個 goroutine 都處理并等待其執行，然后退出程序。我剛剛脫離了 golang，所以現在問這個基本問題:(。有人能幫我解決這個問題嗎？
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)
}
登錄后復制
解決方法
只需將 qui??t <- 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
}
登錄后復制
