基于go interface{}==nil 的几种坑及原理分析-巨人网络通讯

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

本文是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

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

以上就是 nil 相关的坑，希望大家可以牢记，如果 ”幸运“ 的遇到了，可以想到这些可能性。

补充：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（这类反向转换很难发现）。