使用 Golang 的 mongo-driver 实现 MongoDB 自增 ID 自动化

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《使用 Golang 的 mongo-driver 实现 MongoDB 自增 ID 自动化》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

根据文档，go.mongodb.org/mongo-driver 似乎没有提供在更新插入未提供 id 的文档时自动递增 id 的方法。

type Document struct {
        ID             int    `bson:"_id"`
        Foo            string `bson:"foo"`
    }

    document := &Document{Foo: "test"}

    filter := bson.M{"_id": bson.M{"$eq": document.ID}}
    update := bson.M{"$set": document}

    res, err := mongoClient.Database(dbName).
        Collection(collectionName).
        UpdateOne(ctx, filter, update,
            options.Update().SetUpsert(true))

在上面的代码示例中，id 将默认为 int 的零值，即 0，并将在 mongodb 中持久化为 {"_id":0,"foo":"test"}。

当未使用 mongo-driver 提供 id 时，是否有一种干净的方法自动递增 id，而无需自己执行跟踪最后一个 id 的逻辑？例如，假设数据库中已经有 5 个文档，那么在未提供 id 的情况下，运行上面的代码将保留 {"_id":6,"foo":"test"} 。

解决方案

我发现了同样的问题，我认为的解决方案是定义没有id的document结构：

type document struct {
    foo string `bson:"foo"`
}

然后，如果使用 mongo-driver 执行 insertone 操作：

res, err := mongoclient.database(dbname).collection(collectionname).insertone(ctx, document)

_id将在数据库中自动创建（您可以多次重复insertone操作，新的_id将会出现）。

mongodb 文档解释了这种行为：“如果文档未指定 _id 字段，那么 mongod 将在插入之前添加 _id 字段并为文档分配唯一的 objectid。”（您可以阅读更多详细信息https://docs.mongodb.com/manual/reference/method/db.collection.insertOne/#db.collection.insertOne）

如果您出于某种原因需要新创建的_id，您可以使用以下代码片段作为检索它的参考：

fmt.Println("New Document created with mongodb _id: " + res.InsertedID.(primitive.ObjectID).Hex())

（primitive.objectid 与此相关：导入“go.mongodb.org/mongo-driver/bson/primitive”）

希望这有帮助！

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。