获取 golang proto 生成的复杂结构的全部字段名

本篇文章给大家分享《获取 golang proto 生成的复杂结构的全部字段名》，覆盖了Golang的常见基础知识，其实一个语言的全部知识点一篇文章是不可能说完的，但希望通过这些问题，让读者对自己的掌握程度有一定的认识(B 数)，从而弥补自己的不足，更好的掌握它。

我正在尝试获取从 proto 生成的 go 文件中的所有字段名称。 下面是生成的结构。

type action struct {
    name             string            `protobuf:"bytes,1,opt,name=name,proto3" json:"name,omitempty"`
    // types that are valid to be assigned to actiontype:
    //  *action_taskaction
    actiontype           isaction_actiontype `protobuf_oneof:"action_type"`
}

可以看出，actiontype是proto中的一个field，其实现如下。

type isaction_actiontype interface {
    isaction_actiontype()
}

type action_taskaction struct {
    taskaction *taskaction `protobuf:"bytes,16,opt,name=task_action,json=taskaction,proto3,oneof"`
}

type taskaction struct {
    progress             float32  `protobuf:"fixed32,1,opt,name=progress,proto3" json:"progress,omitempty"`
}

因为我想获取 taskaction 结构中的字段名称，即 progress。

我使用下面的代码来获取字段名称，但如果字段类型是接口（对于 oneof 字段），则会遇到问题

func printFieldNames(t reflect.Type) error {
    for i := 0; i < t.NumField(); i++ {
        field := t.Field(i)
        if field.Type.Kind() == reflect.Struct {
            printFieldNames(field.Type)
            continue
        }
        if field.Type.Kind() == reflect.Interface {
            // what to do here.
        }
        column := field.Tag.Get("json")
        fmt.Println("column: ", column)
    }
    return nil
}

解决方案

如果类型是接口，你就不能做太多事情。在实际值中，它可能是一个结构体或实现该接口的任何其他类型，但接口 type 本身无法告诉您这一点，它不限制具体类型。

如果您以 reflect.Value 而不是 reflect.Type 开头，您可以做您想做的事情，因为如果您有一个值，您可以检查存储在接口中的值（或其类型）。要获取封装在接口值中的 reflect.value 描述符，您可以使用 reflect.Elem()。

此外，要处理指向结构的指针，您可以再次使用 reflect.elem() 来获取指向的值。您可以通过将其类型与 reflect.ptr 进行比较来检查某个值是否为指针。

这是 printfieldnames() 的示例，重写为与 reflect.value 一起使用，并且它递归到存储在接口值中的结构。这不是处理所有情况的解决方案，但演示了如何做到这一点：

func printfieldnames(v reflect.value) {
    for i := 0; i < v.numfield(); i++ {
        field := v.field(i)
        if field.kind() == reflect.ptr {
            field = field.elem()
        }
        if field.kind() == reflect.struct {
            printfieldnames(field)
            continue
        }
        if field.kind() == reflect.interface {
            wrapped := field.elem()
            if wrapped.kind() == reflect.ptr {
                wrapped = wrapped.elem()
            }
            printfieldnames(wrapped)
        }
        structfield := v.type().field(i)
        column := structfield.tag.get("json")
        fmt.printf("column: %s, json tag: %s\n", structfield.name, column)
    }
}

测试它：

a := action{
    actiontype: action_taskaction{
        taskaction: &taskaction{},
    },
}
printfieldnames(reflect.valueof(a))

输出将是（在 Go Playground 上尝试）：

column: Name, json tag: name,omitempty
column: Progress, json tag: progress,omitempty
column: ActionType, json tag:

到这里，我们也就讲完了《获取 golang proto 生成的复杂结构的全部字段名》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！