元帅 Go Struct 到 BSON 以进行 mongoimport

Question

我有一段结构要写入 BSON 文件以执行 mongoimport。

这是我在做什么的粗略想法（使用 gopkg.in/mgo.v2/bson）：

type Item struct {
    ID   string `bson:"_id"`
    Text string `bson:"text"`
}

items := []Item{
    {
        ID: "abc",
        Text: "def",
    },
    {
        ID: "uvw",
        Text: "xyz",
    },
}

file, err := bson.Marshal(items)
if err != nil {
    fmt.Printf("Failed to marshal BSON file: '%s'", err.Error())
}

if err := ioutil.WriteFile("test.bson", file, 0644); err != nil {
    fmt.Printf("Failed to write BSON file: '%s'", err.Error())
}

这运行s 并生成文件，但它的格式不正确 - 相反它看起来像这样（使用 bsondump --pretty test.bson）：

{
    "1": {
        "_id": "abc",
        "text": "def"
    },
    "2": {
        "_id": "abc",
        "text": "def"
    }
}

当我认为它应该更像：

{
    "_id": "abc",
    "text": "def"
{
}
    "_id": "abc",
    "text": "def"
}

这在 Go 中可以做到吗？我只想生成一个 .bson 文件，您希望 mongodump 命令生成该文件，这样我就可以运行 mongoimport 并填充一个集合。

Answer 1

您需要独立的 BSON 文档，因此单独编组这些项目：

buf := &bytes.Buffer{}
for _, item := range items {
    data, err := bson.Marshal(item)
    if err != nil {
        fmt.Printf("Failed to marshal BSON item: '%v'", err)
    }
    buf.Write(data)
}

if err := ioutil.WriteFile("test.bson", buf.Bytes(), 0644); err != nil {
    fmt.Printf("Failed to write BSON file: '%v'", err)
}

运行 bsondump --pretty test.bson，输出将是：

{
        "_id": "abc",
        "text": "def"
}
{
        "_id": "uvw",
        "text": "xyz"
}
2022-02-09T10:23:44.886+0100    2 objects found

请注意，如果直接写入文件，则不需要缓冲区：

f, err := os.Create("test.bson")
if err != nil {
    log.Panicf("os.Create failed: %v", err)
}
defer f.Close()

for _, item := range items {
    data, err := bson.Marshal(item)
    if err != nil {
        log.Panicf("bson.Marshal failed: %v", err)
    }
    if _, err := f.Write(data); err != nil {
        log.Panicf("f.Write failed: %v", err)
    }
}

元帅 Go Struct 到 BSON 以进行 mongoimport

Marshal Go Struct to BSON for mongoimport

marshalling

go

mongodb

bson

mongoimport