正确规避常见的 Go 并发陷阱

最新推荐文章于 2021-05-26 13:46:52 发布

原创最新推荐文章于 2021-05-26 13:46:52 发布 · 346 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#多线程 #go #gwt #css3 #css

本文介绍了Go语言并发开发中常见的技巧及陷阱，包括闭包传递参数的问题、panic异常处理及超时控制等，帮助开发者避免并发错误。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

作为一门并发性能优越的语言，Go虽然降低了协程并发开发的难度，但也存在一些「并发陷阱」，这就需要我们在开发时额外注意。

我们再来介绍与Go 并发相关的几个小技巧，帮助你规避 Go 并发开发的一些陷阱。

闭包传递参数的问题

首先是循环并发时闭包传递参数的问题，如下错误例子所示：

func main()  {
for i := 0 ; i < 5 ; i++{
go func() {
fmt.Println("current i is ", i)
}()
}
time.Sleep(time.Second)
}

这段代码极有可能的输出为：

current i is 5
current i is 5
current i is 5
current i is 5
current i is 5

这是因为 i 使用的地址空间在循环中被复用，在 goroutine 执行时，i 的值可能在被主 goroutine 修改，而此时其他 goroutine 也在读取使用，从而导致了并发错误。针对这种错误可以通过「复制拷贝」或者「传参拷贝」的方式规避，如下所示：

func main()  {
	for i := 0 ; i < 5 ; i++{
		go func(v int) {
			fmt.Println("current i is", v)
		}(i)
	}
	time.Sleep(time.Second)
}

panic 异常

上一篇介绍 panic 时我们了解到 panic 异常的出现会导致 Go 程序的崩溃。但其实即使 panic 是出现在其他启动的子 goroutine 中，也会导致 Go 程序的崩溃退出，同时 panic 只能捕获 goroutine 自身的异常，因此「对于每个启动的 goroutine，都需要在入口处捕获 panic，并尝试打印堆栈信息并进行异常处理，从而避免子 goroutine 的 panic 导致整个程序的崩溃退出」。如下面的例子所示：

func RecoverPanic() {
    // 从 panic 中恢复并打印栈信息
if e := recover(); e != nil {
buf := make([]byte, 1024)
buf = buf[:runtime.Stack(buf, false)]
fmt.Printf("[PANIC]%v\n%s\n", e, buf)
}
}
func main() {
for i:= 0 ; i < 5 ; i++{
go func() {
            // defer 注册 panic 捕获函数
defer RecoverPanic()
dothing()
}()
}
}

超时控制

最后一个技巧是要「善于结合使用 select、timer 和 context 进行超时控制」。在 goroutine 中进行一些耗时较长的操作，最好都加上超时timer，在并发的时候也要传递 context，这样在取消的时候就不会有遗漏，进而达到回收 goroutine 的目的，避免内存泄漏的发生。如下面的例子所示，通过 select 同时监听任务和定时器状态，在定时器到达而任务未完成之时，提前结束任务，清理资源并返回。

select {
// do logic process
case msg <- input:
   ....
// has been canceled
case <-ctx.Done():
    // ...资源清理
    return
// 2 second timeout    
case <-time.After(time.Second * 2)  
    // ...资源清理
    return
default:
}