一些背景
首先让我们弄清楚一些与类型有关的事情。引用自Spec: Types:
类型确定特定于该类型值的一组值和操作。类型可以是命名的或未命名的。命名类型由(可能是qualified)type name 指定;未命名的类型使用 类型文字 指定,它从现有类型组成新类型。
所以有(预先声明的)命名类型,例如string、int 等,您还可以使用type declarations 创建新的命名类型(其中涉及type 关键字)比如type MyInt int。还有未命名类型,它们是类型文字(应用于/包括命名或未命名类型)的结果,例如[]int、struct{i int}、*int等等。
您可以使用Type.Name() 方法获取命名类型的名称,该方法“为未命名类型返回一个空字符串”:
var i int = 2
fmt.Printf("%q\n", reflect.TypeOf("abc").Name()) // Named: "string"
fmt.Printf("%q\n", reflect.TypeOf(int(2)).Name()) // Named: "int"
fmt.Printf("%q\n", reflect.TypeOf([]int{}).Name()) // Unnamed: ""
fmt.Printf("%q\n", reflect.TypeOf(struct{ i int }{}).Name()) // Unnamed: ""
fmt.Printf("%q\n", reflect.TypeOf(&struct{ i int }{}).Name()) // Unnamed: ""
fmt.Printf("%q\n", reflect.TypeOf(&i).Name()) // Unnamed: ""
有些类型是预先声明的,可供您使用(按原样,或在类型文字中):
布尔、数字和字符串类型的命名实例是predeclared。复合类型——数组、结构、指针、函数、接口、切片、映射和通道类型——可以使用类型文字来构造。
预先声明的类型是:
bool byte complex64 complex128 error float32 float64
int int8 int16 int32 int64 rune string
uint uint8 uint16 uint32 uint64 uintptr
您可以使用Type.PkgPath() 获取已命名 类型的包路径,如果该类型已预先声明(string、error)或未命名(*T、 struct{}, []int),包路径为空字符串":
fmt.Printf("%q\n", reflect.TypeOf("abc").PkgPath()) // Predeclared: ""
fmt.Printf("%q\n", reflect.TypeOf(A{}).PkgPath()) // Named: "main"
fmt.Printf("%q\n", reflect.TypeOf([]byte{}).PkgPath()) // Unnamed: ""
所以您有 2 个工具可供您使用:Type.Name() 用于判断该类型是否为 已命名 类型,Type.PkgPath() 用于判断该类型是否未预先声明 并且是一个命名类型。
但必须小心。如果您在类型文字中使用自己的命名类型来构造新类型(例如[]A),那将是未命名类型(如果您不使用type 关键字来构造新的命名类型) :
type ASlice []A
fmt.Printf("%q\n", reflect.TypeOf([]A{}).PkgPath()) // Also unnamed: ""
fmt.Printf("%q\n", reflect.TypeOf(ASlice{}).PkgPath()) // Named: "main"
在这种情况下你能做什么?如果类型的Kind 是Array、Chan、Map、Ptr 或Slice,则可以使用Type.Elem() 获取类型的元素类型(否则Type.Elem() 恐慌):
fmt.Printf("%q\n", reflect.TypeOf([]A{}).Elem().Name()) // Element type: "A"
fmt.Printf("%q\n", reflect.TypeOf([]A{}).Elem().PkgPath()) // Which is named, so: "main"
总结
Type.PkgPath() 可用于“过滤掉”预先声明和未命名的类型。如果PkgPath() 返回一个非空字符串,您可以确定它是“自定义”类型。如果它返回一个空字符串,它仍然可能是从“自定义”类型构造的未命名类型(在这种情况下Type.Name() 返回"");为此,您可以使用Type.Elem() 来查看它是否由“自定义”类型构造而成,这可能必须递归地应用:
// [][]A -> Elem() -> []A which is still unnamed: ""
fmt.Printf("%q\n", reflect.TypeOf([][]A{}).Elem().PkgPath())
// [][]A -> Elem() -> []A -> Elem() -> A which is named: "main"
fmt.Printf("%q\n", reflect.TypeOf([][]A{}).Elem().Elem().PkgPath())
尝试Go Playground 上的所有示例。
特殊情况 #1:匿名结构类型
还有匿名结构类型的情况,它是未命名的,但它可能有一个“自定义”类型的字段。这种情况可以通过遍历结构类型的字段并对每个字段执行相同的检查来处理,如果发现其中任何一个是“自定义”类型,我们可以将整个结构类型声明为“自定义” .
特殊情况 #2:地图类型
在地图的情况下,如果它的任何键或值类型是“自定义”,我们可能会考虑未命名的地图类型“自定义”。
map的value类型可以用上面提到的Type.Elem()方法查询,map的key类型可以用Type.Key()方法查询——如果是maps我们也要检查这个。
示例实现
func isCustom(t reflect.Type) bool {
if t.PkgPath() != "" {
return true
}
if k := t.Kind(); k == reflect.Array || k == reflect.Chan || k == reflect.Map ||
k == reflect.Ptr || k == reflect.Slice {
return isCustom(t.Elem()) || k == reflect.Map && isCustom(t.Key())
} else if k == reflect.Struct {
for i := t.NumField() - 1; i >= 0; i-- {
if isCustom(t.Field(i).Type) {
return true
}
}
}
return false
}
测试它(在Go Playground 上试试):
type K int
var i int = 2
fmt.Println(isCustom(reflect.TypeOf(""))) // false
fmt.Println(isCustom(reflect.TypeOf(int(2)))) // false
fmt.Println(isCustom(reflect.TypeOf([]int{}))) // false
fmt.Println(isCustom(reflect.TypeOf(struct{ i int }{}))) // false
fmt.Println(isCustom(reflect.TypeOf(&i))) // false
fmt.Println(isCustom(reflect.TypeOf(map[string]int{}))) // false
fmt.Println(isCustom(reflect.TypeOf(A{}))) // true
fmt.Println(isCustom(reflect.TypeOf(&A{}))) // true
fmt.Println(isCustom(reflect.TypeOf([]A{}))) // true
fmt.Println(isCustom(reflect.TypeOf([][]A{}))) // true
fmt.Println(isCustom(reflect.TypeOf(struct{ a A }{}))) // true
fmt.Println(isCustom(reflect.TypeOf(map[K]int{}))) // true
fmt.Println(isCustom(reflect.TypeOf(map[string]K{}))) // true