Go语言把map接口转换为google.protobuf.Struct

以下是go语言把map[string]interface{}转换为google.protobuf.Struct:

package pb
import (
"fmt"
"reflect"
st "github.com/golang/protobuf/ptypes/struct"
)
// ToStruct转换map[string]interface{} 为ptypes.Struct
func ToStruct(v map[string]interface{}) *st.Struct {
size := len(v)
if size == 0 {
return nil
}
fields := make(map[string]*st.Value, size)
for k, v := range v {
fields[k] = ToValue(v)
}
return &st.Struct{
Fields: fields,
}
}
// ToValue 转换interface{} 为ptypes.Value
func ToValue(v interface{}) *st.Value {
switch v := v.(type) {
case nil:
return nil
case bool:
return &st.Value{
Kind: &st.Value_BoolValue{
BoolValue: v,
},
}
case int:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case int8:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case int32:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case int64:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case uint:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case uint8:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case uint32:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case uint64:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case float32:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v),
},
}
case float64:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: v,
},
}
case string:
return &st.Value{
Kind: &st.Value_StringValue{
StringValue: v,
},
}
case error:
return &st.Value{
Kind: &st.Value_StringValue{
StringValue: v.Error(),
},
}
default:
// 回退为其他类型
return toValue(reflect.ValueOf(v))
}
}
func toValue(v reflect.Value) *st.Value {
switch v.Kind() {
case reflect.Bool:
return &st.Value{
Kind: &st.Value_BoolValue{
BoolValue: v.Bool(),
},
}
case reflect.Int, reflect.Int8, reflect.Int16, reflect.Int32, reflect.Int64:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v.Int()),
},
}
case reflect.Uint, reflect.Uint8, reflect.Uint16, reflect.Uint32, reflect.Uint64, reflect.Uintptr:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: float64(v.Uint()),
},
}
case reflect.Float32, reflect.Float64:
return &st.Value{
Kind: &st.Value_NumberValue{
NumberValue: v.Float(),
},
}
case reflect.Ptr:
if v.IsNil() {
return nil
}
return toValue(reflect.Indirect(v))
case reflect.Array, reflect.Slice:
size := v.Len()
if size == 0 {
return nil
}
values := make([]*st.Value, size)
for i := 0; i < size; i++ {
values[i] = toValue(v.Index(i))
}
return &st.Value{
Kind: &st.Value_ListValue{
ListValue: &st.ListValue{
Values: values,
},
},
}
case reflect.Struct:
t := v.Type()
size := v.NumField()
if size == 0 {
return nil
}
fields := make(map[string]*st.Value, size)
for i := 0; i < size; i++ {
name := t.Field(i).Name
if len(name) > 0 && 'A' <= name[0] && name[0] <= 'Z' {
fields[name] = toValue(v.Field(i))
}
}
if len(fields) == 0 {
return nil
}
return &st.Value{
Kind: &st.Value_StructValue{
StructValue: &st.Struct{
Fields: fields,
},
},
}
case reflect.Map:
keys := v.MapKeys()
if len(keys) == 0 {
return nil
}
fields := make(map[string]*st.Value, len(keys))
for _, k := range keys {
if k.Kind() == reflect.String {
fields[k.String()] = toValue(v.MapIndex(k))
}
}
if len(fields) == 0 {
return nil
}
return &st.Value{
Kind: &st.Value_StructValue{
StructValue: &st.Struct{
Fields: fields,
},
},
}
default:
// 最后排序
return &st.Value{
Kind: &st.Value_StringValue{
StringValue: fmt.Sprint(v),
},
}
}
}

版权声明:著作权归作者所有。

相关推荐

Java把LocalDate转换为Date

有时需要在LocalDate和Date之间互转。LocalDate转换为DateLocalDate localDate = LocalDate.now();System.out.println("LocalDate = " + localDate);Date date = Date.from(localDate.atStartOfDay(ZoneId.systemDefault()).toInst

Java 8使用Collections toMap实现List转换为Map

Collections提供了toMap()来实现集合转换为Map。方法如下:Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper, Function<? super T, ? extends U> valueMapper) keyMapper:实现键的

Go语言日志框架logrus封装

logrus在github上是一个star数比较高的go日志框架,简单封装如下:package logimport ( "fmt" "github.com/Sirupsen/logrus" "runtime" "strings")var logger = logrus.New()// 封装logrus.Fieldstype Fields logrus.Fieldsfunc SetLogLevel(

Java 9数组转换为List

Java 9使用List.of在Java 9,List新增了of()的工厂方法用来产生不可变的列表。返回一个空列表List list = List.of(); 多个元素生成列表List<Integer> list = List.of(1,2,3,4,5); 数组转换为列表Integer[] arr

Java8使用Lambda把List转换为Map

使用Java8的Collections可以很简便地把List转换为Map以User为例List里的User名字唯一Map<String, User> result =     users.stream().collect(Collectors.toMap(User::getName,   &nbs

TypeScript:字符串转换为数字

在TypeScript,String转换为Number有以下方式:Number构造函数(推荐)let n = Number('1234') // 1234 let nan = Number('abcd') // NaN +号操作符let n = +'1234'