同時管理多個 Goroutine 時,常常需要監控活躍 Goroutine 的數量。在 Go 中,原生的 runtime.NumGoroutine() 函數提供了這些資訊。
考慮以下範例:
var element int func deen(queue chan int) { element := <-queue fmt.Println("element is ", element) if element%2 == 0 { fmt.Println("new element is ", element) queue <- (element*100 + 11) queue <- (element*100 + 33) } } func main() { queue := make(chan int, 10) queue <- 1 queue <- 2 queue <- 3 queue <- 0 for len(queue) != 0 { for i := 0; i < 2; i++ { go deen(queue) } } fmt.Scanln() fmt.Println("list has len", len(queue)) //this must be 0 }
雖然此程式碼示範了 Goroutine 的使用,但它沒有提供方法計算任意給定時間活躍 goroutine 的數量。
為了解決這個問題,更有效的方法是利用sync.WaitGroup,協調多個goroutines的完成。
func deen(wg *sync.WaitGroup, queue chan int) { for element := range queue { fmt.Println("element is ", element) if element%2 == 0 { fmt.Println("new element is ", element) wg.Add(2) queue <- (element*100 + 11) queue <- (element*100 + 33) } wg.Done() } } func main() { var wg sync.WaitGroup queue := make(chan int, 10) queue <- 1 queue <- 2 queue <- 3 queue <- 0 for i := 0; i < 4; i++ { wg.Add(1) go deen(&wg, queue) } wg.Wait() close(queue) fmt.Println("list has len", len(queue)) //this must be 0 }
在這個修改後的程式碼中,sync.WaitGroup用來追蹤活動的goroutines數量。每個 Goroutine 在完成後都會遞減計數器,主 Goroutine 會等待所有 Goroutine 執行完畢才繼續執行。
透過利用 runtime.NumGoroutine() 或更有效率的sync.WaitGroup,Go 程式設計師可以有效地監控和管理應用程式中活躍的 goroutine。
以上是如何高效統計和管理Go中活躍的goroutine?的詳細內容。更多資訊請關注PHP中文網其他相關文章!