我們優雅判斷 interface 是否為 nil
背景
很久之前發過一篇文章:《10個令人驚嘆的Go語言技巧,讓你的代碼更加優雅》,這篇文章中第八點有一處錯誤的地方被認真的讀者發現了:
圖片
于是我有空之后,立馬重新看了那篇文章的內容,確實是存在讀者所說的問題。
問題
問題就在于下面這句話,文章也是有列出的:
即使接口持有的值為 nil,也不意味著接口本身為 nil。
但是在執行以下語句的時候,是有可能報 panic 的:
return reflect.ValueOf(x).IsNil()
而輸出也是非常明顯的指出錯誤:
panic: reflect: call of reflect.Value.IsNil on int Value
因為不可 nil 的 interface 是不能使用 reflect.Value.IsNil 方法。
那么問題就很好解決了。
解決方式
我們在執行 reflect.Value.IsNil 方法之前,進行一次判斷是否為指針即可:
func IsNil(x interface{}) bool {
if x == nil {
return true
}
rv := reflect.ValueOf(x)
return rv.Kind() == reflect.Ptr && rv.IsNil()
}
重點在于 rv.Kind() == reflect.Ptr && rv.IsNil() 這段代碼。
這段代碼的作用:
- 判斷 x 的類型是否為指針。
- 判斷 x 的值是否真的為 nil。
下面我們使用幾種常見的數據類型來進行測試:
func IsNil(x interface{}) bool {
if x == nil {
return true
}
rv := reflect.ValueOf(x)
return rv.Kind() == reflect.Ptr && rv.IsNil()
}
func main() {
fmt.Printf("int IsNil: %t\n", IsNil(returnInt()))
fmt.Printf("intPtr IsNil: %t\n", IsNil(returnIntPtr()))
fmt.Printf("slice IsNil: %t\n", IsNil(returnSlice()))
fmt.Printf("map IsNil: %t\n", IsNil(returnMap()))
fmt.Printf("interface IsNil: %t\n", IsNil(returnInterface()))
fmt.Printf("structPtr IsNil: %t\n", IsNil(returnStructPtr()))
}
func returnInt() interface{} {
var value int
return value
}
func returnIntPtr() interface{} {
var value *int
return value
}
func returnSlice() interface{} {
var value []string
return value
}
func returnMap() interface{} {
var value map[string]struct{}
return value
}
func returnInterface() interface{} {
var value interface{}
return value
}
type People struct {
Name string
}
func returnStructPtr() interface{} {
var value *People
return value
}
我們先后使用了 int、*int、slice、map、interface{}、自定義結構體 來測試此 IsNil 方法。運行程序輸出為:
int IsNil: false
intPtr IsNil: true
slice IsNil: false
map IsNil: false
interface IsNil: true
structPtr IsNil: true
從測試結果來看,目前是符合我們對此方法的定位的。