MongoDB 如何实现自增主键？Golang 代码示例

MongoDB 可以设置自增主键，但其实现方式与传统关系型数据库不同。在 MongoDB 中，可以使用名为'计数器'（counter）的集合来实现自增主键。

具体实现步骤如下：

创建一个计数器集合，用于存储当前主键的值。

db.createCollection('counters')

在计数器集合中插入一条记录，用于初始化主键值。

db.counters.insert({_id: 'user_id', seq: 0})

在插入新文档时，先查询计数器集合，获取当前主键值，然后将其加 1，并将其赋值给新文档的主键字段。

func getNextSequence(client *mongo.Client, collectionName string) int {
    collection := client.Database('mydb').Collection('counters')
    filter := bson.M{'_id': collectionName}
    update := bson.M{'$inc': bson.M{'seq': 1}}
    after := options.After
    opt := options.FindOneAndUpdateOptions{ReturnDocument: &after}
    result := collection.FindOneAndUpdate(context.Background(), filter, update, &opt)
    counter := Counter{}
    result.Decode(&counter)
    return counter.Seq
}

func main() {
    client, err := mongo.Connect(context.Background(), options.Client().ApplyURI('mongodb://localhost:27017'))
    if err != nil {
        log.Fatal(err)
    }
    defer client.Disconnect(context.Background())

    collection := client.Database('mydb').Collection('users')
    user := User{
        ID:      getNextSequence(client, 'user_id'),
        Name:    'Alice',
        Age:     25,
        Address: '123 Main St',
    }
    _, err = collection.InsertOne(context.Background(), user)
    if err != nil {
        log.Fatal(err)
    }
}

在上面的代码中，getNextSequence 函数用于获取下一个主键值，它会查询计数器集合，并将seq 字段加 1。然后，在主函数中，创建一个新用户对象，并将其ID 字段设置为getNextSequence 返回的值，然后将其插入到用户集合中。

需要注意的是，由于 MongoDB 是分布式数据库，多个应用程序可能同时访问同一个计数器集合。为了避免出现并发问题，我们可以使用 MongoDB 的乐观锁功能，即在更新计数器记录时指定一个版本号，如果版本号不匹配，则说明有其他应用程序并发更新了计数器记录，此时应该重试获取下一个主键值。