Go 函数并发编程常见的陷阱和避免方法

Go 的并发编程功能强大，但如果不加小心容易出现一些常见的陷阱。让我们探讨一下这些陷阱以及避免它们的最佳实践。

1. 数据竞争

数据竞争发生在多个 goroutine 同时访问共享内存时。这会导致不可预测的行为，包括数据损坏和死锁。

立即学习“go语言免费学习笔记（深入）”；

避免方法：

• 使用锁或并发安全的类型来保护共享数据。
• 使用 channels 在 goroutine 之间安全地传递数据。

2. 死锁

死锁发生在两个或多个 goroutine 相互等待，导致所有 goroutine 都被阻塞。

Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等

下载

避免方法：

• 避免循环等待。
• 使用计时器或超时来防止死锁。
• 仔细考虑 goroutine 的依赖关系。

3. 资源泄漏

如果 goroutine 没有正确释放资源（例如打开的文件或网络连接），会导致资源泄漏。这会消耗系统资源，最终导致性能下降甚至崩溃。

避免方法：

• 使用 defer 语句在 goroutine 退出时自动释放资源（例如 defer file.Close()）。
• 使用 context 对象来传播取消信号，以便在 goroutine 取消时释放资源。

实战案例

考虑以下代码示例：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var count int
    var wg sync.WaitGroup
    
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            count++
            wg.Done()
        }()
    }
    
    wg.Wait()
    fmt.Println(count) // 预期输出：1000
}

在这个示例中，我们创建了 1000 个 goroutine，每个 goroutine 都递增共享变量 count。不幸的是，count 未被同步，这会导致数据竞争。为避免这种情况，我们可以使用互斥锁：

package main

import (
    "fmt"
    "sync"
)

func main() {
    var count int
    var lock sync.Mutex
    
    var wg sync.WaitGroup
    
    for i := 0; i < 1000; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            
            lock.Lock() // 加锁
            count++
            lock.Unlock() // 解锁
        }()
    }
    
    wg.Wait()
    fmt.Println(count) // 预期输出：1000
}

通过使用互斥锁，我们确保每次只有一个 goroutine 可以访问 count，从而避免了数据竞争。