Go 并发编程中的 Goroutine 调度：奇偶数循环次数的影响

本文探讨了 Go 并发编程中一个有趣的现象：当使用 channel 在 goroutine 间传递数据时，循环次数的奇偶性可能会影响程序的执行结果。通过分析示例代码，解释了这种现象背后的原因，并提供了确保 goroutine 完成的正确方法，避免程序提前退出导致数据丢失。

在 Go 语言的并发编程中，goroutine 和 channel 是两个核心概念。Goroutine 是一种轻量级的线程，而 channel 则用于 goroutine 之间的通信。然而，不恰当的使用可能会导致一些难以调试的问题。本文将通过一个具体的例子，深入探讨 Go 调度器的一些特性，以及如何避免潜在的并发问题。

问题现象

一段简单的 Go 代码，使用 channel 向一个 goroutine 发送数据，并在 goroutine 中打印接收到的数据。奇怪的是，当循环次数为奇数时，程序可以正常输出所有数据；而当循环次数为偶数时，最后一个数据却丢失了。

立即进入“豆包AI人工智官网入口”；

立即学习“豆包AI人工智能在线问答入口”；

package main

import "runtime"

func main() {
    c2 := make(chan int)

    go func() {
        for v := range c2 {
            println("c2 =", v, "numof routines:", runtime.NumGoroutine())
        }
    }()

    for i := 1; i <= 10001; i++ { // 尝试 10000 和 10001
        c2 <- i
        // runtime.Gosched() // 取消注释后，问题消失
    }
}

原因分析

这种现象的根本原因在于 Go 程序的退出机制和 goroutine 的调度。当 main 函数返回时，程序会立即终止，而不会等待任何 goroutine 完成。因此，goroutine 是否能够完成所有数据的接收和处理，取决于 Go 调度器的调度策略，以及一些随机因素和外部因素。

循环次数的奇偶性，可能只是影响调度器行为的一个因素。当循环次数为偶数时，main 函数可能更快地执行完毕，导致程序提前退出，从而丢失了最后一个数据。

解决方案

要解决这个问题，关键在于确保 goroutine 在 main 函数退出之前完成所有工作。以下是一些常用的方法：

豆包AI编程

豆包推出的AI编程助手

483

查看详情

1. 使用 sync.WaitGroup

   sync.WaitGroup 提供了一种等待一组 goroutine 完成的机制。main 函数可以使用 Add 方法增加计数器，在每个 goroutine 完成时调用 Done 方法减少计数器，并使用 Wait 方法阻塞，直到计数器变为零。

   package main
   
   import (
       "fmt"
       "runtime"
       "sync"
   )
   
   func main() {
       c2 := make(chan int)
       var wg sync.WaitGroup
   
       wg.Add(1) // 增加一个 goroutine 的计数器
       go func() {
           defer wg.Done() // goroutine 退出时减少计数器
           for v := range c2 {
               fmt.Println("c2 =", v, "numof routines:", runtime.NumGoroutine())
           }
       }()
   
       for i := 1; i <= 10000; i++ {
           c2 <- i
       }
       close(c2) // 关闭 channel，通知 goroutine 退出循环
       wg.Wait()   // 等待 goroutine 完成
   }

   登录后复制

   在这个例子中，close(c2) 非常重要。它关闭了 channel，通知 goroutine 没有更多的数据可以接收了，从而退出循环。如果没有 close(c2)，goroutine 将会一直阻塞，等待新的数据，导致程序死锁。

2. 使用 runtime.Gosched()

   在循环中调用 runtime.Gosched() 可以显式地让出 CPU 时间片，允许其他 goroutine 运行。虽然这可以缓解问题，但并不能保证 goroutine 一定能够完成，因此不推荐使用。

   package main
   
   import (
       "fmt"
       "runtime"
   )
   
   func main() {
       c2 := make(chan int)
   
       go func() {
           for v := range c2 {
               fmt.Println("c2 =", v, "numof routines:", runtime.NumGoroutine())
           }
       }()
   
       for i := 1; i <= 10000; i++ {
           c2 <- i
           runtime.Gosched() // 让出 CPU 时间片
       }
       close(c2)
   }

   登录后复制

   同样，close(c2) 也是必须的。

总结与建议

• Go 程序的退出机制不会等待 goroutine 完成，因此需要使用 sync.WaitGroup 等机制来确保 goroutine 在程序退出之前完成所有工作。
• 使用 channel 时，需要注意 channel 的关闭，避免 goroutine 永久阻塞。
• runtime.Gosched() 可以用于让出 CPU 时间片，但不能保证 goroutine 一定能够完成。
• 在并发编程中，需要仔细考虑各种边界情况，避免潜在的并发问题。

理解 Go 调度器的工作原理，以及正确使用并发原语，是编写健壮的并发程序的关键。希望本文能够帮助你更好地理解 Go 并发编程，并避免一些常见的陷阱。

以上就是Go 并发编程中的 Goroutine 调度：奇偶数循环次数的影响的详细内容，更多请关注php中文网其它相关文章！