本文深入探讨go语言中goroutine的`defer`语句行为。当主程序提前终止时，子goroutine的`defer`可能不会被执行。文章分析了这一现象的根本原因，即goroutine的生命周期与主程序的关联，并详细介绍了如何通过`sync.waitgroup`和channel等显式同步机制，确保goroutine的`defer`逻辑得以正确执行，从而避免潜在的资源泄露或状态不一致问题。
在Go语言中，defer语句用于确保函数在返回前执行某些操作，例如资源清理、解锁互斥量等。然而，当defer语句存在于一个新启动的goroutine中时，其行为可能会出乎意料，尤其是在主程序过早结束的情况下。理解这种行为对于编写健壮的并发程序至关重要。
1. 问题现象与原因分析
考虑以下Go程序代码：
package mainimport ( "fmt" "time")func main() { fmt.Println("1") defer fmt.Println("-1") // 主goroutine的defer go func() { fmt.Println("2") defer fmt.Println("-2") // 子goroutine的defer time.Sleep(9 * time.Second) // 模拟长时间运行的任务 }() time.Sleep(1 * time.Second) // 主goroutine短暂等待 fmt.Println("3")}
运行上述代码，您可能会观察到如下输出：
123-1
与预期中的 1 2 3 -2 -1 不同，子goroutine中的 defer fmt.Println(“-2”) 并没有被执行。
立即学习“go语言免费学习笔记（深入）”；
原因分析：
这个现象的根本原因在于Go程序的生命周期管理。当main函数执行完毕并返回时，Go程序会立即终止，无论是否有其他goroutine仍在运行。defer语句的执行时机是其所在函数即将返回之前。在上述示例中，主goroutine在执行完 time.Sleep(1 * time.Second) 并打印 “3” 之后，就迅速进入了结束阶段，并执行了其自身的 defer fmt.Println(“-1”)。
与此同时，子goroutine被启动后，开始执行 time.Sleep(9 * time.Second)。由于主goroutine没有等待子goroutine完成，它在子goroutine的9秒睡眠结束之前就已经退出。当主程序终止时，所有尚未完成的goroutine（包括那个正在睡眠的子goroutine）都会被强制终止，它们内部的defer语句自然也就没有机会执行了。这并非defer放置位置的问题，而是并发程序中常见的竞态条件（race condition）——主程序与子goroutine的执行顺序存在不确定性，且主程序没有等待子goroutine完成。
2. Goroutine与程序生命周期
Go程序默认的主goroutine是main函数。当main函数完成执行时，整个Go程序就会退出。这意味着，如果你启动了其他goroutine，但main函数没有显式地等待这些goroutine完成，那么这些子goroutine可能会在完成任务之前就被“杀死”。
为了确保子goroutine有机会完成其任务并执行其defer语句，主goroutine必须等待它。这需要引入显式的同步机制。
稿定抠图
AI自动消除图片背景
76 查看详情
3. 解决方案：显式同步
Go语言提供了多种强大的同步原语，用于协调不同goroutine的执行。对于本例中的“等待子goroutine完成”的需求，sync.WaitGroup和channel是两种最常用且有效的方案。
3.1 使用 sync.WaitGroup
sync.WaitGroup 是一个计数器，用于等待一组goroutine完成。它提供了三个方法：
Add(delta int)：增加WaitGroup的计数器。通常在启动新的goroutine之前调用。Done()：减少WaitGroup的计数器。通常在goroutine完成任务后调用，并常常结合defer使用。Wait()：阻塞直到WaitGroup的计数器归零。通常在主goroutine中调用，以等待所有子goroutine完成。
下面是使用sync.WaitGroup改进后的代码示例：
package mainimport ( "fmt" "sync" // 引入sync包 "time")func main() { fmt.Println("1") defer fmt.Println("-1") // 主goroutine的defer var wg sync.WaitGroup // 声明一个WaitGroup wg.Add(1) // 增加计数器，表示有一个goroutine需要等待 go func() { defer wg.Done() // 在子goroutine函数返回前调用Done，减少计数器 fmt.Println("2") defer fmt.Println("-2") // 子goroutine的defer time.Sleep(9 * time.Second) // 模拟长时间运行的任务 fmt.Println("子goroutine: 任务完成") // 确认子goroutine完成 }() time.Sleep(1 * time.Second) // 主goroutine短暂等待 fmt.Println("3") wg.Wait() // 阻塞主goroutine，直到所有Add的goroutine都调用了Done fmt.Println("主goroutine: 已等待子goroutine完成")}
运行这段代码，输出将变为：
123子goroutine: 任务完成-2主goroutine: 已等待子goroutine完成-1
现在，子goroutine中的 defer fmt.Println(“-2”) 得到了正确执行。这是因为 wg.Wait() 确保了主goroutine会一直阻塞，直到子goroutine执行 wg.Done() 并且其函数返回（从而执行了 defer fmt.Println(“-2”)）之后，主goroutine才会继续执行。
3.2 使用 Channel 进行同步
Channel是Go语言中用于goroutine之间通信的强大工具，也可以用于简单的同步。我们可以创建一个channel，让子goroutine在完成任务后发送一个信号，主goroutine则等待接收这个信号。
package mainimport ( "fmt" "time")func main() { fmt.Println("1") defer fmt.Println("-1") // 主goroutine的defer done := make(chan struct{}) // 创建一个用于同步的空结构体channel go func() { fmt.Println("2") defer fmt.Println("-2") // 子goroutine的defer time.Sleep(9 * time.Second) // 模拟长时间运行的任务 fmt.Println("子goroutine: 任务完成") close(done) // 关闭channel，向主goroutine发送完成信号 }() time.Sleep(1 * time.Second) // 主goroutine短暂等待 fmt.Println("3") <-done // 阻塞主goroutine，直到从done channel接收到信号（或channel被关闭） fmt.Println("主goroutine: 已等待子goroutine完成")}
这段代码的输出与使用sync.WaitGroup的示例相似，同样能确保子goroutine的defer被执行。<-done操作会阻塞主goroutine，直到done channel被关闭或有数据发送。当子goroutine执行 close(done) 时，主goroutine解除阻塞，继续执行后续代码。
4. 注意事项与最佳实践
显式同步是关键： 当主goroutine的执行依赖于其他goroutine的完成，或者需要确保其他goroutine的资源清理（如defer）得以执行时，必须使用显式同步机制。选择合适的同步原语：sync.WaitGroup：适用于“等待N个goroutine完成”的场景，尤其当你只关心完成数量而不关心具体通信内容时。Channel：适用于goroutine之间需要传递数据、发送特定信号或进行更复杂协调的场景。defer与WaitGroup.Done()的结合： 将wg.Done()放在defer语句中是最佳实践，它能确保无论goroutine正常返回还是发生panic，计数器都能被正确递减，防止主goroutine永远阻塞。避免滥用 runtime.Gosched()： 有些新手可能会尝试使用 runtime.Gosched() 来“让出CPU”以期望子goroutine能完成。然而，runtime.Gosched() 只是一个调度提示，它并不能保证子goroutine一定会在主goroutine之前完成，也不是一个可靠的同步机制。其效果往往是偶然的，并且高度依赖于运行时调度器的具体实现。错误处理： 在实际应用中，子goroutine中可能发生错误。同步机制通常还需要与错误处理机制（如通过channel传递错误）结合使用。
总结
Go语言中goroutine的defer语句执行依赖于其所在函数的正常返回。如果主程序在子goroutine函数返回之前就已终止，那么子goroutine的defer将不会被执行。为了避免这种“隐形”的资源泄露或逻辑中断，开发者必须通过sync.WaitGroup或channel等显式同步机制，确保主goroutine等待子goroutine完成任务。理解并正确运用这些同步原语是编写高效、健壮Go并发程序的基石。
以上就是Go语言中Goroutine的defer行为与显式同步机制详解的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1017843.html
微信扫一扫 
支付宝扫一扫 
