基于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(这类反向转换很难发现)。
那我们如何处理上述这类问题呢。我这边整理了三个点
1,充分了解 interface 原理,使用过程中需要谨慎小心
2,谨慎使用泛型编程,接收变量使用接口类型,也需要保证接口返回为接口类型,而不应该是实例类型
3,判空是使用反射 typeof 和 valueof 转换成实例对象后再进行判空