go 线程同步WaitGroup结束示例

本文介绍如何利用Go语言中的sync.WaitGroup实现线程同步。通过实例展示如何等待多个goroutines执行完毕后再继续执行后续代码,适用于并发编程场景。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

WaitGroup用于线程同步,WaitGroup等待一组线程集合完成,才会继续向下执行。

示例中 main 等待 job1 job2 job3结束后, wg.Wait 才返回

wg.Add(3) 表明要等待3个 goroutines 

每调用一次 Done ,减少一个等待的goroutine计数

如果我不小心, wg.Add(4), 那么就程序就会永远无法结束

如果wg.Add(2) , 那么只要等到 wg.Done 两次 ,  main wg.Wait() 就返回了,

package main

import (
	"sync"
	"time"
	"log"// "github.com/sirupsen/logrus"
)

func job(wg *sync.WaitGroup, name string, sleepSecond time.Duration) {
	defer wg.Done()
	
	time.Sleep(time.Second*sleepSecond)
	log.Print(name + " done")
}

func main() {
	log.Print("hello world")

	var wg sync.WaitGroup
	wg.Add(3)
	go job(&wg, "job1", 1)
	go job(&wg, "job2", 2)
	go job(&wg, "job3", 1)

	wg.Wait()
	log.Print("all job done")
}
### 关于 `sync.WaitGroup` 和 `sync.WaitGroup{}` 的区别 #### 定义上的差异 `sync.WaitGroup` 是一个结构体类型,定义了一个用于等待一组 Goroutine 完成的同步机制[^2]。而 `sync.WaitGroup{}` 则是一个具体的实例化对象,表示创建了一个新的零值状态下的 `WaitGroup` 对象。 具体来说,`sync.WaitGroup` 只是一种类型的声明,只有当其被实例化为变量时才能真正发挥作用。例如: ```go var wg sync.WaitGroup // 声明并初始化一个 WaitGroup 类型的变量 ``` 或者直接使用匿名方式实例化: ```go wg := sync.WaitGroup{} // 创建一个新的 WaitGroup 实例 ``` 这两种写法都完成了对 `WaitGroup` 的初始化,但前者更常见于需要显式命名变量的情况下[^4]。 --- #### 使用场景分析 ##### 场景一:多个 Goroutine 并发执行后的统一等待 假设有一个程序启动了若干个 Goroutine 来处理不同的任务,并希望主线程能够等到这些子线程全部完成后才继续运行,则可以利用 `sync.WaitGroup` 提供的功能来实现这一需求[^1]。 下面给出一段简单的代码示例说明如何应用 `sync.WaitGroup`: ```go package main import ( "fmt" "sync" ) func worker(id int, wg *sync.WaitGroup) { defer wg.Done() // 每个工作单元结束前调用 Done 函数通知已完成 fmt.Printf("Worker %d starting\n", id) // Simulate some work being done. for i := 0; i < 3; i++ { fmt.Printf("Worker %d doing task %d\n", id, i+1) } fmt.Printf("Worker %d finished\n", id) } func main() { var wg sync.WaitGroup // 初始化 WaitGroup 结构体变量 numWorkers := 5 // 设定工作数量 wg.Add(numWorkers) // 设置要等待的任务数目 for i := 1; i <= numWorkers; i++ { go worker(i, &wg) // 启动每个 goroutine 工作者函数 } wg.Wait() // 主协程阻塞直到所有工作者完成 fmt.Println("All workers have completed their tasks.") } ``` 上述例子展示了通过传递指针形式管理不同 Goroutines 生命周期的过程。 ##### 场景二:局部作用域内的短期协作 如果只是在一个较小的作用范围内需要用到临时性的协调逻辑,那么可以直接采用如下简洁的方式构建和销毁 `WaitGroup` 实例: ```go func performTasksInParallel(taskCount int) { wg := sync.WaitGroup{} wg.Add(taskCount) for i := 0; i < taskCount; i++ { go func(index int){ defer wg.Done() processTask(index) }(i) } wg.Wait() } func processTask(index int) { fmt.Printf("Processing Task #%v...\n", index) } ``` 这里无需额外声明全局范围的变量名,仅需简单地定义即可满足即时的需求[^3]。 --- ### 总结 无论是选择单独定义还是立即创建的形式,核心在于理解何时何地去运用这种强大的工具以简化复杂的并发流程控制问题。两者并无本质优劣之分,更多取决于实际编码习惯和个人偏好等因素影响下做出的选择。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值