Golang 語言怎么避免空指針引發(fā)的 Panic
01介紹
在 Golang 語言項目開發(fā)中,變量操作不當就會觸發(fā)空指針引發(fā)程序 panic??罩羔樉褪俏捶峙鋬?nèi)存的指針類型的變量,變量的值是 nil,因為操作空指針會引發(fā) panic,所以我們在程序開發(fā)中要特別小心。
02結(jié)構(gòu)體指針類型返回值
在調(diào)用結(jié)構(gòu)體指針類型返回值的函數(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
- }
閱讀上面這段代碼,我們通過調(diào)用函數(shù) GetUser() 獲取 *User 類型的返回值,因為返回值變量是空指針,當我們訪問返回值的字段時,程序引發(fā) panic。
避免此類空指針問題,一是可以在返回值包含指針類型變量的函數(shù)或方法中,在函數(shù)體開頭初始化返回值的指針類型變量;二是在調(diào)用結(jié)構(gòu)體指針類型返回值的函數(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() {
- }
閱讀上面這段代碼,我們通過調(diào)用函數(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結(jié)構(gòu)體指針類型 value 的 Map
在 Golang 語言程序開發(fā)中,經(jīng)常會操作結(jié)構(gòu)體指針類型 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 是結(jié)構(gòu)體指針類型,我們訪問 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
- }
04defer 延遲調(diào)用
關(guān)鍵字 defer 延遲調(diào)用函數(shù),雖然被調(diào)用函數(shù)會延遲調(diào)用,但是被調(diào)用函數(shù)的變量會先被注冊。所以,如果被調(diào)用函數(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 延遲調(diào)用函數(shù)釋放資源,因為我們將 defer 放在錯誤檢查之后,所以如果返回值 res 是空指針,就會引發(fā) panic。
避免此類空指針問題,我們可以在使用 defer 調(diào)用之前,先做錯誤檢查,并且遇到錯誤后停止向下執(zhí)行。
05總結(jié)
本文我們介紹一些 Golang 語言開發(fā)需要避免空指針引發(fā) panic 的場景,雖然都比較簡單,但是新手很容易踩“坑”。歡迎讀者朋友們在評論區(qū)與大家分享更多因為觸發(fā)空指針引發(fā) panic 的場景。