本文深入探讨了go语言中，当主协程（main goroutine）过早退出时，子协程内部的`defer`语句为何可能不被执行的问题。我们将通过具体代码示例，分析其根本原因在于并发场景下的竞态条件，而非`defer`的放置或调度问题。文章将重点介绍如何利用`sync.waitgroup`或通道（channels）等go语言提供的同步原语，确保子协程能够完整执行其逻辑，包括所有延迟调用，从而构建健壮的并发应用程序。
理解Go语言中的defer与Goroutine生命周期
在Go语言中，defer语句用于延迟函数的执行，直到包含它的函数即将返回时才被调用。这些延迟调用会按照后进先出（LIFO）的顺序执行。然而，当defer语句存在于一个新创建的Goroutine（子协程）中时，其行为可能会出乎意料，尤其是在主协程（main函数）过早退出的情况下。
考虑以下示例代码：
package mainimport ( "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协程在此处退出}
运行上述代码，我们可能会观察到如下输出：
123-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完成。
稿定抠图
AI自动消除图片背景
76 查看详情
sync.WaitGroup有三个主要方法：
Add(delta int)：增加内部计数器，通常在启动Goroutine之前调用。Done()：减少内部计数器，通常在Goroutine即将完成时调用（常与defer配合使用）。Wait()：阻塞直到内部计数器归零，表示所有被Add的Goroutine都已完成。
下面是使用sync.WaitGroup修改后的示例代码：
package mainimport ( "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协程即将退出。")}
运行上述代码，我们将得到预期的输出：
123子协程完成工作-2所有协程已完成，main协程即将退出。-1
通过wg.Wait()，主协程被强制等待子协程完成其所有任务，包括执行其内部的defer fmt.Println(“-2”)语句。当子协程调用wg.Done()后，wg.Wait()解除阻塞，主协程继续执行，最终调用自己的defer fmt.Println(“-1”)并退出。
使用通道（Channels）进行同步
除了sync.WaitGroup，通道（channels）也是Go语言中常用的同步机制。如果子协程需要向主协程发送结果或仅仅发送一个完成信号，通道会是更合适的选择。
例如，通过一个空结构体通道来发送完成信号：
package mainimport ( "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语言的并发模型和正确使用同步机制，我们可以编写出更加健壮、可预测且易于维护的并发应用程序。
以上就是Go语言并发编程中，defer语句的执行时机与主协程退出机制深度解析的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1019814.html
微信扫一扫 
支付宝扫一扫 
