Golang计算 struct struct 中的字段数

发布于09月25日

给定由两个嵌入的 struct StructA和StructB组成的 struct CompleteStruct，其中StructB具有ImbStructC.

type StructA struct {
    AA int
    AB int
    AC int
}
type ImbStructC struct {
    BCC int
}
type StructB struct {
    BA int
    BB int
    ImbStructC
}

type CompleteStruct struct {
    StructA
    StructB
}

如何提取内部 struct 中的字段总数？

reflect.TypeOf(CompleteStruct{}).NumField())

我假设返回2，因为CompleteStruct由2个嵌入的 struct 组成.

我可以使用什么代码来显示CompleteStruct有6个字段？

推荐答案

需要递归来解决这个问题，因为嵌入的 struct 字段本身可能嵌入另一个 struct .

此外，还应注意不要将嵌入的 struct 视为字段--它们在reflect包中被列为"匿名"字段:

func countFields(v any) int {
    return rvCountFields(reflect.ValueOf(v))
}

func rvCountFields(rv reflect.Value) (count int) {

    if rv.Kind() != reflect.Struct {
        return
    }

    fs := rv.NumField()
    count += fs

    for i := 0; i < fs; i++ {
        f := rv.Field(i)
        if rv.Type().Field(i).Anonymous {
            count-- // don't count embedded structs (listed as anonymous fields) as a field
        }

        // recurse each field to see if that field is also an embedded struct
        count += rvCountFields(f)
    }

    return
}

https://go.dev/play/p/IjOllo86_xk个

输出:

main.CompleteStruct : count = 5
main.StructA        : count = 3
main.StructB        : count = 2
main.StructC        : count = 6
main.StructD        : count = 12
main.Empty          : count = 0
int                 : count = 0