Golang語言如何避免空指針引發的panic詳解

01、介紹

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

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

在調用結構體指針類型返回值的函數或方法時,並且需要操作返回值的字段或方法,此時,我們就需要註意觸發空指針引發的 panic。

操作返回值的字段:

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

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

type User struct {
 Id   int
 Name string
}

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

避免此類空指針問題,一是可以在返回值包含指針類型變量的函數或方法中,在函數體開頭初始化返回值的指針類型變量;二是在調用結構體指針類型返回值的函數或方法時,在操作返回值的字段或方法時,先判定返回值是否為 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() {

}

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

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

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 語言程序開發中,經常會操作結構體指針類型 value 的 Map,也需要註意觸發空指針引發 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 的值時,因為值是空指針,所以會引發 panic。

避免此類空指針問題,我們可以使用 ok-idiom 模式判斷鍵值是否存在,如果鍵值存在(判斷鍵值是否為 nil),我們訪問鍵值的字段,否則不訪問。通過這種方式,也可以避免觸發空指針引發 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 延遲調用函數,雖然被調用函數會延遲調用,但是被調用函數的變量會先被註冊。所以,如果被調用函數的變量是空指針,就會引發 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 延遲調用函數釋放資源,因為我們將 defer 放在錯誤檢查之後,所以如果返回值 res 是空指針,就會引發 panic。

避免此類空指針問題,我們可以在使用 defer 調用之前,先做錯誤檢查,並且遇到錯誤後停止向下執行。

05、總結

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

到此這篇關於Golang語言如何避免空指針引發panic的文章就介紹到這瞭,更多相關Go語言避免空指針引發panic內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!

參考資料:

  • 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

推薦閱讀: