并发管理多个 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中文网其他相关文章!