共計 661 個字符,預計需要花費 2 分鐘才能閱讀完成。
在高并發(fā)場景下,Go 的 WaitGroup 是一種有效的工具,用于等待一組 goroutine 完成其任務。
WaitGroup 有三個方法可以使用:
Add(n int)
:表示等待的 goroutine 數量增加 n。Done()
:表示等待的 goroutine 數量減少 1。Wait()
:阻塞等待,直到等待的 goroutine 數量為 0。
下面是一個使用 WaitGroup 的示例代碼,在高并發(fā)場景下,使用 goroutine 處理任務,并使用 WaitGroup 等待所有 goroutine 完成:
package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
for i := 0; i < 10; i++ {wg.Add(1)
go func(n int) {defer wg.Done()
time.Sleep(time.Second) // 模擬任務處理時間
fmt.Printf("Task %d completed\n", n)
}(i)
}
wg.Wait() // 等待所有任務完成
fmt.Println("All tasks completed")
}
在上面的代碼中,我們創(chuàng)建了 10 個 goroutine 來處理任務。每個 goroutine 在完成任務后,調用 Done()
方法來減少等待的 goroutine 數量。最后,調用 Wait()
方法來阻塞等待,直到所有 goroutine 完成任務。
通過使用 WaitGroup,我們可以保證在高并發(fā)場景下,所有的 goroutine 都能夠完成任務后再執(zhí)行后續(xù)的操作。
丸趣 TV 網 – 提供最優(yōu)質的資源集合!
正文完