[[429635]]

先聲明：我被坑過。

之前寫 Go 專欄時(shí)，寫過一篇文章：Go 專欄|錯(cuò)誤處理：defer，panic 和 recover。有小伙伴留言說：道理都懂，但還是不知道怎么用，而且還總出現(xiàn)莫名奇妙的問題。

出問題就對了，這個(gè)小東西壞的很，一不留神就出錯(cuò)。

所以，面對這種情況，我們今天就不講道理了。直接把我珍藏多年的代碼一把梭，憑借多年踩坑經(jīng)歷和寫 BUG 經(jīng)驗(yàn)，我要站著把這個(gè)坑邁過去。

一、

先來一個(gè)簡單的例子熱熱身：

package main 
 
import ( 
    "fmt" 
) 
 
func main() { 
    defer func() { 
        fmt.Println("first") 
    }() 
 
    defer func() { 
        fmt.Println("second") 
    }() 
 
    fmt.Println("done") 
} 

輸出：

done 
second 
first 

這個(gè)比較簡單，defer 語句的執(zhí)行順序是按調(diào)用 defer 語句的倒序執(zhí)行。

二、

看看這段代碼有什么問題?

for _, filename := range filenames { 
    f, err := os.Open(filename) 
    if err != nil { 
        return err 
    } 
    defer f.Close() 
} 

這段代碼其實(shí)很危險(xiǎn)，很可能會(huì)用盡所有文件描述符。因?yàn)?defer 語句不到函數(shù)的最后一刻是不會(huì)執(zhí)行的，也就是說文件始終得不到關(guān)閉。所以切記，一定不要在 for 循環(huán)中使用 defer 語句。

那怎么優(yōu)化呢?可以將循環(huán)體單獨(dú)寫一個(gè)函數(shù)，這樣每次循環(huán)的時(shí)候都會(huì)調(diào)用關(guān)閉函數(shù)。

如下：

for _, filename := range filenames { 
    if err := doFile(filename); err != nil { 
        return err 
    } 
} 
 
func doFile(filename string) error { 
    f, err := os.Open(filename) 
    if err != nil { 
        return err 
    } 
    defer f.Close() 
} 

三、

看看這三個(gè)函數(shù)的輸出結(jié)果是什么?

package main 
 
import ( 
    "fmt" 
) 
 
func a() (r int) { 
    defer func() { 
        r++ 
    }() 
    return 0 
} 
 
func b() (r int) { 
    t := 5 
    defer func() { 
        t = t + 5 
    }() 
    return t 
} 
 
func c() (r int) { 
    defer func(r int) { 
        r = r + 5 
    }(r) 
    return 1 
} 
 
func main() { 
    fmt.Println("a = ", a()) 
    fmt.Println("b = ", b()) 
    fmt.Println("c = ", c()) 
} 

公布答案：

a =  1 
b =  5 
c =  1 

你答對了嗎?

說實(shí)話剛開始看到這個(gè)結(jié)果時(shí)，我是相當(dāng)費(fèi)解，完全不知道怎么回事。

但可以看到，這三個(gè)函數(shù)都有一個(gè)共同特點(diǎn)，它們都有一個(gè)命名返回值，并且都在函數(shù)中引用了這個(gè)返回值。

引用的方式分兩種：分別是閉包和函數(shù)參數(shù)。

先看 a() 函數(shù)：

閉包通過 r++ 修改了外部變量，返回值變成了 1。

相當(dāng)于：

func aa() (r int) { 
    r = 0 
    // 在 return 之前，執(zhí)行 defer 函數(shù) 
    func() { 
        r++ 
    }() 
    return 
} 

再看 b() 函數(shù)：

閉包內(nèi)修改的只是局部變量 t，而外部變量 t 不受影響，所以還是返回 5。

相當(dāng)于：

func bb() (r int) { 
    t := 5 
    // 賦值 
    r = t 
    // 在 return 之前，執(zhí)行 defer 函數(shù) 
    // defer 函數(shù)沒有對返回值 r 進(jìn)行修改，只是修改了變量 t 
    func() { 
        t = t + 5 
    }() 
    return 
} 

最后是 c 函數(shù)：

參數(shù)傳遞是值拷貝，實(shí)參不受影響，所以還是返回 1。

相當(dāng)于：

func cc() (r int) { 
    // 賦值 
    r = 1 
    // 這里修改的 r 是函數(shù)形參的值 
    // 值拷貝，不影響實(shí)參值 
    func(r int) { 
        r = r + 5 
    }(r) 
    return 
} 

那么，為了避免寫出這么令人意外的代碼，最好在定義函數(shù)時(shí)就不要使用命名返回值。或者如果使用了，就不要在 defer 中引用。

再看下面兩個(gè)例子：

func d() int { 
    r := 0 
    defer func() { 
        r++ 
    }() 
    return r 
} 
 
func e() int { 
    r := 0 
    defer func(i int) { 
        i++ 
    }(r) 
    return 0 
} 
d =  0 
e =  0 

返回值符合預(yù)期，再也不用絞盡腦汁猜了。

四、

defer 表達(dá)式的函數(shù)如果在 panic 后面，則這個(gè)函數(shù)無法被執(zhí)行。

func main() { 
    panic("a") 
    defer func() { 
        fmt.Println("b") 
    }() 
} 

輸出如下，b 沒有打印出來。

panic: a 
 
goroutine 1 [running]: 
main.main() 
    xxx.go:87 +0x4ce 
exit status 2 

而如果 defer 在前，則可以執(zhí)行。

func main() { 
    defer func() { 
        fmt.Println("b") 
    }() 
    panic("a") 
} 

輸出：

b 
panic: a 
 
goroutine 1 [running]: 
main.main() 
    xxx.go:90 +0x4e7 
exit status 2 

五、

看看下面這段代碼的執(zhí)行順序：

func G() { 
    defer func() { 
        fmt.Println("c") 
    }() 
 
    F() 
    fmt.Println("繼續(xù)執(zhí)行") 
} 
 
func F() { 
    defer func() { 
        if err := recover(); err != nil { 
            fmt.Println("捕獲異常:", err) 
        } 
        fmt.Println("b") 
    }() 
    panic("a") 
} 
 
func main() { 
    G() 
} 

順序如下：

1. 調(diào)用 G() 函數(shù);
2. 調(diào)用 F() 函數(shù);
3. F() 中遇到 panic，立刻終止，不執(zhí)行 panic 之后的代碼;
4. 執(zhí)行 F() 中 defer 函數(shù)，遇到 recover 捕獲錯(cuò)誤，繼續(xù)執(zhí)行 defer 中代碼，然后返回;
5. 執(zhí)行 G() 函數(shù)后續(xù)代碼，最后執(zhí)行 G() 中 defer 函數(shù)。

輸出：

捕獲異常: a 
b 
繼續(xù)執(zhí)行 
c 

五、

看看下面這段代碼的執(zhí)行順序：

func G() { 
    defer func() { 
        if err := recover(); err != nil { 
            fmt.Println("捕獲異常:", err) 
        } 
        fmt.Println("c") 
    }() 
 
    F() 
    fmt.Println("繼續(xù)執(zhí)行") 
} 
 
func F() { 
    defer func() { 
        fmt.Println("b") 
    }() 
    panic("a") 
} 
 
func main() { 
    G() 
} 

順序如下：

1. 調(diào)用 G() 函數(shù);
2. 調(diào)用 F() 函數(shù);
3. F() 中遇到 panic，立刻終止，不執(zhí)行 panic 之后的代碼;
4. 執(zhí)行 F() 中 defer 函數(shù)，由于沒有 recover，則將 panic 拋到 G() 中;
5. G() 收到 panic 則不會(huì)執(zhí)行后續(xù)代碼，直接執(zhí)行 defer 函數(shù);
6. defer 中捕獲 F() 拋出的異常 a，然后繼續(xù)執(zhí)行，最后退出。

輸出：

b 
捕獲異常: a 
c 

六、

看看下面這段代碼的執(zhí)行順序：

func G() { 
    defer func() { 
        fmt.Println("c") 
    }() 
 
    F() 
    fmt.Println("繼續(xù)執(zhí)行") 
} 
 
func F() { 
    defer func() { 
        fmt.Println("b") 
    }() 
    panic("a") 
} 
 
func main() { 
    G() 
} 

順序如下：

1. 調(diào)用 G() 函數(shù);
2. 調(diào)用 F() 函數(shù);
3. F() 中遇到 panic，立刻終止，不執(zhí)行 panic 之后的代碼;
4. 執(zhí)行 F() 中 defer 函數(shù)，由于沒有 recover，則將 panic 拋到 G() 中;
5. G() 收到 panic 則不會(huì)執(zhí)行后續(xù)代碼，直接執(zhí)行 defer 函數(shù);
6. 由于沒有 recover，直接拋出 F() 拋過來的異常 a，然后退出。

輸出：

b 
c 
panic: a 
 
goroutine 1 [running]: 
main.F() 
    xxx.go:90 +0x5b 
main.G() 
    xxx.go:82 +0x48 
main.main() 
    xxx.go:107 +0x4a5 
exit status 2 

七、

看看下面這段代碼的執(zhí)行順序：

func G() { 
    defer func() { 
        // goroutine 外進(jìn)行 recover 
        if err := recover(); err != nil { 
            fmt.Println("捕獲異常:", err) 
        } 
        fmt.Println("c") 
    }() 
 
    // 創(chuàng)建 goroutine 調(diào)用 F 函數(shù) 
    go F() 
    time.Sleep(time.Second) 
} 
 
func F() { 
    defer func() { 
        fmt.Println("b") 
    }() 
    // goroutine 內(nèi)部拋出panic 
    panic("a") 
} 
 
func main() { 
    G() 
} 

順序如下：

1. 調(diào)用 G() 函數(shù);
2. 通過 goroutine 調(diào)用 F() 函數(shù);
3. F() 中遇到 panic，立刻終止，不執(zhí)行 panic 之后的代碼;
4. 執(zhí)行 F() 中 defer 函數(shù)，由于沒有 recover，則將 panic 拋到 G() 中;
5. 由于 goroutine 內(nèi)部沒有進(jìn)行 recover，則 goroutine 外部函數(shù)，也就是 G() 函數(shù)是沒辦法捕獲的，程序直接崩潰退出。

輸出：

b 
panic: a 
 
goroutine 6 [running]: 
main.F() 
    xxx.go:96 +0x5b 
created by main.G 
    xxx.go:87 +0x57 
exit status 2 

八、

最后再說一個(gè) recover 的返回值問題：

defer func() { 
    if err := recover(); err != nil { 
        fmt.Println("捕獲異常:", err.Error()) 
    } 
}() 
panic("a") 

recover 返回的是 interface {} 類型，而不是 error 類型，所以這樣使用的話會(huì)報(bào)錯(cuò)：

err.Error undefined (type interface {} is interface with no methods) 

可以這樣來轉(zhuǎn)換一下：

defer func() { 
    if err := recover(); err != nil { 
        fmt.Println("捕獲異常:", fmt.Errorf("%v", err).Error()) 
    } 
}() 
panic("a") 

或者直接打印結(jié)果：

defer func() { 
    if err := recover(); err != nil { 
        fmt.Println("捕獲異常:", err) 
    } 
}() 
panic("a") 

輸出：

捕獲異常: a 

以上就是本文的全部內(nèi)容，其實(shí)寫過其他的語言的同學(xué)都知道，關(guān)閉文件句柄，釋放鎖等操作是很容易忘的。而 Go 語言通過 defer 很好地解決了這個(gè)問題，但在使用過程中還是要小心。

本文總結(jié)了一些容踩坑的點(diǎn)，希望能夠幫助大家少寫 BUG，如果大家覺得有用的話，歡迎點(diǎn)贊和轉(zhuǎn)發(fā)。

文章中的腦圖和源碼都上傳到了 GitHub，有需要的同學(xué)可自行下載。

源碼地址：

https://github.com/yongxinz/gopher/tree/main/sc

本文轉(zhuǎn)載自微信公眾號「AlwaysBeta」，可以通過以下二維碼關(guān)注。轉(zhuǎn)載本文請聯(lián)系A(chǔ)lwaysBeta公眾號。