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

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

GoLang 函数中使用 WaitGroup 协调并发任务

简介

当在 Go 中使用并发特性时，经常需要协调多个并发执行的任务。WaitGroup 是一种同步机制，用于等待一组并发的 goroutine 完成他们的任务。

WaitGroup 的用法

WaitGroup 是一个用于管理并发的 goroutine 组的数据结构。它包含两个主要字段：

• Counter：一个计数器，跟踪正在等待完成的 goroutine 数。
• Wait：一个 channel，用于阻塞直到所有 goroutine 完成。

要使用 WaitGroup，请执行以下步骤：

立即学习“go语言免费学习笔记（深入）”；

1. 创建一个新的 WaitGroup：

   import "sync/atomic"
   
   var wg sync.WaitGroup

   登录后复制

2. 在每个要等待的 goroutine 中，调用 Add(1) 方法以增加计数器：

   wg.Add(1)

   登录后复制

3. 当 goroutine 完成时，调用 Done() 方法以减少计数器：

   wg.Done()

   登录后复制

4. 在主 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来协调并发任务？的详细内容，更多请关注php中文网其它相关文章！