golang 等待协程结束

Golang等待协程结束的几种方法

并发编程是Go语言的一大特色，Go协程的轻量级和高效性使得并发操作变得非常简单。然而，在实际开发中，我们经常需要等待所有协程执行完成后再继续进行后续操作。下面将介绍几种常用的在Go语言中等待协程结束的方法。

1. 使用计数器

一种常见的方式是使用计数器来记录当前活跃的协程数量。每个协程在执行完毕后将计数器减一，当计数器为零时，表示所有协程都执行完毕。下面是一个示例代码：

``` package main import ( "fmt" "sync" ) func main() { var wg sync.WaitGroup count := 5 wg.Add(count) for i := 0; i < count; i++ { go func(i int) { defer wg.Done() fmt.Printf("协程 %d 执行完毕\n", i) }(i) } wg.Wait() fmt.Println("所有协程执行完毕") } ```

在上述代码中，我们定义了一个sync.WaitGroup类型的变量wg，并通过调用Add方法初始化计数器。接着，在每个协程执行完毕后调用wg.Done()方法将计数器减一。最后，我们调用wg.Wait()方法阻塞主协程，直到计数器为零，表示所有协程都执行完毕。

2. 使用通道

另一种常见的方式是使用通道来等待协程执行完成。我们可以创建一个带有缓冲区的通道，并通过在每个协程执行完毕后往通道中发送信号来表示协程已完成。下面是一个示例代码：

``` package main import ( "fmt" "sync" ) func main() { var wg sync.WaitGroup count := 5 ch := make(chan struct{}, count) for i := 0; i < count; i++ { wg.Add(1) go func(i int) { defer wg.Done() fmt.Printf("协程 %d 执行完毕\n", i) ch <- struct{}{} }(i) } wg.Wait() close(ch) for range ch { // 等待所有协程完成 } fmt.Println("所有协程执行完毕") } ```

在上述代码中，我们创建了一个带有缓冲区的通道ch，并使用for range循环从通道中接收信号，直到通道被关闭，表示所有协程都执行完毕。这种方法相较于计数器的方式更加灵活，可以在通道接收过程中进行一些其他操作。

3. 使用sync.WaitGroup和context.Context

在Go语言1.7版本开始，标准库中引入了context包，用于在协程之间传递请求的截止日期、取消信号以及其他与请求有关的值。我们可以结合sync.WaitGroup和context.Context来等待协程执行完成。下面是一个示例代码：

``` package main import ( "context" "fmt" "sync" "time" ) func main() { ctx, cancel := context.WithCancel(context.Background()) var wg sync.WaitGroup count := 5 wg.Add(count) for i := 0; i < count; i++ { go func(ctx context.Context, i int) { defer wg.Done() fmt.Printf("协程 %d 执行完毕\n", i) // 模拟耗时操作 time.Sleep(time.Second * time.Duration(i+1)) }(ctx, i) } wg.Wait() cancel() fmt.Println("所有协程执行完毕") } ```

在上述代码中，我们通过调用context.WithCancel函数创建了一个带有取消功能的上下文类型ctx，并通过defer cancel()语句确保在main函数退出之前取消所有协程。每个协程在执行完毕后自动取消操作。这种方式可以更加精确地控制协程的执行时间，并在需要时取消协程的执行。

总结

本文介绍了在Go语言中等待协程结束的几种常见方法，包括使用计数器、使用通道以及结合sync.WaitGroup和context.Context。根据实际需求选择适合的方式，可以提高程序的并发性能和灵活性。无论是开发高性能的服务器程序还是处理大规模数据的并发任务，Go语言都提供了丰富的工具和库来支持并发编程。希望本文对您在使用Go语言进行并发编程时有所帮助。