Go WaitGroup的使用方式及实例详解

Go语言是一种以并发编程为核心的编程语言，在处理并发任务时非常高效。在Go语言的标准库中，有一个非常好用的并发工具——WaitGroup。本文将详细介绍WaitGroup的使用方式，并通过实例来演示其具体应用。

首先，我们需要了解一下WaitGroup的作用。WaitGroup用于等待一组并发任务完成后再继续执行后续的代码。通常情况下，我们需要在主协程等待其他协程完成某个任务再继续向下执行时，就可以使用WaitGroup。

在Go语言的sync包下，有一个WaitGroup结构体，它具有3个主要的方法，分别是Add()、Done()和Wait()。下面我们逐一介绍它们的作用。

首先是Add()方法，它用于增加等待的任务数。我们可以通过调用Add()方法来告诉WaitGroup我们有多少个任务需要等待完成。具体的调用方式是wg.Add(n)，其中n表示需要等待的任务数量。例如，如果我们有10个任务需要等待完成，就可以调用wg.Add(10)。

然后是Done()方法，它用于标志当前任务已经完成。每个任务完成时，都需要调用Done()方法告诉WaitGroup当前任务已经完成。具体的调用方式是wg.Done()。通常情况下，我们会在任务完成后的defer语句中调用Done()方法，以确保任务完成后一定会调用Done()方法。

最后是Wait()方法，它用于阻塞主协程，直到所有任务都已经完成。当主协程需要等待一组任务完成后再继续执行后续的代码时，可以调用Wait()方法。具体的调用方式是wg.Wait()。Wait()方法会一直阻塞，直到所有任务都已经完成。

下面我们通过一个实例来演示WaitGroup的具体应用。假设我们有一个任务列表，我们需要并发地处理这些任务。在每个任务完成后，我们都会将结果写入一个结果列表。最后，我们需要等待所有任务都完成后，才能打印出结果列表。代码如下所示：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup
    taskList := []string{"task1", "task2", "task3", "task4"}

    // 创建结果列表
    resultList := make([]string, len(taskList))

    // 遍历任务列表，开启多个协程处理任务
    for i, task := range taskList {
        wg.Add(1) // 增加等待的任务数

        go func(index int, task string) {
            defer wg.Done() // 标志任务已经完成

            // 模拟任务处理过程
            result := fmt.Sprintf("task %s processed", task)

            // 将结果写入结果列表
            resultList[index] = result
        }(i, task)
    }

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

    // 打印结果列表
    for _, result := range resultList {
        fmt.Println(result)
    }
}

在上面的代码中，首先我们创建了一个WaitGroup对象wg。然后，我们遍历任务列表，针对每个任务调用了一次wg.Add(1)，表示有一个任务需要等待。在每个任务的处理函数中，我们在任务完成后调用了一次wg.Done()。最后，我们在主协程中调用了wg.Wait()来等待所有任务都完成。

通过这样的设计，我们可以并发地处理任务，并且确保在所有任务都完成后，再继续主协程的后续执行。最后，我们打印出了结果列表，验证了任务的并发处理和结果的正确性。

总结来说，WaitGroup是Go语言中非常实用的并发工具。通过合理使用WaitGroup，我们可以更好地管理并发任务，确保任务的顺序和正确性。希望本文能够帮助读者更好地理解和应用WaitGroup。

以上是Go WaitGroup的使用方式及实例详解的详细内容。更多信息请关注PHP中文网其他相关文章！