首页 > 文章列表 > 掌握Go语言文档中的sync.WaitGroup函数实现并发控制

掌握Go语言文档中的sync.WaitGroup函数实现并发控制

go语言 WaitGroup sync
228 2023-11-03

掌握Go语言文档中的sync.WaitGroup函数实现并发控制,需要具体代码示例

在Go语言中实现并发控制是非常重要的,而sync包中的WaitGroup函数提供了一种简单和有效的方法来实现并发控制。本文将详细介绍如何使用sync.WaitGroup函数,同时提供具体的代码示例。

在并发编程中,常常需要等待所有的goroutine完成后才能继续执行后续的代码。使用sync.WaitGroup函数可以轻松实现这一点。sync包中的WaitGroup函数包含三个主要的方法:Add、Done和Wait。

首先,使用Add方法设置要等待的goroutine的数量。然后,在每个goroutine的最后使用Done方法表示该goroutine已经完成了。最后,使用Wait方法来等待所有的goroutine完成。

下面是一个简单的示例,展示了如何使用sync.WaitGroup函数实现并发控制:

package main

import (
    "fmt"
    "sync"
)

func main() {
    var wg sync.WaitGroup

    // 设置要等待的goroutine的数量
    wg.Add(2)

    go func() {
        defer wg.Done()
        // 模拟第一个任务
        fmt.Println("Executing Task 1")
    }()

    go func() {
        defer wg.Done()
        // 模拟第二个任务
        fmt.Println("Executing Task 2")
    }()

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

    // 所有的goroutine已经完成,执行后续的代码
    fmt.Println("All tasks completed")
}

在上述示例中,我们首先使用sync.WaitGroup类型的变量wg创建一个WaitGroup实例。然后,通过调用Add方法设置要等待的goroutine的数量,这里是2个goroutine。然后,我们创建两个goroutine,每个goroutine都会执行一些任务,并在最后通过调用Done方法来表示它已经完成了。最后,通过调用Wait方法,主goroutine将等待所有的goroutine完成。

上述示例中的任务非常简单,只是打印一些文本。但是可以根据实际需求修改任务的逻辑。

总结来说,sync.WaitGroup函数提供了一种简单和有效的方法来实现并发控制。通过合理设置WaitGroup的计数器,并在每个goroutine中调用Done方法,可以达到等待所有goroutine完成的效果。这对于并发任务的控制非常有用,可以确保在继续执行后续代码之前,所有的goroutine都已经完成了自己的任务。