本文探讨了 Go 并发编程中一个有趣的现象：当使用 channel 在 goroutine 间传递数据时，循环次数的奇偶性可能会影响程序的执行结果。通过分析示例代码，解释了这种现象背后的原因，并提供了确保 goroutine 完成的正确方法，避免程序提前退出导致数据丢失。
在 Go 语言的并发编程中，goroutine 和 channel 是两个核心概念。Goroutine 是一种轻量级的线程，而 channel 则用于 goroutine 之间的通信。然而，不恰当的使用可能会导致一些难以调试的问题。本文将通过一个具体的例子，深入探讨 Go 调度器的一些特性，以及如何避免潜在的并发问题。
问题现象
一段简单的 Go 代码，使用 channel 向一个 goroutine 发送数据，并在 goroutine 中打印接收到的数据。奇怪的是，当循环次数为奇数时，程序可以正常输出所有数据；而当循环次数为偶数时，最后一个数据却丢失了。
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 函数退出之前完成所有工作。以下是一些常用的方法：
-
使用 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 将会一直阻塞，等待新的数据，导致程序死锁。
-
使用 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 并发编程，并避免一些常见的陷阱。
