01、介紹

在 Golang 語言項目開發(fā)中，變量操作不當就會觸發(fā)空指針引發(fā)程序 panic。空指針就是未分配內存的指針類型的變量，變量的值是 nil，因為操作空指針會引發(fā) panic，所以我們在程序開發(fā)中要特別小心。

02、結構體指針類型返回值

在調用結構體指針類型返回值的函數(shù)或方法時，并且需要操作返回值的字段或方法，此時，我們就需要注意觸發(fā)空指針引發(fā)的 panic。

操作返回值的字段：

func main() {
?user := GetUser()
?fmt.Println(user)
?fmt.Println(user.Id)
}

func GetUser() (user *User) {
?return
}

type User struct {
?Id ? int
?Name string
}

閱讀上面這段代碼，我們通過調用函數(shù) GetUser() 獲取 *User 類型的返回值，因為返回值變量是空指針，當我們訪問返回值的字段時，程序引發(fā) panic。

避免此類空指針問題，一是可以在返回值包含指針類型變量的函數(shù)或方法中，在函數(shù)體開頭初始化返回值的指針類型變量；二是在調用結構體指針類型返回值的函數(shù)或方法時，在操作返回值的字段或方法時，先判定返回值是否為 nil（空指針）。

func main() {
?user := GetUser()
?fmt.Println(user)
?if user != nil {
? fmt.Println(user.Id)
?}
}

func GetUser() (user *User) {
?user = new(User)
?// user = &User{}
?return
}

type User struct {
?Id ? int
?Name string
}

操作返回值的方法：

func main() {
?user := GetUser()
?user.Login()
}

func GetUser() (user *User) {
?return
}

type User struct {
?Id ? int
?Name string
}

func (u User) Login() {

}

閱讀上面這段代碼，我們通過調用函數(shù) GetUser() 獲取 *User 類型的返回值，因為返回值變量是空指針，當我們訪問返回值的方法 Login() 時，程序觸發(fā)空指針引發(fā) panic。

避免此類空指針問題，一是可以在返回值是指針類型變量的函數(shù)或方法的函數(shù)體中，開頭先初始化返回值的指針類型變量；二是類型方法的接收者使用指針類型。

func main() {
?user := GetUser()
?user.Login()
}

func GetUser() (user *User) {
?user = new(User)
?// user = &User{}
?return
}

type User struct {
?Id ? int
?Name string
}

func (u *User) Login() {

}

03、結構體指針類型 value 的 Map

在 Golang 語言程序開發(fā)中，經(jīng)常會操作結構體指針類型 value 的 Map，也需要注意觸發(fā)空指針引發(fā) panic。

func main() {
?var userData map[int]*User
?fmt.Println(userData[1].Name)
}

type User struct {
?Id ? int
?Name string
}

閱讀上面這段代碼，我們定義 map 類型的變量 userData，key 是 int 類型，value 是結構體指針類型，我們訪問 map 的值時，因為值是空指針，所以會引發(fā) panic。

避免此類空指針問題，我們可以使用 ok-idiom 模式判斷鍵值是否存在，如果鍵值存在（判斷鍵值是否為 nil），我們訪問鍵值的字段，否則不訪問。通過這種方式，也可以避免觸發(fā)空指針引發(fā) panic。

func main() {
?var userData map[int]*User
?if val, ok := userData[1]; ok {
? fmt.Println(val.Name)
?}
}

type User struct {
?Id ? int
?Name string
}

04、defer 延遲調用

關鍵字 defer 延遲調用函數(shù)，雖然被調用函數(shù)會延遲調用，但是被調用函數(shù)的變量會先被注冊。所以，如果被調用函數(shù)的變量是空指針，就會引發(fā) panic。

func main() {
?res, err := http.Get("http://www.baidu2022.com/robots.txt") // 偽造錯誤請求
?defer res.Body.Close()
?if err != nil {
? log.Fatal(err)
?}
?body, err := io.ReadAll(res.Body)
?if err != nil {
? log.Fatal(err)
?}
?fmt.Printf("%s", body)
}

閱讀上面這段代碼，使用 defer 延遲調用函數(shù)釋放資源，因為我們將 defer 放在錯誤檢查之后，所以如果返回值 res 是空指針，就會引發(fā) panic。

避免此類空指針問題，我們可以在使用 defer 調用之前，先做錯誤檢查，并且遇到錯誤后停止向下執(zhí)行。

05、總結

本文我們介紹一些 Golang 語言開發(fā)需要避免空指針引發(fā) panic 的場景，雖然都比較簡單，但是新手很容易踩“坑”。

到此這篇關于Golang語言如何避免空指針引發(fā)panic的文章就介紹到這了,更多相關Go語言避免空指針引發(fā)panic內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

參考資料：

• https://yourbasic.org/golang/gotcha-nil-pointer-dereference/
• https://blog.wuhsun.com/panic-runtime-error-invalid-memory-address-or-nil-pointer-dereference/
• https://programmerah.com/go-solve-panic-runtime-error-invalid-memory-address-or-nil-pointer-dereference-in-golang-28179/
• https://stackoverflow.com/questions/16280176/go-panic-runtime-error-invalid-memory-address-or-nil-pointer-dereference

最新評論