waitgroup协调并发任务:创建 waitgroup:创建一个waitgroup数据结构来跟踪并发任务。加入任务:在每个并发任务开始时,调用add(1)增加计数器。完成任务:当任务完成时,调用done()减少计数器。等待完成:在主任务中,调用wait()阻塞,直到所有任务完成,确保数据一致性。

GoLang 函数中使用 WaitGroup 协调并发任务
简介
当在 Go 中使用并发特性时,经常需要协调多个并发执行的任务。WaitGroup 是一种同步机制,用于等待一组并发的 goroutine 完成他们的任务。
WaitGroup 的用法
WaitGroup 是一个用于管理并发的 goroutine 组的数据结构。它包含两个主要字段:
-
Counter:一个计数器,跟踪正在等待完成的 goroutine 数。 -
Wait:一个 channel,用于阻塞直到所有 goroutine 完成。
要使用 WaitGroup,请执行以下步骤:
-
创建一个新的 WaitGroup:
import "sync/atomic" var wg sync.WaitGroup
-
在每个要等待的 goroutine 中,调用
Add(1)方法以增加计数器:
Golang Spf13 Viper下载Go 配置库,使用 spf13/viper — 分层优先级(flag > env >file > KV > default),提供 BindPFlag/BindPFlags、SetEnvPrefix + SetEnvKeyReplace 等功能。
wg.Add(1)
-
当 goroutine 完成时,调用
Done()方法以减少计数器:wg.Done()
-
在主 goroutine 中,调用
Wait()方法阻塞,直到所有 goroutine 完成。wg.Wait()
实战案例
让我们考虑一个打印数字列表的并发程序。主 goroutine 创建一组 goroutine,每个 goroutine 打印列表中的一组数字。
package main
import (
"fmt"
"os"
"runtime"
"strconv"
"sync"
)
func main() {
// 创建 WaitGroup
var wg sync.WaitGroup
// 获取CPU核心数
numCores := runtime.NumCPU()
// 分配任务给 goroutine 组
for i := 0; i <p>在这个示例中,主 goroutine 创建了一个 <code>WaitGroup</code> 并调用 <code>Add(1)</code> 方法,为每个 goroutine 增加计数器。然后,它创建一组 goroutine,每个 goroutine调用 <code>printNumbers</code> 函数打印一组数字。</p><p><code>printNumbers</code> 函数每打印一组数字就会减少计数器。当所有 goroutine 完成时,主 goroutine 调用 <code>Wait()</code> 方法等待所有 goroutine 完成。这确保了列表中的所有数字都被打印出来,并且不存在goroutine竞争条件。</p>golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!










