当前位置 博文首页 > 基于go interface{}==nil 的几种坑及原理分析

    基于go interface{}==nil 的几种坑及原理分析

    作者:mukebb 时间:2021-06-03 18:11

    本文是Go比较有名的一个坑,在以前面试的时候也被问过,为什么想起来写这个?

    因为我们线上就真实出现过这个坑,写给不了解的人在使用 if err != nil 的时候提高警惕。

    Go语言的interface{}在使用过程中有一个特别坑的特性,当你比较一个interface{}类型的值是否是nil的时候,这是需要特别注意避免的问题。

    先来看看一个demo:

    package main
    import "fmt"
    type ErrorImpl struct{}
    func (e *ErrorImpl) Error() string {
       return ""
    }
    var ei *ErrorImpl
    var e error
    func ErrorImplFun() error {
       return ei
    }
    func main() {
       f := ErrorImplFun()
       fmt.Println(f == nil)
    }
    

    输出:

    false

    为什么不是true?

    想要理解这个问题,首先需要理解interface{}变量的本质。在Go语言中,一个interface{}类型的变量包含了2个指针,一个指针指向值的在编译时确定的类型,另外一个指针指向实际的值。

    // InterfaceStructure 定义了一个interface{}的内部结构
    type InterfaceStructure struct {
      pt uintptr // 到值类型的指针
      pv uintptr // 到值内容的指针
    }
    // asInterfaceStructure 将一个interface{}转换为InterfaceStructure
    func asInterfaceStructure(i interface{}) InterfaceStructure {
      return *(*InterfaceStructure)(unsafe.Pointer(&i))
    }
    func main() {
      var i1, i2 interface{}
      var v1 int = 23
      var v2 int = 23
      i1 = v1
      i2 = v2
      fmt.Printf("sizeof interface{} = %d\n", unsafe.Sizeof(i1))
      fmt.Printf("i1 %v %+v\n", i1, asInterfaceStructure(i1))
      fmt.Printf("i2 %v %+v\n", i2, asInterfaceStructure(i2))
      var nilInterface interface{}
      var str *string
      fmt.Printf("nil interface = %+v\n", asInterfaceStructure(nilInterface))
      fmt.Printf("nil string = %+v\n", asInterfaceStructure(str))
      fmt.Printf("nil = %+v\n", asInterfaceStructure(nil))
    }
    

    输出:

    sizeof interface{} = 16

    i1 23 {pt:4812032 pv:825741246928}

    i2 23 {pt:4812032 pv:825741246936}

    nil interface = {pt:0 pv:0}

    nil string = {pt:4802400 pv:0}

    nil = {pt:0 pv:0}

    当我们将一个具体类型的值赋值给一个interface{}类型的变量的时候,就同时把类型和值都赋值给了interface{}里的两个指针。如果这个具体类型的值是nil的话,interface{}变量依然会存储对应的类型指针和值指针。

    如何解决?

    方法一

    返回的结果进行非nil检查,然后再赋值给interface{}变量

    type ErrorImpl struct{}
    func (e *ErrorImpl) Error() string {
       return ""
    }
    var ei *ErrorImpl
    var e error
    func ErrorImplFun() error {
       if ei == nil {
          return nil
       }
       return ei
    }
    func main() {
       f := ErrorImplFun()
       fmt.Println(f == nil)
    }
    

    输出:

    true

    方法二

    返回具体实现的类型而不是interface{}

    package main
    import "fmt"
    type ErrorImpl struct{}
    func (e *ErrorImpl) Error() string {
       return ""
    }
    var ei *ErrorImpl
    var e error
    func ErrorImplFun() *ErrorImpl {
       return ei
    }
    func main() {
       f := ErrorImplFun()
       fmt.Println(f == nil)
    }
    

    输出:

    true

    解决由于第三方包带来的坑

    由于有的error是第三方包返回的,又自己不想改第三方包,只好接收处理的时候想办法。

    方法一

    利用interface{}原理

     is:=*(*InterfaceStructure)(unsafe.Pointer(&i))
     if is.pt==0 && is.pv==0 {
         //is nil do something
     }
    

    将底层指向值和指向值的类型的指针打印出来如果都是0,表示是nil

    方法二

    利用断言,断言出来具体类型再判断非空

    type ErrorImpl struct{}
    func (e ErrorImpl) Error() string {
       return "demo"
    }
    var ei *ErrorImpl
    var e error
    func ErrorImplFun() error {
       //ei = &ErrorImpl{}
       return ei
    }
    func main() {
       f := ErrorImplFun()
       //当然error实现类型较多的话使用  
     //switch case方式断言更清晰
       res, ok := f.(*ErrorImpl)
       fmt.Printf("ok:%v,f:%v,res:%v", 
       ok, f == nil, res == nil)
    }
    

    输出:

    ok:true,f:false,res:true

    方法三

    利用反射

    type ErrorImpl struct{}
    func (e ErrorImpl) Error() string {
       return "demo"
    }
    var ei *ErrorImpl
    var e error
    func ErrorImplFun() error {
       //ei = &ErrorImpl{}
       return ei
    }
    func main() {
       f := ErrorImplFun()
       rv := reflect.ValueOf(f)
       fmt.Printf("%v", rv.IsNil())
    }
    

    输出:

    true

    注意⚠:

    断言和反射性能不是特别好,如果不得已再使用,控制使用有助于提升程序性能。

    由于函数接收类型导致的panic:

    type ErrorImpl struct{}
    func (e ErrorImpl) Error() string {
       return "demo"
    }
    var ei *ErrorImpl
    var e error
    func ErrorImplFun() error {
       return ei
    }
    func main() {
       f := ErrorImplFun()
       fmt.Printf(f.Error())
    }
    

    输出:

    panic: value method main.ErrorImpl.Error called using nil *ErrorImpl pointer

    解决:

    func (e *ErrorImpl) Error() string {
       return "demo"
    }

    输出:

    demo

    可以发现将接收类型变成指针类型就可以了。

    补充:go 语言 interface{} 的易错点

    如果说 goroutine 和 channel 是 go 语言并发的两大基石,那 interface 就是 go 语言类型抽象的关键。

    在实际项目中,几乎所有的数据结构最底层都是接口类型。

    说起 C++ 语言,我们立即能想到是三个名词:封装、继承、多态。go 语言虽然没有严格意义上的对象,但通过 interface,可以说是实现了多态性。(由以组合结构体实现了封装、继承的特性)

    package main
    type animal interface {
        Move()
    }
    type bird struct{}
    func (self *bird) Move() {
        println("bird move")
    }
    type beast struct{}
    func (self *beast) Move() {
        println("beast move")
    }
    func animalMove(v animal) {
        v.Move()
    }
    func main() {
        var a *bird
        var b *beast
        animalMove(a) // bird move
        animalMove(b) // beast move
    }

    go 语言中支持将 method、struct、struct 中成员定义为 interface 类型,使用 struct 举一个简单的栗子

    使用 go 语言的 interface 特性,就能实现多态性,进行泛型编程。

    二,interface 原理

    如果没有充分了解 interface 的本质,就直接使用,那最终肯定会踩到很深的坑,要用就先要了解,先来看看 interface 源码

     type eface struct {
         _type *_type
         data  unsafe.Pointer
     }  
     type _type struct {
         size       uintptr // type size
         ptrdata    uintptr // size of memory prefix holding all pointers
         hash       uint32  // hash of type; avoids computation in hash tables
         tflag      tflag   // extra type information flags
         align      uint8   // alignment of variable with this type
         fieldalign uint8   // alignment of struct field with this type
         kind       uint8   // enumeration for C
         alg        *typeAlg  // algorithm table
         gcdata    *byte    // garbage collection data
         str       nameOff  // string form
         ptrToThis typeOff  // type for pointer to this type, may be zero
     }

    可以看到 interface 变量之所以可以接收任何类型变量,是因为其本质是一个对象,并记录其类型和数据块的指针。(其实 interface 的源码还包含函数结构和内存分布,由于不是本文重点,有兴趣的同学可以自行了解)

    三,interface 判空的坑

    对于一个空对象,我们往往通过 if v == nil 的条件语句判断其是否为空,但在代码中充斥着 interface 类型的情况下,很多时候判空都并不是我们想要的结果(其实了解或聪明的同学从上述 interface 的本质是对象已经知道我想要说的是什么)

    package main 
     type animal interface {
         Move()
     } 
     type bird struct{} 
     func (self *bird) Move() {
         println("bird move")
     } 
     type beast struct{} 
     func (self *beast) Move() {
         println("beast move")
     } 
     func animalMove(v animal) {
         if v == nil {
             println("nil animal")
         }
         v.Move()
     } 
     func main() {
         var a *bird   // nil
         var b *beast  // nil
         animalMove(a) // bird move
         animalMove(b) // beast move
     }

    还是刚才的栗子,其实在 go 语言中 var a *bird 这种写法,a 只是声明了其类型,但并没有申请一块空间,所以这时候 a 本质还是指向空指针,但我们在 aminalMove 函数进行判空是失败的,并且下面的 v.Move() 的调用也是成功的,本质的原因就是因为 interface 是一个对象,在进行函数调用的时候,就会将 bird 类型的空指针进行隐式转换,转换成实例的 interface animal 对象,所以这时候 v 其实并不是空,而是其 data 变量指向了空。

    这时候看着执行都正常,那什么情况下坑才会绊倒我们呢?只需要加一段代码

    package main 
     type animal interface {
         Move()
     } 
     type bird struct {
        name string
     } 
     func (self *bird) Move() {
         println("bird move %s", self.name) // panic
     } 
     type beast struct {
         name string
     } 
     func (self *beast) Move() {
         println("beast move %s", self.name) // panic
     } 
     func animalMove(v animal) {
         if v == nil {
             println("nil animal")
         }
         v.Move()
     } 
     func main() {
         var a *bird   // nil
         var b *beast  // nil
         animalMove(a) // panic
         animalMove(b) // panic
     }

    在代码中,我们给派生类添加 name 变量,并在函数的实现中进行调用,就会发生 panic,这时候的 self 其实是 nil 指针。所以这里坑就出来了。

    有些人觉得这类错误谨慎一些还是可以避免的,那是因为我们是正向思维去代入接口,但如果反向编程就容易造成很难发现的 bug

    package main 
     type animal interface {
         Move()
     } 
     type bird struct {
         name string
     } 
     func (self *bird) Move() {
         println("bird move %s", self.name)
     } 
     type beast struct {
         name string
     } 
     func (self *beast) Move() {
         println("beast move %s", self.name)
     } 
     func animalMove(v animal) {
         if v == nil {
             println("nil animal")
         }
         v.Move()
     } 
     func getBirdAnimal(name string) *bird {
         if name != "" {
             return &bird{name: name}
         }
         return nil
     } 
     func main() {
         var a animal
         var b animal
         a = getBirdAnimal("big bird")
         b = getBirdAnimal("") // return interface{data:nil}
         animalMove(a) // bird move big bird
         animalMove(b) // panic
     }
    

    这里我们看到通过函数返回实例类型指针,当返回 nil 时,因为接收的变量为接口类型,所以进行了隐性转换再次导致了 panic(这类反向转换很难发现)。

    那我们如何处理上述这类问题呢。我这边整理了三个点

    1,充分了解 interface 原理,使用过程中需要谨慎小心

    2,谨慎使用泛型编程,接收变量使用接口类型,也需要保证接口返回为接口类型,而不应该是实例类型

    3,判空是使用反射 typeOf 和 valueOf 转换成实例对象后再进行判空

    js