温馨提示×

并发控制实例解析:Golang中使用Go WaitGroup完成任务

小云
102
2023-10-08 16:20:48
栏目: 编程语言

在Golang中,我们可以使用sync包中的WaitGroup来实现并发控制。WaitGroup用于等待一组goroutine完成执行。

下面是一个使用WaitGroup来完成任务的示例:

package main
import (
"fmt"
"sync"
"time"
)
func main() {
var wg sync.WaitGroup
// 设置要执行的任务数量
taskCount := 5
wg.Add(taskCount)
for i := 1; i <= taskCount; i++ {
go performTask(i, &wg)
}
// 等待所有任务完成
wg.Wait()
fmt.Println("所有任务已完成")
}
func performTask(id int, wg *sync.WaitGroup) {
defer wg.Done()
fmt.Printf("开始执行任务 %d\n", id)
time.Sleep(2 * time.Second)
fmt.Printf("任务 %d 完成\n", id)
}

在上面的示例中,我们首先创建了一个WaitGroup实例wg,然后设置要执行的任务数量为5,通过调用wg.Add(taskCount)方法来增加等待的goroutine数量。

接下来,我们使用一个循环来创建5个goroutine,并将每个goroutine的任务ID和WaitGroup实例的指针传递给performTask函数。

performTask函数中,我们使用defer关键字来确保在函数执行完成后调用wg.Done()方法,表示任务已完成。

在每个goroutine中,我们首先打印出任务开始执行的信息,然后通过time.Sleep函数模拟任务执行的时间。最后,我们打印出任务完成的信息。

最后,我们调用wg.Wait()方法来等待所有的goroutine完成任务。一旦所有的goroutine都调用了wg.Done()方法,Wait方法就会返回,程序继续执行后面的代码。

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

开始执行任务 1
开始执行任务 2
开始执行任务 3
开始执行任务 4
开始执行任务 5
任务 3 完成
任务 1 完成
任务 4 完成
任务 5 完成
任务 2 完成
所有任务已完成

从输出可以看出,5个任务是并发执行的,但是最后的输出表明所有的任务都已完成。

使用WaitGroup来完成任务的并发控制,它可以确保在所有的goroutine执行完成之前,主goroutine会一直等待。这在需要等待一组goroutine完成执行后再进行后续操作的场景中非常有用。

0