Go:一种很“新”的类型断言
分享组件:go-kratos-ecosystem/components
注:该功能计划在 v2.11.0 版本中发布。PR
引言
我们在早期使用 Go 做类型断言的时候,大多是这么用:
1 | var v interface{} |
那现在,我们只需要这么用:
1 | if IsType[int](v) { |
主角
这就是今天要给大家分享的一个新方法:IsType
。这个想法来自于 关于增加 is[T any](any) bool
的提案 。
我们先看看实际的代码和单测效果。
代码:
1
2
3
4func IsType[T any](value any) bool {
_, ok := value.(T)
return ok
}单测:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24func TestIsType(t *testing.T) {
assert.True(t, IsType[int](10))
assert.False(t, IsType[int](int8(10)))
assert.True(t, IsType[string]("foo"))
assert.False(t, IsType[string](10))
assert.True(t, IsType[time.Time](time.Now()))
assert.False(t, IsType[time.Time](10))
assert.True(t, IsType[foo](foo{}))
assert.False(t, IsType[foo](10))
assert.True(t, IsType[*foo](&foo{}))
assert.False(t, IsType[foo](&foo{}))
assert.False(t, IsType[*foo](nil))
assert.True(t, IsType[testInterface](testStruct{}))
assert.True(t, IsType[interface{}](testStruct{}))
assert.True(t, IsType[any](testStruct{}))
assert.True(t, IsType[error](errors.New("foo")))
assert.False(t, IsType[error](nil))
}单测结果:
1
2
3=== RUN TestIsType
--- PASS: TestIsType (0.00s)
PASS
底层其实就是应用了 Go 1.18+ 中的泛型特性,通过这种方式,我们可以很方便地判断一个值的类型。原来三行代码,现在只需要一行代码就可以搞定。
1 | // 之前 |
性能
考虑到使用了泛型,我们来比较下使用原生和使用泛型的性能损耗。
相关脚本:https://github.com/go-kratos-ecosystem/components/pull/185/files
原生:
单测脚本:
1
2
3
4
5
6
7
8
9
10
11
12func testIsType(value any) bool {
_, ok := value.(int)
return ok
}
func BenchmarkIsType_Native(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
testIsType(10)
}
})
}单测结果:
1
2BenchmarkIsType_Native
BenchmarkIsType_Native-8 1000000000 0.2330 ns/op泛型:
单测脚本:
1
2
3
4
5
6
7func BenchmarkIsType_Generics(b *testing.B) {
b.RunParallel(func(pb *testing.PB) {
for pb.Next() {
IsType[int](10)
}
})
}单测结果:
1
2BenchmarkIsType_Generics
BenchmarkIsType_Generics-8 1000000000 0.2403 ns/op
可以看到,性能损耗几乎可以忽略不计。
应用
实际应用中,类似很多语言的 instanceof
操作,我们可以使用这种方式来判断类型。
1 | type Named interface { |