Golang并发编程进阶：深入理解WaitGroup-Golang-PHP中文网

Golang并发编程进阶：深入理解WaitGroup

王林

发布： 2023-09-27 08:36:21

原创

1745人浏览过

golang并发编程进阶：深入理解waitgroup

Golang并发编程进阶：深入理解WaitGroup

引言：
并发编程是Golang最强大的功能之一。在并行执行多个任务时，一种常见的场景是等待所有任务完成后再继续执行下一步操作。Golang中的sync包提供了一个很好的工具WaitGroup来解决这个问题。本文将深入介绍WaitGroup的使用方法，并给出具体的代码示例。

一、什么是WaitGroup？
WaitGroup是Golang中sync包中的一个结构体，它提供了一种简单而有效的机制来等待所有的goroutine完成任务。WaitGroup内部维护了一个计数器，可以通过Add()方法增加计数器的值，Done()方法减少计数器的值，Wait()方法用于阻塞当前线程，直到计数器归零。

二、WaitGroup的基本用法
首先，我们需要导入sync包：

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

import "sync"

登录后复制

然后，我们可以按照以下步骤使用WaitGroup：

创建WaitGroup对象：
```
var wg sync.WaitGroup
```
登录后复制
使用Add()方法设置需要等待的任务数量：
```
wg.Add(2)
```
登录后复制
启动goroutine执行任务：

PHP 网络编程技术与实例(曹衍龙)
PHP网络编程技术详解由浅入深，全面、系统地介绍了PHP开发技术，并提供了大量实例，供读者实战演练。另外，笔者专门为本书录制了相应的配套教学视频，以帮助读者更好地学习本书内容。这些视频和书中的实例源代码一起收录于配书光盘中。本书共分4篇。第1篇是PHP准备篇，介绍了PHP的优势、开发环境及安装；第2篇是PHP基础篇，介绍了PHP中的常量与变量、运算符与表达式、流程控制以及函数；第3篇是进阶篇，介绍

386

查看详情
```
go task1()
go task2()
```
登录后复制

在任务结束之前调用Done()方法减少计数器的值：

func task1() {
 defer wg.Done()
 // 执行task1的操作
}

func task2() {
 defer wg.Done()
 // 执行task2的操作
}

登录后复制

最后，在需要等待所有任务完成的地方调用Wait()方法：
```
wg.Wait()
```
登录后复制
这样，主线程将等待所有任务完成后再继续执行下一步操作。

三、WaitGroup的实际应用示例
下面我们通过一个具体的示例来说明WaitGroup的用法。

假设我们有一个需求，要并发下载多个网络图片并保存到本地。当所有图片下载完成后，我们需要进行后续处理。代码示例如下：

package main

import (
    "fmt"
    "io"
    "net/http"
    "os"
    "sync"
)

var wg sync.WaitGroup

func main() {
    // 假设有3个图片需要下载
    imageUrls := []string{
        "https://example.com/image1.jpg",
        "https://example.com/image2.jpg",
        "https://example.com/image3.jpg",
    }

    // 设置需要等待的任务数量
    wg.Add(len(imageUrls))

    // 并发下载图片
    for _, imageUrl := range imageUrls {
        go downloadImage(imageUrl)
    }

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

    fmt.Println("所有图片下载完成，进行后续处理")
}

func downloadImage(imageUrl string) {
    defer wg.Done()

    // 发送HTTP GET请求获取图片数据
    resp, err := http.Get(imageUrl)
    if err != nil {
        fmt.Printf("下载图片失败: %v
", err)
        return
    }
    defer resp.Body.Close()

    // 创建本地文件
    fileName := imageUrl[strings.LastIndex(imageUrl, "/")+1:]
    imgFile, err := os.Create(fileName)
    if err != nil {
        fmt.Printf("创建图片文件失败: %v
", err)
        return
    }
    defer imgFile.Close()

    // 将图片数据保存到本地文件
    _, err = io.Copy(imgFile, resp.Body)
    if err != nil {
        fmt.Printf("保存图片失败: %v
", err)
        return
    }

    fmt.Printf("图片下载成功: %v
", imageUrl)
}

登录后复制

在上面的示例代码中，我们首先定义了一个包级变量wg用于管理等待任务完成的计数器。在主函数中，我们设置了等待任务数量为图片URL的数量，然后并发启动每个图片下载任务。每个任务完成后，调用wg.Done()方法来减少计数器的值。最后，调用wg.Wait()方法来等待所有任务完成。当所有图片下载完成后，继续执行后续处理。

总结：
本文详细介绍了Golang中WaitGroup的使用方法，并给出了一个具体的多任务并发下载图片的示例。通过深入理解WaitGroup的使用，可以更好地掌握Golang的并发编程能力，提高程序的性能和效率。在实际应用中，我们可以根据具体需求灵活运用WaitGroup来管理和等待多个goroutine的完成。

以上就是Golang并发编程进阶：深入理解WaitGroup的详细内容，更多请关注php中文网其它相关文章！