登录
首页 >  Golang >  Go问答

测试嵌套结构中的 nil 值

来源:Golang技术栈

时间:2023-03-05 09:40:42 394浏览 收藏

你在学习Golang相关的知识吗?本文《测试嵌套结构中的 nil 值》,主要介绍的内容就涉及到golang,如果你想提升自己的开发能力,就不要错过这篇文章,大家要知道编程理论基础和实战操作都是不可或缺的哦!

问题内容

我有一个深度嵌套的结构。这些是由 json unmarshaller 构建的。

然而,这个结构中的一些字段是“省略的”,所以我以一个可以在各个地方有 nills 的结构结束 op。

示例(真实的嵌套更深,而且很大:400 行结构):

package main

import "fmt"

type Foo struct {
    Foo string
    Bar *Bar
}

type Bar struct {
    Bar string
    Baz *Baz
}

type Baz struct {
    Baz string
}

func main() {
    f1 := Foo{Foo: "f1"}
    f2 := Foo{Foo: "f2", Bar: &Bar{Bar: "br2"}}
    f3 := Foo{Foo: "f3", Bar: &Bar{Bar: "br3", Baz: &Baz{Baz: "bz3"}}}

    fmt.Println(f3.Bar.Baz.Baz) //-> bz3
    fmt.Println(f2.Bar.Baz.Baz) //-> panic: runtime error: invalid memory address or nil pointer dereference
    fmt.Println(f1.Bar.Baz.Baz) //-> panic: runtime error: invalid memory address or nil pointer dereference    
    //so far so good, but

    //is there a more generic way to do this kind of testing?
    if f2.Bar != nil && f2.Bar.Baz != nil {
        fmt.Println(f2.Bar.Baz.Baz)
    } else {
        fmt.Println("something nil")
    }
}

问题是是否有更通用的方法来测试参考树中的某个节点是否为零?我需要得到很多不同的项目,编写所有这些 if 语句会很痛苦。哦,速度令人担忧。

正确答案

处理它的一种优雅方式(在我看来)是将 getter 添加到用作指针的结构中。[protobuf](https://developers.google.com/protocol- buffers/docs/gotutorial)生成的 Go 代码也使用了这种“技术” ,它允许方法调用的自然链接,而不必担心由于nil指针导致的运行时恐慌。

在您的示例中,BarandBaz结构用作指针,因此用 getter 武装它们。重点是添加带有指针接收器的方法,首先必须检查接收器是否为nil. 如果是,则返回结果类型的零值。如果不是,则继续返回结构的字段:

func (b *Bar) GetBaz() *Baz {
    if b == nil {
        return nil
    }
    return b.Baz
}

func (b *Baz) GetBaz() string {
    if b == nil {
        return ""
    }
    return b.Baz
}

带有指针接收器的方法的好处是您可以使用接收器调用它们nil。在您尝试引用它们的字段之前,它不会导致运行时恐慌,而我们不会,这就是为什么我们首先检查接收器是否是nil(最终,接收器充当正常参数

  • 传递nil作为指针参数)。

有了上面的 getter,使用就被简化了,在这些例子中都不会出现运行时恐慌:

fmt.Println(f3.Bar.GetBaz().GetBaz()) // naturally no panic
fmt.Println(f2.Bar.GetBaz().GetBaz()) // No panic
fmt.Println(f1.Bar.GetBaz().GetBaz()) // No panic

if baz := f2.Bar.GetBaz(); baz != nil {
    fmt.Println(baz.GetBaz())
} else {
    fmt.Println("something nil")
}

在Go Playground上尝试一下。

到这里,我们也就讲完了《测试嵌套结构中的 nil 值》的内容了。个人认为,基础知识的学习和巩固,是为了更好的将其运用到项目中,欢迎关注golang学习网公众号,带你了解更多关于golang的知识点!

声明:本文转载于:Golang技术栈 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>