新闻资讯

质量为本、客户为根、勇于拼搏、务实创新

< 返回新闻资讯列表

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

发布时间:2023-10-08 16:20:48

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

在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
", id)
time.Sleep(2 * time.Second)
fmt.Printf("任务 %d 完成
", 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完成履行后再进行后续操作的场景中非常有用。