Golang 语言怎么避免空指针引发的 Panic

时间：2021-07-13 04:21:45

Golang 语言怎么避免空指针引发的 Panic

01介绍

在 Golang 语言项目开发中，变量操作不当就会触发空指针引发程序 panic。空指针就是未分配内存的指针类型的变量，变量的值是 nil，因为操作空指针会引发 panic，所以我们在程序开发中要特别小心。

02结构体指针类型返回值

在调用结构体指针类型返回值的函数或方法时，并且需要操作返回值的字段或方法，此时，我们就需要注意触发空指针引发的 panic。

操作返回值的字段：

  1. func main() {
  2. user := GetUser()
  3. fmt.Println(user)
  4. fmt.Println(user.Id)
  5. }
  6.  
  7. func GetUser() (user *User) {
  8. return
  9. }
  10.  
  11. type User struct {
  12. Id int
  13. Name string
  14. }

阅读上面这段代码，我们通过调用函数 GetUser() 获取 *User 类型的返回值，因为返回值变量是空指针，当我们访问返回值的字段时，程序引发 panic。

避免此类空指针问题，一是可以在返回值包含指针类型变量的函数或方法中，在函数体开头初始化返回值的指针类型变量;二是在调用结构体指针类型返回值的函数或方法时，在操作返回值的字段或方法时，先判定返回值是否为 nil(空指针)。

  1. func main() {
  2. user := GetUser()
  3. fmt.Println(user)
  4. if user != nil {
  5. fmt.Println(user.Id)
  6. }
  7. }
  8.  
  9. func GetUser() (user *User) {
  10. user = new(User)
  11. // user = &User{}
  12. return
  13. }
  14.  
  15. type User struct {
  16. Id int
  17. Name string
  18. }

操作返回值的方法：

  1. func main() {
  2. user := GetUser()
  3. user.Login()
  4. }
  5.  
  6. func GetUser() (user *User) {
  7. return
  8. }
  9.  
  10. type User struct {
  11. Id int
  12. Name string
  13. }
  14.  
  15. func (u User) Login() {
  16.  
  17. }

阅读上面这段代码，我们通过调用函数 GetUser() 获取 *User 类型的返回值，因为返回值变量是空指针，当我们访问返回值的方法 Login() 时，程序触发空指针引发 panic。

避免此类空指针问题，一是可以在返回值是指针类型变量的函数或方法的函数体中，开头先初始化返回值的指针类型变量;二是类型方法的接收者使用指针类型。

  1. func main() {
  2. user := GetUser()
  3. user.Login()
  4. }
  5.  
  6. func GetUser() (user *User) {
  7. user = new(User)
  8. // user = &User{}
  9. return
  10. }
  11.  
  12. type User struct {
  13. Id int
  14. Name string
  15. }
  16.  
  17. func (u *User) Login() {
  18.  
  19. }

03结构体指针类型 value 的 Map

在 Golang 语言程序开发中，经常会操作结构体指针类型 value 的 Map，也需要注意触发空指针引发 panic。

  1. func main() {
  2. var userData map[int]*User
  3. fmt.Println(userData[1].Name)
  4. }
  5.  
  6. type User struct {
  7. Id int
  8. Name string
  9. }

阅读上面这段代码，我们定义 map 类型的变量 userData，key 是 int 类型，value 是结构体指针类型，我们访问 map 的值时，因为值是空指针，所以会引发 panic。

避免此类空指针问题，我们可以使用 ok-idiom 模式判断键值是否存在，如果键值存在(判断键值是否为 nil)，我们访问键值的字段，否则不访问。通过这种方式，也可以避免触发空指针引发 panic。

  1. func main() {
  2. var userData map[int]*User
  3. if val, ok := userData[1]; ok {
  4. fmt.Println(val.Name)
  5. }
  6. }
  7.  
  8. type User struct {
  9. Id int
  10. Name string
  11. }

04defer 延迟调用

关键字 defer 延迟调用函数，虽然被调用函数会延迟调用，但是被调用函数的变量会先被注册。所以，如果被调用函数的变量是空指针，就会引发 panic。

  1. func main() {
  2. res, err := http.Get("http://www.baidu2022.com/robots.txt") // 伪造错误请求
  3. defer res.Body.Close()
  4. if err != nil {
  5. log.Fatal(err)
  6. }
  7. body, err := io.ReadAll(res.Body)
  8. if err != nil {
  9. log.Fatal(err)
  10. }
  11. fmt.Printf("%s", body)
  12. }

阅读上面这段代码，使用 defer 延迟调用函数释放资源，因为我们将 defer 放在错误检查之后，所以如果返回值 res 是空指针，就会引发 panic。

避免此类空指针问题，我们可以在使用 defer 调用之前，先做错误检查，并且遇到错误后停止向下执行。

05总结

本文我们介绍一些 Golang 语言开发需要避免空指针引发 panic 的场景，虽然都比较简单，但是新手很容易踩“坑”。欢迎读者朋友们在评论区与大家分享更多因为触发空指针引发 panic 的场景。

原文链接：https://mp.weixin.qq.com/s/9s6YXJsZcXyfgWDYG-WZOQ

标签：

相关文章