在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完成执行后再进行后续操作的场景中非常有用。