在 Go
语言中可以使用类型断言来判断变量的类型,类型断言的使用有两种方式:
- 返回两个值,分别是断言为设置类型后的值和断言成功的标志符;
- 返回一个值,表示断言为设置类型的值;
1. 返回单个值的使用
1.1 断言成功
如下代码,假如我们无法判断变量 a
的类型,可以使用下面的断言方式
func main() {
a := []int{1, 2, 3}
value := interface{}(a).([]int)
fmt.Printf("value is %#v\n", value) // value is []int{1, 2, 3}
}
其中 interface{}(a)
表示将 a
转换为接口类型,如果修改为以下语句
value := (a).([]int)
会报错误
invalid type assertion: a.([]int) (non-interface type []int on left)
意思是说 a
是非接口类型。
1.2 断言失败
看下面代码,如果我们将变量 a
断言为 int
那么会发生什么情况呢?
func main() {
a := []int{1, 2, 3}
value := interface{}(a).(int)
fmt.Printf("value is %#v\n", value)
}
输出结果为:
panic: interface conversion: interface {} is []int, not int
触发 panic
原因是断言 a
为 int
,但实际上 a
是 []int
,所以触发异常。
那怎样避免这种 panic
呢? 接着往下看
2. 返回多个值的使用
在这一小节,我们使用返回的两个值来判断断言是否成功。
func main() {
a := []int{1, 2, 3}
value, ok := interface{}(a).(int)
if ok {
fmt.Printf("value is %#v\n", value)
} else {
fmt.Println("断言失败, value is ", value)
}
}
其中 OK
表示断言结果的 bool
值,代码输出结果为:
断言失败, value is 0
在这里如果断言失败后,此时的 value 值为断言类型的零值,此处为 int
的零值。
3. 要断言的变量类型不确定
有这么一种情况是如果要断言的类型不确定,即变量的类型有可能是好几种类型中的一种,那么我们该如何应对呢?
使用 switch
语句可以帮助我们更好的做这个事情。
func main() {
a := []int{1, 2, 3}
// value, ok := interface{}(a).(int)
switch t := interface{}(a).(type) {
case int:
fmt.Println("type is int")
case string:
fmt.Println("type is string")
default:
fmt.Printf("other type %T\n", t)
}
}
注意 interface{}(a).(type)
此处使用的是 .(type)
,并且 .(type)
只能用于 switch
语句中。