文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

Golang中的同步工具Sync.WaitGroup详解

2024-11-30 13:00

关注

sync.WaitGroup

sync.WaitGroup作用是等待一组goroutine执行完成,当使用多个goroutine处理任务的时候,如果要等待所有的goroutine都执行完成后再执行下一步操作,就可以使用sync.WaitGroup来实现。

sync.WaitGroup有三个方法:

  1. Add(delta int):增加或减少等待goroutine的数量,delta可以为负数;
  2. Done():标记goroutine已执行完成,等价于Add(-1);
  3. Wait():等待所有的goroutine执行完成。

使用方法和示例

假如有n个任务需要处理,可以先创建一个WaitGroup,用于等待所有的任务完成:

var wg sync.WaitGroup

在处理每一个任务之前,调用一次Add方法:

wg.Add(1)

在每一个任务处理完成之后,调用一次Done方法:

wg.Done()

最后,在等待所有的任务完成之前,需要调用一次Wait方法:

wg.Wait()

具体示例代码如下:

package main

import (
	"log"
	"sync"
	"time"
)

func main() {
	// 创建一个等待组
	var wg sync.WaitGroup

	for i := 1; i <= 5; i++ {
		// 每一个任务开始前, 增加1
		wg.Add(1)

		go func(index int) {
			// 处理业务逻辑
			//...
			time.Sleep(1 * time.Second)
			log.Printf("第 %d 个goroutine执行完了", index)
			wg.Done()
		}(i)
	}
	// 等待所有的任务完成
	wg.Wait()
	log.Println("所有goroutine都执行完了")
}

使用过程中需要注意的点

  1. Done和Add方法的顺序非常重要,Done方法必须在goroutine最后执行,否则可能会导致计数器无法正确的达到0;
  2. 如果使用Add或Done方法将wg维护的计数更改成了负数会导致panic;
  3. Wait方法会一直阻塞,直到所有的goroutine都执行完成为止(wg维护的计数达到0),因此在使用时需要小心;
  4. sync.WaitGroup是开箱即用的,并且是并发安全的。
来源:今日头条内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     813人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     354人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     318人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     435人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     224人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯