进击的协程，谁是舞者的领导者？——Go sync.WaitGroup：弹性编排，掌控并行

后端

2023-11-13 16:00:20

并发协奏曲：用 WaitGroup 掌控优雅舞步

协奏曲的序章

并发编程的世界如同一场壮观的交响乐，无数协程宛如乐队中的演奏者，各司其职，奏响美妙的乐章。然而，为了让协程之间的合作井然有序，我们需要一位指挥家来协调它们。

Go语言为我们提供了这样一位指挥家——sync.WaitGroup 。WaitGroup如同一个计数器，它记录了需要等待的协程数量。当所有协程都完成任务时，WaitGroup的计数器归零，从而释放等待的协程继续执行。

掌控并行洪流的艺术

WaitGroup的使用很简单，只需三个步骤：

创建 WaitGroup 对象
在每个协程中调用 WaitGroup.Add 方法 ，增加需要等待的协程数量
在主协程中调用 WaitGroup.Wait 方法 ，等待所有协程完成任务

掌握了这三个步骤，你便能轻松掌控协程之间的并行执行，让它们有序完成任务。

WaitGroup 的实战演练：提升性能风暴

为了更深入理解 WaitGroup 的妙用，让我们通过一个实际案例来领略它的魅力。假设我们要处理一个包含 1000 个文件的目录，并希望并发读取每个文件的内容。

package main

import (
    "fmt"
    "io/ioutil"
    "sync"
)

func main() {
    // 创建 WaitGroup 对象
    var wg sync.WaitGroup

    // 并发读取 1000 个文件的内容
    for i := 0; i < 1000; i++ {
        wg.Add(1) // 增加需要等待的协程数量

        go func(i int) {
            // 读取文件内容
            data, err := ioutil.ReadFile(fmt.Sprintf("file-%d.txt", i))
            if err != nil {
                fmt.Printf("Error reading file: %s\n", err)
            }

            // 任务完成后，调用 WaitGroup.Done 方法，减少需要等待的协程数量
            wg.Done()
        }(i)
    }

    // 等待所有协程完成任务
    wg.Wait()

    // 所有协程都已完成任务，可以继续执行主协程
    fmt.Println("All files have been processed.")
}