温馨提示×

基础教程:Go WaitGroup及其在Golang中的应用

小云
101
2023-10-08 16:06:14
栏目: 编程语言

在Golang中,WaitGroup是一种用于等待一组goroutine完成执行的机制。它可以用来确保在主goroutine结束之前,所有的子goroutine都已经执行完毕。

要使用WaitGroup,首先需要导入sync包。然后,可以创建一个WaitGroup对象,并使用Add方法来设置需要等待的goroutine数量。接下来,在每个子goroutine的末尾使用Done方法来标记一个子goroutine的结束。最后,在主goroutine中使用Wait方法来等待所有的子goroutine完成执行。

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

package main
import (
"fmt"
"sync"
)
func main() {
var wg sync.WaitGroup
for i := 0; i < 5; i++ {
wg.Add(1)
go func(i int) {
defer wg.Done()
fmt.Println("Goroutine", i, "is done")
}(i)
}
wg.Wait()
fmt.Println("All goroutines finished executing")
}

在上面的示例中,我们创建了一个包含5个子goroutine的循环。每个子goroutine都会输出一个简单的消息,并在结束时调用Done方法。最后,我们在主goroutine中调用Wait方法来等待所有的子goroutine完成执行。

运行上面的代码,你会看到类似以下的输出:

Goroutine 0 is done
Goroutine 1 is done
Goroutine 3 is done
Goroutine 4 is done
Goroutine 2 is done
All goroutines finished executing

可以看到,所有的子goroutine都已经执行完毕,然后主goroutine输出了"所有goroutine都已经执行完毕"的消息。

总结来说,WaitGroup是一种在Golang中等待一组goroutine完成执行的机制。通过调用Add方法设置需要等待的goroutine数量,调用Done方法标记一个goroutine的结束,以及调用Wait方法来等待所有的goroutine完成执行。这是在并发编程中非常常用的一种技术。

0