本文深入探讨了go语言中，当主协程（main goroutine）过早退出时，子协程内部的`defer`语句为何可能不被执行的问题。我们将通过具体代码示例，分析其根本原因在于并发场景下的竞态条件，而非`defer`的放置或调度问题。文章将重点介绍如何利用`sync.waitgroup`或通道（channels）等go语言提供的同步原语，确保子协程能够完整执行其逻辑，包括所有延迟调用，从而构建健壮的并发应用程序。
理解Go语言中的defer与Goroutine生命周期
在Go语言中，defer语句用于延迟函数的执行，直到包含它的函数即将返回时才被调用。这些延迟调用会按照后进先出（LIFO）的顺序执行。然而，当defer语句存在于一个新创建的Goroutine（子协程）中时，其行为可能会出乎意料，尤其是在主协程（main函数）过早退出的情况下。
考虑以下示例代码：
package main
import (
"fmt"
"time"
)
func main() {
fmt.Println("1")
defer fmt.Println("-1") // 属于main协程的defer
go func() {
fmt.Println("2")
defer fmt.Println("-2") // 属于子协程的defer
time.Sleep(9 * time.Second) // 模拟长时间工作
}()
time.Sleep(1 * time.Second) // 主协程等待1秒
fmt.Println("3")
// main协程在此处退出
}运行上述代码，我们可能会观察到如下输出：
1 2 3 -1
预期的输出可能包括子协程的-2，即：1 2 3 -2 -1。然而，实际输出中子协程的defer语句fmt.Println("-2")并未被执行。这并非defer放置错误，也不是Go调度器的问题，而是由于Go程序的生命周期管理。
立即学习“go语言免费学习笔记（深入）”；
当main函数执行完毕并退出时，整个Go程序会终止。这意味着，如果main函数在其他Goroutine完成其工作之前就退出了，那么那些尚未完成的Goroutine（包括其内部的defer调用）将被强制终止，而不会有机会执行它们的defer语句。在上面的例子中，主协程仅等待了1秒，而子协程被设定为工作9秒，因此主协程在子协程完成前就退出了。
错误的解决方案与正确理解
一些开发者可能会尝试使用runtime.Gosched()来尝试“让出CPU时间”，期望能解决此问题。例如：
// ... (代码同上) ...
time.Sleep(1 * time.Second)
fmt.Println("3")
// runtime.Gosched() // 尝试让出CPU，但无法保证子协程完成
}runtime.Gosched()的作用是让当前Goroutine放弃CPU，允许其他Goroutine运行。虽然在某些特定调度环境下，这可能“偶然”导致子协程获得足够时间完成，但它并不能保证子协程一定会完成，更不是解决这种竞态条件的可靠方法。其效果是不可预测且依赖于运行时实现细节的。
根本问题在于主协程与子协程之间缺乏明确的同步机制。要确保子协程能够完整执行并调用其defer语句，主协程必须等待子协程完成。
使用sync.WaitGroup进行同步
Go语言提供了sync包，其中包含了一系列用于并发同步的工具。sync.WaitGroup是解决此类问题的最常用且推荐的方法之一。它允许我们等待一组Goroutine完成。
sync.WaitGroup有三个主要方法：
- Add(delta int)：增加内部计数器，通常在启动Goroutine之前调用。
- Done()：减少内部计数器，通常在Goroutine即将完成时调用（常与defer配合使用）。
- Wait()：阻塞直到内部计数器归零，表示所有被Add的Goroutine都已完成。
下面是使用sync.WaitGroup修改后的示例代码：
package main
import (
"fmt"
"sync" // 引入sync包
"time"
)
func main() {
var wg sync.WaitGroup // 声明一个WaitGroup
fmt.Println("1")
defer fmt.Println("-1") // 属于main协程的defer
wg.Add(1) // 启动一个子协程前，将计数器加1
go func() {
defer wg.Done() // 子协程结束前，将计数器减1，确保无论如何都会被调用
fmt.Println("2")
defer fmt.Println("-2") // 属于子协程的defer
time.Sleep(9 * time.Second) // 模拟长时间工作
fmt.Println("子协程完成工作")
}()
time.Sleep(1 * time.Second) // 主协程等待1秒
fmt.Println("3")
wg.Wait() // 阻塞main协程，直到所有wg.Add(1)对应的wg.Done()被调用，即子协程完成
fmt.Println("所有协程已完成，main协程即将退出。")
}运行上述代码，我们将得到预期的输出：
1 2 3 子协程完成工作 -2 所有协程已完成，main协程即将退出。 -1
通过wg.Wait()，主协程被强制等待子协程完成其所有任务，包括执行其内部的defer fmt.Println("-2")语句。当子协程调用wg.Done()后，wg.Wait()解除阻塞，主协程继续执行，最终调用自己的defer fmt.Println("-1")并退出。
使用通道（Channels）进行同步
除了sync.WaitGroup，通道（channels）也是Go语言中常用的同步机制。如果子协程需要向主协程发送结果或仅仅发送一个完成信号，通道会是更合适的选择。
例如，通过一个空结构体通道来发送完成信号：
package main
import (
"fmt"
"time"
)
func main() {
fmt.Println("1")
defer fmt.Println("-1")
done := make(chan struct{}) // 创建一个用于发送完成信号的通道
go func() {
fmt.Println("2")
defer fmt.Println("-2")
time.Sleep(9 * time.Second)
fmt.Println("子协程完成工作")
close(done) // 关闭通道，通知主协程子协程已完成
}()
time.Sleep(1 * time.Second)
fmt.Println("3")
<-done // 阻塞main协程，直到从done通道接收到信号（或通道被关闭）
fmt.Println("所有协程已完成，main协程即将退出。")
}此方法也能达到相同的同步效果，确保子协程在主协程退出前完成。
总结与注意事项
- defer与函数作用域： defer语句的执行与它所在的函数（或Goroutine的执行函数）的退出紧密相关。
- 主协程的生命周期： main函数是程序的入口点，当main函数退出时，整个Go程序都会终止，无论其他Goroutine是否完成。
- 并发同步的重要性： 在并发编程中，如果主协程需要依赖其他Goroutine的结果或确保它们完成特定任务，就必须使用sync.WaitGroup、通道或其他同步原语进行显式同步。
- 避免竞态条件： 不进行同步可能导致竞态条件，造成程序行为不可预测，如本例中defer语句未执行。
-
选择合适的同步机制：
- 当只需要等待一组Goroutine完成时，sync.WaitGroup通常是最简洁高效的选择。
- 当Goroutine之间需要传递数据或更复杂的协调逻辑时，通道是更强大的工具。
通过理解Go语言的并发模型和正确使用同步机制，我们可以编写出更加健壮、可预测且易于维护的并发应用程序。
