温馨提示×

Golang中的并发编程与WaitGroup的结合应用

小云
101
2023-10-08 16:07:27
栏目: 编程语言

在Golang中,可以使用WaitGroup来管理并发的goroutine,以确保在所有goroutine完成之前,主程序不会退出。

WaitGroup是一个计数信号量,它的值初始为0。每个goroutine在执行之前,通过调用WaitGroup的Add方法将计数加1。在goroutine的末尾,通过调用WaitGroup的Done方法将计数减1。主程序调用WaitGroup的Wait方法,将会被阻塞,直到所有goroutine的计数变为0。

下面是一个使用WaitGroup的示例:

package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
for i := 0; i < 5; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
fmt.Printf("goroutine %d\n", id)
}(i)
}
wg.Wait()
fmt.Println("All goroutines have finished")
}

在上面的例子中,主程序启动了5个goroutine,并通过调用wg.Add(1)将计数加1。每个goroutine在执行完任务后,通过调用wg.Done()将计数减1。最后,主程序调用wg.Wait(),将会被阻塞,直到所有goroutine的计数变为0。当所有goroutine完成后,主程序会打印"All goroutines have finished"。

使用WaitGroup可以很方便地管理并发的goroutine,确保它们的执行顺序和主程序的退出顺序一致。

0