Golang中的并发控制和Go WaitGroup

Golang中的并发控制和Go WaitGroup

在Golang中，我们可以使用goroutine实现并发执行任务。然而，在某些情况下，我们需要控制并发执行的数量，以避免资源过度消耗或者出现并发竞争的问题。Golang中提供了一些方法来实现并发控制，其中最常用的是使用Go WaitGroup。

Go WaitGroup是一个计数信号量，用于等待一组goroutine完成执行。当我们启动一组goroutine时，我们可以使用WaitGroup来追踪这些goroutine的状态，并在所有goroutine完成后进行下一步的操作。

下面我们通过一个具体的代码示例来演示Golang中的并发控制和Go WaitGroup的使用。

package main

import (
    "fmt"
    "sync"
    "time"
)

func main() {
    numWorkers := 5
    var wg sync.WaitGroup

    for i := 0; i < numWorkers; i++ {
        wg.Add(1) // 每启动一个goroutine，等待组加1
        go worker(i, &wg)
    }

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

    fmt.Println("All workers have finished")
}

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done() // goroutine执行完毕，等待组减1
    
    fmt.Printf("Worker %d started
", id)
    time.Sleep(time.Second) // 模拟耗时操作
    fmt.Printf("Worker %d finished
", id)
}

在这个示例中，我们有5个goroutine，每个执行的任务是调用worker函数。首先，我们定义了一个WaitGroup变量wg，并在主函数中启动了5个goroutine。每个goroutine都会调用worker函数，并通过传递指向wg的指针来操作WaitGroup。

在worker函数中，我们使用defer wg.Done()来在goroutine执行完毕后减少等待组的计数。这意味着每个goroutine在完成后都会调用Done()函数来通知WaitGroup。然后，我们在每个worker函数中加入了一些模拟的耗时操作，以便观察并发执行的效果。

在主函数中，我们调用wg.Wait()来等待所有goroutine完成。这将导致主函数在所有goroutine执行完毕后阻塞，直到WaitGroup的计数归零。

运行上述代码，你将看到类似以下的输出：

Worker 0 started
Worker 1 started
Worker 2 started
Worker 3 started
Worker 4 started
Worker 3 finished
Worker 2 finished
Worker 0 finished
Worker 1 finished
Worker 4 finished
All workers have finished

从输出结果可以看出，所有的goroutine都按照并发的方式启动和运行，并在完成后通知主函数。

通过使用Go WaitGroup，我们可以很方便地控制并发执行的数量，并在所有goroutine完成后进行后续操作。这对于处理大规模并发任务或限制资源消耗非常有帮助。

总结起来，Golang中的并发控制和Go WaitGroup是实现并发编程的重要工具。我们可以使用WaitGroup来追踪和控制一组goroutine的执行，确保并发操作的正确性和稳定性。通过这种方式，我们可以更好地利用多核处理器和资源，并提高程序的执行效率。