快捷導(dǎo)航

Golang中defer與recover的組合使用示例代碼

更新時(shí)間：2025年06月10日 11:02:12 作者：星星點(diǎn)點(diǎn)洲

Go語(yǔ)言的defer和recover機(jī)制為開發(fā)者提供了一種優(yōu)雅處理錯(cuò)誤的方式,幫助保持程序的穩(wěn)定性和可維護(hù)性,這篇文章主要介紹了Golang中defer與recover組合使用的相關(guān)資料,需要的朋友可以參考下

1. defer 的應(yīng)用場(chǎng)景

defer用于延遲執(zhí)行函數(shù)調(diào)用，確保在函數(shù)退出前執(zhí)行特定操作。主要用途包括：

資源釋放

文件操作：確保文件句柄關(guān)閉。

func readFile(filename string) error {
    file, err := os.Open(filename)
    if err != nil {
        return err
    }
    defer file.Close() // 確保函數(shù)返回前關(guān)閉文件
    // 處理文件內(nèi)容...
    return nil
}

鎖釋放：防止死鎖。

var mu sync.Mutex
func updateData() {
    mu.Lock()
    defer mu.Unlock() // 函數(shù)退出時(shí)自動(dòng)釋放鎖
    // 修改共享數(shù)據(jù)...
}

事務(wù)回滾

數(shù)據(jù)庫(kù)或業(yè)務(wù)邏輯中，確保操作失敗時(shí)回滾。

func transferMoney() {
    tx := db.Begin()
    defer func() {
        if r := recover(); r != nil { // 結(jié)合recover處理panic
            tx.Rollback()
        }
    }()
    // 執(zhí)行轉(zhuǎn)賬操作，可能觸發(fā)panic
    tx.Commit()
}

2. recover 的應(yīng)用場(chǎng)景

recover用于捕獲panic，防止程序非正常終止。必須在defer函數(shù)中調(diào)用。

全局異常恢復(fù)

防止因未處理的panic導(dǎo)致程序崩潰。

func safeCall() {
    defer func() {
        if r := recover(); r != nil {
            fmt.Println("Recovered from panic:", r)
        }
    }()
    // 可能觸發(fā)panic的代碼
    panic("unexpected error")
}

保護(hù)Goroutine

避免某個(gè)Goroutine的panic影響整個(gè)程序。

func startWorker() {
    go func() {
        defer func() {
            if r := recover(); r != nil {
                log.Println("Worker panic:", r)
            }
        }()
        // Goroutine的業(yè)務(wù)邏輯...
    }()
}

優(yōu)雅降級(jí)

將panic轉(zhuǎn)換為錯(cuò)誤，保持服務(wù)可用性。

func safeHandler() (err error) {
    defer func() {
        if r := recover(); r != nil {
            err = fmt.Errorf("internal error: %v", r)
        }
    }()
    // 可能panic的代碼
    return nil
}

3. 結(jié)合使用示例

func processRequest() (err error) {
    // 恢復(fù)panic并轉(zhuǎn)為錯(cuò)誤
    defer func() {
        if r := recover(); r != nil {
            err = fmt.Errorf("panic occurred: %v", r)
        }
    }()

    // 資源管理示例
    file, err := os.Open("data.txt")
    if err != nil {
        return err
    }
    defer file.Close() // 確保文件關(guān)閉

    // 業(yè)務(wù)邏輯（可能觸發(fā)panic）
    if someCondition {
        panic("data corruption")
    }

    return nil
}

4. 注意事項(xiàng)

recover僅在defer中有效：非defer上下文中調(diào)用會(huì)返回nil。
避免濫用recover：隱藏panic可能導(dǎo)致未知狀態(tài)，應(yīng)僅在必要時(shí)使用。
明確錯(cuò)誤處理：優(yōu)先返回錯(cuò)誤而非依賴panic/recover，后者適用于不可恢復(fù)的異常（如程序邏輯錯(cuò)誤）。

通過(guò)合理使用defer和recover，可以顯著提升Go程序的健壯性和可維護(hù)性，尤其在資源管理和異?；謴?fù)場(chǎng)景中。

在Go語(yǔ)言中，上述代碼無(wú)法正確捕獲panic，原因如下：

5. 問題分析

代碼示例

func safeCall() {
    // 直接調(diào)用recover（不在defer中）
    if r := recover(); r != nil {
        fmt.Println("Recovered from panic:", r)
    }
    // 觸發(fā)新的panic
    panic("unexpected error")
}

// 將safeCall包裹在defer中
defer func() { safeCall() }()

關(guān)鍵問題

recover未在defer中調(diào)用：
- safeCall中的recover直接調(diào)用，而非通過(guò)defer注冊(cè)的函數(shù)。此時(shí)recover會(huì)在safeCall正常執(zhí)行時(shí)立即觸發(fā)，而非在panic發(fā)生后被動(dòng)調(diào)用。
- 若此時(shí)未發(fā)生panic，recover返回nil，無(wú)法捕獲后續(xù)觸發(fā)的panic。
panic與defer執(zhí)行順序：
- 當(dāng)外層函數(shù)觸發(fā)panic時(shí)，會(huì)先執(zhí)行已注冊(cè)的defer函數(shù)。
- 在defer中調(diào)用safeCall，此時(shí)safeCall內(nèi)的recover嘗試捕獲當(dāng)前panic，但隨后safeCall自身又觸發(fā)了一個(gè)新的panic("unexpected error")，而新的panic未被任何recover處理，導(dǎo)致程序崩潰。

6. 正確寫法

修復(fù)方案

將recover放在defer函數(shù)中，并直接與可能觸發(fā)panic的代碼關(guān)聯(lián)：

func safeCall() {
    // 可能觸發(fā)panic的代碼
    defer func() {
        // 在defer中調(diào)用recover
        if r := recover(); r != nil {
            fmt.Println("Recovered from panic:", r)
        }
    }()
    panic("unexpected error")
}

// 注冊(cè)defer
defer safeCall()

執(zhí)行邏輯

調(diào)用defer safeCall()，注冊(cè)safeCall到外層函數(shù)的defer棧。
當(dāng)外層函數(shù)觸發(fā)panic時(shí)，執(zhí)行safeCall。
safeCall內(nèi)部的defer函數(shù)中的recover會(huì)捕獲當(dāng)前panic，阻止其繼續(xù)傳播。
若safeCall自身觸發(fā)panic，該panic會(huì)被其自身的defer recover捕獲。

7. 錯(cuò)誤示例的詳細(xì)解釋

原代碼執(zhí)行流程

假設(shè)外層函數(shù)觸發(fā)panic：

外層函數(shù)執(zhí)行panic("outer panic")。
程序開始處理defer，調(diào)用defer func() { safeCall() }()。
safeCall執(zhí)行：
- recover()嘗試捕獲外層panic("outer panic")，打印恢復(fù)信息。
- 隨后觸發(fā)新的panic("unexpected error")。
新的panic未被任何recover處理，導(dǎo)致程序崩潰。