Go 语言中的 `defer`：基础应用详解-程序员茄子

编程 Go 语言中的 `defer`：基础应用详解

2024-11-18 10:59:04 +0800 CST views 607

Go 语言中的 `defer`：基础应用详解

defer 是 Go 语言中的一个强大特性，主要用于在函数返回之前执行一些操作，比如资源清理、记录日志等。defer 通过延迟执行操作，不仅可以确保资源的正确释放，还可以让代码更加简洁和易于维护。

1.1 使用场景

1.1.1 资源清理

defer 主要用于在函数退出前执行资源清理操作，比如关闭文件、网络连接、数据库连接等。这种用法在 Go 中十分常见，可以有效避免资源泄露。

func readFile(filename string) {
    file, err := os.Open(filename)
    if err != nil {
        log.Fatal(err)
    }
    defer file.Close()
    // 执行其他操作
}

1.1.2 解锁资源

在并发编程中，锁的使用是非常普遍的，defer 可以帮助确保锁被正确解锁。通过 defer 解锁，可以避免因为程序中的 return 或异常导致的锁未解开问题。

var mu sync.Mutex

func criticalSection() {
    mu.Lock()
    defer mu.Unlock()
    // 执行临界区的操作
}

1.1.3 确保后续操作执行

在一些情况下，我们希望函数在结束时进行一些后续操作，如记录日志或分析函数执行时间，defer 可以确保这些操作即使在 return 之前也能执行。

func timeTrack(start time.Time, name string) {
    elapsed := time.Since(start)
    fmt.Printf("%s took %s\n", name, elapsed)
}

func someFunction() {
    defer timeTrack(time.Now(), "someFunction")
    defer fmt.Println("Function completed")

    fmt.Println("Executing someFunction...")
    time.Sleep(2 * time.Second)
}

1.2 触发时机

defer 语句会在函数返回之前执行，具体时机如下：

函数执行 return 语句时。
函数正常执行到结束时。
函数发生 panic 时。

func deferExample() {
    defer fmt.Println("defer")
    fmt.Println("function body")
    return
}

// 输出顺序为：
// function body
// defer

1.3 执行顺序与使用须知

在 Go 中，多个 defer 语句的执行顺序遵循栈的原则，即后定义的 defer 语句会先执行（LIFO - 后进先出）。

func main() {
    defer fmt.Println("1")
    defer fmt.Println("2")
    defer fmt.Println("3")
    fmt.Println("End of function")
}

// 输出：
// End of function
// 3
// 2
// 1

1.4 返回值处理

在 Go 中，defer 的执行发生在 return 之后、函数真正返回之前。这意味着在返回值已经确定的情况下，defer 仍然可以访问并修改命名返回值。

func deferReturn() (res int) {
    res = 1
    defer func() {
        res = 2 // 修改返回值
    }()
    return
}

// 输出：2

1.5 使用注意点

使用 defer 时有几点需要特别注意：

defer 会在函数返回前执行，因此应该尽量靠近资源分配的地方使用，以确保资源在整个函数中都得到正确管理。
defer 语句的函数参数在调用时就会被求值，而不是在 defer 实际执行时。

func main() {
    startedAt := time.Now()
    defer func() {
        fmt.Println(time.Since(startedAt))  // 输出 1 秒
    }()
    time.Sleep(1 * time.Second)
}