温馨提示×

Golang异步处理与Go WaitGroup的巧妙结合

小云
104
2023-10-08 16:59:17
栏目: 编程语言

在Golang中,可以使用Go WaitGroup来同步并发操作。WaitGroup是一个计数器,用于等待一组goroutine完成任务。

当我们需要异步处理多个任务时,可以使用WaitGroup来等待所有任务完成。下面是一种巧妙地结合异步处理和WaitGroup的方法:

  1. 首先,创建一个WaitGroup对象,并设置计数器的初始值为任务的数量。
var wg sync.WaitGroup
wg.Add(numTasks)
  1. 在处理每个任务时,使用goroutine来异步执行任务,并在任务完成时调用Done方法减少计数器的值。
for _, task := range tasks {
go func(t Task) {
// 处理任务 t
// ...
// 任务完成时调用Done方法
wg.Done()
}(task)
}
  1. 在所有任务都启动后,使用Wait方法来等待所有任务完成。
wg.Wait()

这样,我们就可以同时处理多个任务,并在所有任务完成后继续执行后续操作。

完整的示例代码如下:

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,我们可以高效地处理并发任务,并在所有任务完成后继续执行后续操作。

0