簡體   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