You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
50 lines
1.0 KiB
50 lines
1.0 KiB
2 years ago
|
package utils
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"github.com/golang/protobuf/jsonpb"
|
||
|
|
"github.com/golang/protobuf/proto"
|
||
|
|
"reflect"
|
||
|
|
)
|
||
|
|
|
||
|
|
func JsonDataList(resp interface{}) []map[string]interface{} {
|
||
|
|
var list []proto.Message
|
||
|
|
if reflect.TypeOf(resp).Kind() == reflect.Slice {
|
||
|
|
s := reflect.ValueOf(resp)
|
||
|
|
for i := 0; i < s.Len(); i++ {
|
||
|
|
ele := s.Index(i)
|
||
|
|
list = append(list, ele.Interface().(proto.Message))
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
result := make([]map[string]interface{}, 0)
|
||
|
|
for _, v := range list {
|
||
|
|
m := ProtoToMap(v, false)
|
||
|
|
result = append(result, m)
|
||
|
|
}
|
||
|
|
return result
|
||
|
|
}
|
||
|
|
|
||
|
|
func JsonDataOne(pb proto.Message) map[string]interface{} {
|
||
|
|
return ProtoToMap(pb, false)
|
||
|
|
}
|
||
|
|
|
||
|
|
func ProtoToMap(pb proto.Message, idFix bool) map[string]interface{} {
|
||
|
|
marshaler := jsonpb.Marshaler{
|
||
|
|
OrigName: true,
|
||
|
|
EnumsAsInts: false,
|
||
|
|
EmitDefaults: false,
|
||
|
|
}
|
||
|
|
|
||
|
|
s, _ := marshaler.MarshalToString(pb)
|
||
|
|
out := make(map[string]interface{})
|
||
|
|
json.Unmarshal([]byte(s), &out)
|
||
|
|
if idFix {
|
||
|
|
if _, ok := out["id"]; ok {
|
||
|
|
out["_id"] = out["id"]
|
||
|
|
delete(out, "id")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return out
|
||
|
|
}
|