Golang如何使用defer释放资源

defer用于延迟执行函数,确保资源释放;多个defer按后进先出顺序执行;常用于关闭文件、网络连接、释放锁等;注意参数在defer时求值,闭包中需避免变量覆盖问题。

在Go语言中,defer 是一个非常实用的关键字,用于延迟执行函数调用,通常用来确保资源被正确释放,比如关闭文件、释放锁或关闭网络连接。它的核心作用是:无论函数如何退出(正常返回或发生panic),被 defer 的语句都会执行,从而避免资源泄漏。

1. defer 的基本用法

defer 后面跟一个函数或方法调用,该调用会被推迟到当前函数返回前执行。

例如,打开文件后需要关闭:

func readFile(filename string) error {
    file, err := os.Open(filename)
    if err != nil {
        return err
    }
    defer file.Close() // 函数返回前自动调用

    // 读取文件内容
    data := make([]byte, 100)
    _, err = file.Read(data)
    return err
}

即使 read 过程出错,file.Close() 依然会被调用,保证文件描述符不会泄漏。

2. 多个 defer 的执行顺序

当有多个 defer 时,它们按后进先出(LIFO)的顺序执行。

func example() {
    defer fmt.Println("first")
    defer fmt.Println("second")
    defer fmt.Println("third")
}
// 输出:third → second → first

这个特性在需要按顺序释放资源时很有用,比如多次加锁后逆序解锁。

3. 常见资源释放场景

除了文件操作,defer 广泛用于以下资源管理:

  • 关闭网络连接:
    conn, _ := net.Dial("tcp", "example.com:80")
        defer conn.Close()
  • 释放互斥锁:
    mu.Lock()
        defer mu.Unlock()
    防止因提前 return 或 panic 导致死锁。
  • 关闭数据库连接或事务:
    tx, _ := db.Begin()
        defer tx.Rollback() // 如果未 Commit,自动回滚
        // ... 执行SQL
        tx.Commit()

4. 注意事项与陷阱

使用 defer 时要注意以下几点:

  • 参数在 defer 时求值:
    func example(i int) {
            defer fmt.Println(i) // 输出的是 defer 时刻的 i 值
            i++
        }
    上例中,尽管 i++,输出仍是传入值。
  • 闭包 defer 需小心:
    for i := 0; i < 3; i++ {
            defer func() { fmt.Println(i) }() // 全部输出 3
        }
    应改为传参捕获变量:
    defer func(val int) { fmt.Println(val) }(i)
  • 性能考虑: defer 有轻微开销,高频循环中谨慎使用,但多数场景可忽略。

基本上就这些。合理使用 defer 能显著提升代码的安全性和可读性,让资源释放逻辑更清晰、不易遗漏。