01介紹

在 Golang 語(yǔ)言項(xiàng)目開(kāi)發(fā)中，變量操作不當(dāng)就會(huì)觸發(fā)空指針引發(fā)程序 panic?？罩羔樉褪俏捶峙鋬?nèi)存的指針類型的變量，變量的值是 nil，因?yàn)椴僮骺罩羔槙?huì)引發(fā) panic，所以我們?cè)诔绦蜷_(kāi)發(fā)中要特別小心。

02結(jié)構(gòu)體指針類型返回值

在調(diào)用結(jié)構(gòu)體指針類型返回值的函數(shù)或方法時(shí)，并且需要操作返回值的字段或方法，此時(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 
} 

閱讀上面這段代碼，我們通過(guò)調(diào)用函數(shù) GetUser() 獲取 *User 類型的返回值，因?yàn)榉祷刂底兞渴强罩羔?，?dāng)我們?cè)L問(wèn)返回值的字段時(shí)，程序引發(fā) panic。

避免此類空指針問(wèn)題，一是可以在返回值包含指針類型變量的函數(shù)或方法中，在函數(shù)體開(kāi)頭初始化返回值的指針類型變量;二是在調(diào)用結(jié)構(gòu)體指針類型返回值的函數(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() { 
 
} 

閱讀上面這段代碼，我們通過(guò)調(diào)用函數(shù) GetUser() 獲取 *User 類型的返回值，因?yàn)榉祷刂底兞渴强罩羔?，?dāng)我們?cè)L問(wèn)返回值的方法 Login() 時(shí)，程序觸發(fā)空指針引發(fā) panic。

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

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 語(yǔ)言程序開(kāi)發(fā)中，經(jīng)常會(huì)操作結(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)體指針類型，我們?cè)L問(wèn) map 的值時(shí)，因?yàn)橹凳强罩羔?，所以?huì)引發(fā) panic。

避免此類空指針問(wèn)題，我們可以使用 ok-idiom 模式判斷鍵值是否存在，如果鍵值存在(判斷鍵值是否為 nil)，我們?cè)L問(wèn)鍵值的字段，否則不訪問(wèn)。通過(guò)這種方式，也可以避免觸發(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ù)會(huì)延遲調(diào)用，但是被調(diào)用函數(shù)的變量會(huì)先被注冊(cè)。所以，如果被調(diào)用函數(shù)的變量是空指針，就會(huì)引發(fā) panic。

func main() { 
 res, err := http.Get("http://www.baidu2022.com/robots.txt") // 偽造錯(cuò)誤請(qǐng)求 
 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ù)釋放資源，因?yàn)槲覀儗?defer 放在錯(cuò)誤檢查之后，所以如果返回值 res 是空指針，就會(huì)引發(fā) panic。

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

05總結(jié)

本文我們介紹一些 Golang 語(yǔ)言開(kāi)發(fā)需要避免空指針引發(fā) panic 的場(chǎng)景，雖然都比較簡(jiǎn)單，但是新手很容易踩“坑”。歡迎讀者朋友們?cè)谠u(píng)論區(qū)與大家分享更多因?yàn)橛|發(fā)空指針引發(fā) panic 的場(chǎng)景。