在Golang中,我们可以使用`sync`包中的`WaitGroup`来实现并发控制。`WaitGroup`用于等待一组goroutine完成执行。
下面是一个使用`WaitGroup`来完成任务的示例:
```go
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完成执行后再进行后续操作的场景中非常有用。