我正在从mongo中获取嵌套数据,我想将其展平到一个结构中以将其存储在csv文件中。
数据如下:
{
"_id" : "bec7bfaa-7a47-4f61-a463-5966a2b5c8ce",
"data" : {
"driver" : {
"etaToStore" : 156
},
"createdAt" : 1532590052,
"_id" : "07703a33-a3c3-4ad5-9e06-d05063474d8c"
}
}
我最终想要得到的结构应该是这样的
type EventStruct struct {
Id string `bson:"_id"`
DataId string `bson:"data._id"`
EtaToStore string `bson:"data.driver.etaToStore"`
CreatedAt int `bson:"data.createdAt"`
}
这不起作用,因此在some SO answers之后,我将其分解为多个结构:
// Creating a structure for the inner struct that I will receive from the query
type DriverStruct struct {
EtaToStore int `bson:"etaToStore"`
}
type DataStruct struct {
Id string `bson:"_id"`
Driver DriverStruct `bson:"driver"`
CreatedAt int `bson:"createdAt"`
}
// Flattenning out the structure & getting the fields we need only
type EventStruct struct {
Id string `bson:"_id"`
Data DataStruct `bson:"data"`
}
这将从Mongo查询结果中获取所有数据,但它是嵌套的:
{
"Id": "bec7bfaa-7a47-4f61-a463-5966a2b5c8ce",
"Data": {
"Id": a33-a3c3-4ad5-9e06-d05063474d8c,
"Driver": {
"EtaToStore": 156
},
"CreatedAt": 1532590052
}
}
我想结束的是:
{
"Id": "bec7bfaa-7a47-4f61-a463-5966a2b5c8ce",
"DataId": "a33-a3c3-4ad5-9e06-d05063474d8c",
"EtaToStore": 156,
"CreatedAt": 1532590052
}
我敢肯定有一个简单的方法可以做到这一点,但我不知道该怎么做,帮助!
答案 0 :(得分:1)
您可以使用与以下基本相同的逻辑:
package utils
// FlattenIntegers flattens nested slices of integers
func FlattenIntegers(slice []interface{}) []int {
var flat []int
for _, element := range slice {
switch element.(type) {
case []interface{}:
flat = append(flat, FlattenIntegers(element.([]interface{}))...)
case []int:
flat = append(flat, element.([]int)...)
case int:
flat = append(flat, element.(int))
}
}
return flat
}
(来源:https://gist.github.com/Ullaakut/cb1305ede48f2391090d57cde355074f)
通过对其进行调整以适应JSON中的内容。如果您希望它是通用的,则需要支持它可以包含的所有类型。
答案 1 :(得分:1)
您可以实现json.Unmarshaler
接口来添加自定义方法来解组json。然后,在该方法中,可以使用嵌套的struct格式,但最后返回展平的格式。
func (es *EventStruct) UnmarshalJSON(data []byte) error {
// define private models for the data format
type driverInner struct {
EtaToStore int `bson:"etaToStore"`
}
type dataInner struct {
ID string `bson:"_id" json:"_id"`
Driver driverInner `bson:"driver"`
CreatedAt int `bson:"createdAt"`
}
type nestedEvent struct {
ID string `bson:"_id"`
Data dataInner `bson:"data"`
}
var ne nestedEvent
if err := json.Unmarshal(data, &ne); err != nil {
return err
}
// create the struct in desired format
tmp := &EventStruct{
ID: ne.ID,
DataID: ne.Data.ID,
EtaToStore: ne.Data.Driver.EtaToStore,
CreatedAt: ne.Data.CreatedAt,
}
// reassign the method receiver pointer
// to store the values in the struct
*es = *tmp
return nil
}
答案 2 :(得分:1)
这个问题已有一年半的历史了,但是我今天遇到这个问题的时候是在对API更新做出反应时,这使我处于同样的情况,所以这是我的解决方案(坦白地说,我还没有使用{{ 1}},但我假设bson
和json
字段标签阅读器实现以相同的方式处理它们)
嵌入式(有时称为匿名)字段可以捕获JSON,因此您可以将多个结构组合为一个像单个结构一样的复合结构。
bson
{
"_id" : "bec7bfaa-7a47-4f61-a463-5966a2b5c8ce",
"data" : {
"driver" : {
"etaToStore" : 156
},
"createdAt" : 1532590052,
"_id" : "07703a33-a3c3-4ad5-9e06-d05063474d8c"
}
}
您可以访问嵌入式结构的嵌套字段,就像父结构包含等效字段一样,例如type DriverStruct struct {
EtaToStore string `bson:"etaToStore"`
type DataStruct struct {
DriverStruct `bson:"driver"`
DataId string `bson:"_id"`
CreatedAt int `bson:"createdAt"`
}
type EventStruct struct {
DataStruct `bson:"data"`
Id string `bson:"_id"`
}
是达到目标的有效方法。
好处:
EventStructInstance.EtaToStore
或Marshaller
接口,对于这个问题来说有点过分了