在go语言并发编程中，新手常遇到的一个问题是，即使在协程（goroutine）中使用了`fmt.println`，程序却没有任何输出。这通常是由于主协程在子协程完成执行前便已退出，导致整个程序终止。本文将深入探讨这一现象的根本原因，并提供使用`sync.waitgroup`这一go语言标准库提供的强大工具来正确同步协程，确保所有并发任务都能顺利完成并输出结果的专业解决方案。
Go协程输出问题解析
当我们在Go程序中启动一个或多个协程（goroutine）来执行并发任务时，有时会发现这些协程内部的打印语句（如fmt.Println）并没有按预期输出。考虑以下示例代码：
package main
import "fmt"
func f(msg string) {
fmt.Println(msg)
}
func main() {
go f("goroutine")
go func(msg string) {
fmt.Println(msg)
}("going")
// main函数在此处直接返回
}运行上述代码，你可能会发现控制台没有任何输出。如果移除go关键字，程序则会正常打印"goroutine"和"going"。
问题根源：主协程的生命周期
这种现象的根本原因在于Go程序的执行机制。main函数本身也是一个协程，被称为主协程（main goroutine）。当主协程执行完毕并退出时，Go运行时会立即终止整个程序，而不管此时是否有其他非主协程仍在运行或等待执行。
在上述示例中，main函数启动了两个新的协程，然后立即到达函数末尾并返回。由于协程的调度是异步的，这两个新启动的协程可能还没有来得及被Go调度器执行，或者即使开始执行也未能及时完成其内部的fmt.Println操作，主协程就已经退出了，从而导致程序没有任何输出。
立即学习“go语言免费学习笔记（深入）”；
不推荐的临时解决方案：time.Sleep
为了“看到”协程的输出，一种常见的初学者尝试是让主协程暂停一段时间，以期望给子协程足够的时间来完成任务。例如：
package main
import (
"fmt"
"time" // 导入 time 包
)
func f(msg string) {
fmt.Println(msg)
}
func main() {
go f("goroutine")
go func(msg string) {
fmt.Println(msg)
}("going")
time.Sleep(2 * time.Second) // 暂停2秒
}这段代码在大多数情况下可能会打印出预期的内容。然而，这种做法被认为是不良实践，原因如下：
- 不确定性： time.Sleep引入了一个任意的、硬编码的延迟。我们无法保证2秒（或任何其他固定时间）对于所有协程来说都足够长。在不同的系统负载、处理器速度或协程执行的任务复杂性下，协程可能需要更长的时间才能完成。
- 效率低下： 如果协程在很短的时间内就完成了，那么主协程不必要的等待会浪费系统资源。
- 竞态条件： 这种方式实际上是在引入一个隐式的竞态条件。你无法确保协程在time.Sleep结束前一定完成。
因此，time.Sleep不应作为协程同步的解决方案。
专业的协程同步方案：sync.WaitGroup
Go语言标准库提供了sync包，其中包含了一系列用于并发编程的同步原语。sync.WaitGroup是其中最常用且最适合解决此类问题的工具。它允许我们等待一组协程完成执行。
sync.WaitGroup的工作原理如下：
- Add(delta int): 用于设置或增加等待协程的数量。通常在启动每个协程之前调用wg.Add(1)，表示有一个新的协程需要等待。
- Done(): 由每个协程在完成其任务后调用，表示该协程已完成。它会递减内部计数器。
- Wait(): 由主协程调用，它会阻塞当前协程（主协程），直到内部计数器归零，即所有注册的协程都调用了Done()。
下面是使用sync.WaitGroup重构后的示例代码：
package main
import (
"fmt"
"sync" // 导入 sync 包
)
// f 函数现在接收一个 WaitGroup 指针
func f(msg string, wg *sync.WaitGroup) {
defer wg.Done() // 确保协程完成时调用 Done()
fmt.Println(msg)
}
func main() {
var wg sync.WaitGroup // 声明一个 WaitGroup 变量
// 启动第一个协程
wg.Add(1) // 增加计数器，表示有一个协程需要等待
go f("goroutine", &wg)
// 启动第二个协程
wg.Add(1) // 再次增加计数器
go func(msg string) {
defer wg.Done() // 匿名协程也确保调用 Done()
fmt.Println(msg)
}("going")
wg.Wait() // 阻塞主协程，直到所有注册的协程都完成
fmt.Println("所有协程已完成。") // 确认主协程在等待后才退出
}代码解析
- var wg sync.WaitGroup: 在main函数中声明一个WaitGroup变量。
- wg.Add(1): 在启动每个协程之前，调用wg.Add(1)。这会使WaitGroup的内部计数器增加1。这意味着主协程现在知道它需要等待一个额外的协程完成。
- defer wg.Done(): 在f函数内部和匿名协程的开始处，使用defer wg.Done()。defer语句确保无论协程如何退出（正常完成或发生panic），wg.Done()都会被调用。wg.Done()会使WaitGroup的内部计数器减1。
- wg.Wait(): 在main函数的末尾调用wg.Wait()。这会阻塞主协程，直到WaitGroup的内部计数器变为零。只有当所有之前通过Add注册的协程都调用了Done()之后，wg.Wait()才会解除阻塞，主协程才能继续执行并最终退出。
通过这种方式，我们确保了主协程会一直等待，直到所有子协程都明确地表示它们已经完成任务，从而避免了主协程过早退出的问题，保证了所有fmt.Println都能被执行并输出。
总结与最佳实践
当你在Go语言中遇到协程不按预期打印输出的问题时，几乎总是因为主协程没有等待子协程完成就退出了。正确的解决方案是使用sync.WaitGroup进行同步。
核心原则：
- 启动协程前调用wg.Add(1)。
- 协程完成任务时调用wg.Done()（通常使用defer确保）。
- 主协程调用wg.Wait()等待所有协程完成。
掌握sync.WaitGroup是Go并发编程中的一项基本技能，它能帮助你构建健壮、可预测且高效的并发程序。避免使用time.Sleep进行协程同步，因为它引入了不确定性和潜在的竞态条件。
