深度解析sync WaitGroup源码及其实现原理

最新推荐文章于 2024-08-04 12:01:31 发布

猿面试

最新推荐文章于 2024-08-04 12:01:31 发布

阅读量1.2k

点赞数 3

分类专栏：面试源码博客文章标签： go golang

本文链接：https://blog.csdn.net/qq_41066066/article/details/113986923

版权

本文介绍了Go语言中的sync.WaitGroup，用于解决并发等待问题。WaitGroup包括Add、Done和Wait三个方法。Add设置计数值，Done减少计数，Wait在计数为0时释放阻塞。文章详细剖析了WaitGroup的源码，包括数据结构、Add方法实现、Done方法以及Wait方法的工作原理。

摘要由CSDN通过智能技术生成

在这里插入图片描述

WaitGroup介绍

waitGroup ，也是在go语言并发中比较常用的语法，所以在这里我们一起剖析 waitGroup 的使用方式及其源码解读。

WaitGroup 也是sync 包下一份子，用来解决任务编排的一个并发原语。它主要解决了并发-等待问题：比如现在有三个goroutine，分别为goroutineA，goroutineB，goroutineC，而goroutineA需要等待goroutineB和goroutineC这一组goroutine全部执行完毕后，才可以执行后续业务逻辑。此时就可以使用 WaitGroup 轻松解决。

在这个场景中，goroutineA为主goroutine，goroutineB和goroutineC为子goroutine。goroutineA则需要在检查点(checkout point) 等待goroutineB和goroutineC全部执行完毕，如果在执行任务的goroutine还没全部完成，那么goroutineA就会阻塞在检查点，直到所有goroutine都完成后才能继续执行。

代码实现：

package main

import (
  "fmt"
  "sync"
)

func goroutineB(wg *sync.WaitGroup) {
   
  defer wg.Done()
  fmt.Println("goroutineB Execute")
  time.Sleep(time.Second)
}

func goroutineC(wg *sync.WaitGroup) {
   
  defer wg.Done()
  fmt.Println("goroutineC Execute")
  time.Sleep(time.Second)
}

func main() {
   
  var wg sync.WaitGroup
  wg.Add(2