繁体   English   中英

在结构体中具有 go interface{} 字段的 protobuf 文件中使用 google.protobuf.Value,反之亦然

[英]Use google.protobuf.Value in a protobuf file with go interface{} field in a struct and vice versa

我有看起来像这样的 protobuf 消息:

// Rule : Includes an entity , an operation and a comparison value
message Rule {
  string entity = 1;
  string operator = 2;
  google.protobuf.Value value = 3;
}

我在这里使用 Value 是因为 value 可以是任何类型。

在我使用生成代码的 Golang 代码中:

type Rule struct {
 Entity string
 Operator string
 Value interface{}
}

现在,我的问题是:如何将 Kind of google.protobuf.Value (nil, number, string, bool, struct, list) 表示的类型转换为 interface{} 并在运行时将 interface{} 的动态类型转换回输入 google.protobuf.Value ?

我真的可以使用一些帮助,在这里找到了一个排序解决方案( https://go.googlesource.com/protobuf/+/4c057caa84dc5e1bc6bf0d9fe8af9180a6151e07/internal/value/convert.go )但希望有一种更简单的方法来解决这个问题。

为了完整起见评论:

func convertToValue(v interface{}, from reflect.Type) (*types.Value, error) {
    switch from.Kind() {
    case reflect.String:
        if reflect.ValueOf(v).String() == "true" || reflect.ValueOf(v).String() == "false" {
            boolVal, err := strconv.ParseBool(reflect.ValueOf(v).String())
            if err != nil {
                return nil, err
            }
            return &types.Value{Kind:&types.Value_BoolValue{BoolValue:boolVal}}, nil
        }
        return &types.Value{Kind:&types.Value_StringValue{StringValue:reflect.ValueOf(v).String()}}, nil
    case reflect.Int64:
    case reflect.Float64:
        return &types.Value{Kind:&types.Value_NumberValue{NumberValue:reflect.ValueOf(v).Float()}}, nil
    case reflect.Slice:
        list := reflect.ValueOf(v)
        outputList := make([]*types.Value,0)
        for i:=0 ; i < list.Len(); i++ {
            val, err := convertToValue(list.Index(i).Interface(), reflect.TypeOf(list.Index(i).Interface()))
            if err != nil {
                return nil, err
            }
            outputList = append(outputList, val)
        }
        return &types.Value{Kind:&types.Value_ListValue{ListValue:&types.ListValue{Values:outputList}}}, nil
    case reflect.Struct:
        valStruct := reflect.ValueOf(v)
        outputMap := make(map[string]*types.Value)
        numOfField := valStruct.NumField()
        for i := 0; i < numOfField; i++ {
            val, err := convertToValue(valStruct.Field(i).Interface(), reflect.TypeOf(valStruct.Field(i).Interface()))
            if err != nil {
                return nil, err
            }
            outputMap[valStruct.Field(i).Type().Name()] = val
        }

        return &types.Value{Kind:&types.Value_StructValue{StructValue:&types.Struct{Fields:outputMap}}}, nil
    default:
        return nil, nil
}

func convertValueToInterface(value types.Value) (interface{}, error) {

    switch {
    case _ , ok := value.Kind.(*types.Value_NullValue); ok:
        return nil, nil
    case x , ok := value.Kind.(*types.Value_StringValue); ok:
        return x.StringValue, nil
    case x , ok := value.Kind.(*types.Value_NumberValue); ok:
        return x.NumberValue, nil
    case x , ok := value.Kind.(*types.Value_BoolValue); ok:
        return strconv.FormatBool(x.BoolValue), nil
    case x , ok := value.Kind.(*types.Value_ListValue); ok:
        if x == nil || x.ListValue == nil || x.ListValue.Values == nil {
            return nil, nil
        }
        listValue := x.ListValue.Values
        if len(listValue) == 0 {
            return nil, nil
        }
        val, err := convertValueToInterface(*listValue[0])
        if err != nil {
            return nil, err
        }
        typ := reflect.TypeOf(val)
        outputList := reflect.MakeSlice(reflect.SliceOf(typ),0,0)
        for _, value := range listValue {
            if value != nil {
                val, err := convertValueToInterface(*value)
                if err != nil {
                    return nil, err
                }
                outputList = reflect.Append(outputList, reflect.ValueOf(val))
            }
        }
        return outputList.Interface(), nil
    case x , ok := value.Kind.(*types.Value_StructValue); ok:
        if x == nil || x.StructValue == nil || x.StructValue.Fields == nil {
            return nil, nil
        }
        mapValue := x.StructValue.Fields
        var keyTyp reflect.Type
        var typ reflect.Type
        for key,value := range mapValue {
            if value != nil {
                val, err := convertValueToInterface(*value)
                if err != nil {
                    return nil, err
                }
                keyTyp = reflect.TypeOf(key)
                typ = reflect.TypeOf(val)
                break
            } else {
                return nil, nil
            }
        }
        outputMap := reflect.MakeMap(reflect.MapOf(keyTyp, typ))
        for key,value := range mapValue {
            if value != nil {
                val, err := convertValueToInterface(*value)
                if err != nil {
                    return nil, err
                }
                outputMap.SetMapIndex(reflect.ValueOf(key), reflect.ValueOf(val))
            }
        }
        return outputMap.Interface(), nil
    default:
        return nil, nil
}

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM