当前位置：首页 > 文章列表 > Golang > Go教程 > Golang 函数如何使用WaitGroup来协调并发任务？

Golang 函数如何使用WaitGroup来协调并发任务？

2024-10-06 18:18:56 0浏览收藏

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《Golang 函数如何使用WaitGroup来协调并发任务？》，文章讲解的知识点主要包括，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

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) 方法以增加计数器：
```
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 < numCores; i++ {
        from := i * 10
        to := (i + 1) * 10
        wg.Add(1)
        go printNumbers(wg, from, to)
    }

    // 等待所有 goroutine 完成
    wg.Wait()

    fmt.Println("所有数字已打印。")
}

// printNumbers 打印一组数字
func printNumbers(wg sync.WaitGroup, from, to int) {
    defer wg.Done()

    for i := from; i < to; i++ {
        fmt.Fprintf(os.Stdout, "%d ", i)
    }
}

在这个示例中，主 goroutine 创建了一个 WaitGroup 并调用 Add(1) 方法，为每个 goroutine 增加计数器。然后，它创建一组 goroutine，每个 goroutine调用 printNumbers 函数打印一组数字。

printNumbers 函数每打印一组数字就会减少计数器。当所有 goroutine 完成时，主 goroutine 调用 Wait() 方法等待所有 goroutine 完成。这确保了列表中的所有数字都被打印出来，并且不存在goroutine竞争条件。

以上就是《Golang 函数如何使用WaitGroup来协调并发任务？》的详细内容，更多关于的资料请关注golang学习网公众号！