Go语言利用协程实现等待机制的方法

在Go语言中，协程（goroutine）是一种轻量级的并发执行单元，它使得并发编程变得更加简单和高效。然而，在实际应用中，我们常常需要等待多个协程完成任务后再进行下一步操作，这就需要用到等待机制。本文将介绍Go语言利用协程实现等待机制的方法。

Go语言中常用的等待机制是通过sync.WaitGroup来实现的。sync.WaitGroup是一个结构体，它提供了三个方法：Add、Done和Wait。Add方法用于增加等待的协程数量，Done方法用于标记一个协程完成任务，Wait方法用于阻塞当前协程，直到所有等待的协程都完成任务。

下面是一个简单的示例代码，演示了如何使用sync.WaitGroup实现等待机制：

package main

import (
    "fmt"
    "sync"
)

func worker(id int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Printf("Worker %d starting\n", id)
    // 模拟任务执行
    for i := 0; i < 3; i++ {
        fmt.Printf("Worker %d: %d\n", id, i)
    }
    fmt.Printf("Worker %d done\n", id)
}

func main() {
    var wg sync.WaitGroup
    for i := 1; i <= 3; i++ {
        wg.Add(1)
        go worker(i, &wg)
    }
    wg.Wait()
    fmt.Println("All workers done")
}

在上述代码中，我们首先创建了一个sync.WaitGroup实例wg，然后使用wg.Add方法增加了3个等待的协程数量。接着，我们使用go关键字启动了3个协程，每个协程都会执行worker函数。在worker函数中，我们使用defer wg.Done标记协程完成任务。最后，我们使用wg.Wait方法阻塞当前协程，直到所有等待的协程都完成任务。

除了sync.WaitGroup，Go语言还提供了其他的等待机制，例如channel。channel可以用于协程之间的通信和同步，通过在channel上发送和接收数据，可以实现协程之间的等待和协作。

Go语言提供了多种方法来实现协程的等待机制，开发者可以根据具体的需求选择合适的方法。通过合理使用等待机制，可以更好地控制协程的执行顺序，提高程序的并发性能和效率。