在Golang中,可以使用Go WaitGroup来同步并发操作。WaitGroup是一个计数器,用于等待一组goroutine完成任务。
当我们需要异步处理多个任务时,可以使用WaitGroup来等待所有任务完成。下面是一种巧妙地结合异步处理和WaitGroup的方法:
1. 首先,创建一个WaitGroup对象,并设置计数器的初始值为任务的数量。
```go
var wg sync.WaitGroup
wg.Add(numTasks)
```
2. 在处理每个任务时,使用goroutine来异步执行任务,并在任务完成时调用Done方法减少计数器的值。
```go
for _, task := range tasks {
go func(t Task) {
// 处理任务 t
// ...
// 任务完成时调用Done方法
wg.Done()
}(task)
}
```
3. 在所有任务都启动后,使用Wait方法来等待所有任务完成。
```go
wg.Wait()
```
这样,我们就可以同时处理多个任务,并在所有任务完成后继续执行后续操作。
完整的示例代码如下:
```go
package main
import (
"fmt"
"sync"
"time"
)
type Task struct {
id int
}
func main() {
numTasks := 10
tasks := make([]Task, numTasks)
var wg sync.WaitGroup
wg.Add(numTasks)
for i := 0; i < numTasks; i++ {
tasks[i] = Task{id: i}
go func(t Task) {
// 模拟处理任务
time.Sleep(1 * time.Second)
fmt.Printf("Task %d done\n", t.id)
// 任务完成时调用Done方法
wg.Done()
}(tasks[i])
}
// 等待所有任务完成
wg.Wait()
fmt.Println("All tasks done")
}
```
输出结果:
```
Task 1 done
Task 2 done
Task 5 done
Task 8 done
Task 4 done
Task 9 done
Task 6 done
Task 7 done
Task 3 done
Task 0 done
All tasks done
```
通过巧妙地结合异步处理和WaitGroup,我们可以高效地处理并发任务,并在所有任务完成后继续执行后续操作。