java - MongoDB 使用 Spring Data Mongo 在集合中自动递增 Integer Id

Question

我在域对象中有 @Id private Long id;

Caused by: org.springframework.dao.InvalidDataAccessApiUsageException: Cannot autogenerate id of type java.lang.Long for entity of type ...

原因:org.springframework.dao.InvalidDataAccessApiUsageException

这可以通过使用 String as @Id private String id;

来解决

但如果我没有多线程(或适当处理它)，也没有分布式 MongoDB 集群，我确实希望和 Id 更友好，
如何使用 Spring Data MongoDB 在 Java 中使整数自动递增 MongoDB Id？

使用的版本:

• MongoDB 3.2
• java mongodb-driver 3.2.2
• spring-data-mongodb 1.9.2.RELEASE

相关:

• Is it bad to change _id type in MongoDB to integer?
• Is it a good practice to have a sequential integer id in mongo

当前带字符串的代码

import org.springframework.data.annotation.Id;

public @lombok.Data class Item {
    @Id private String id;  
    private String name;

}

Answer 1

MongoDB 建议使用 2 种方法来处理这种情况:auto-increment-optimistic-loop

1. Use Counters Collection

创建一个文档来保存您的序列号

db.counters.insert(
   {
      _id: "mySequenceName",
      seq: 0
   }
)

在 mongodb 中创建一个 getNextSequenceNumber javascript 函数

function getNextSequence() {
   var ret = db.counters.findAndModify(
          {
            query: { _id: mySequenceName },
            update: { $inc: { seq: 1 } },
            new: true
          }
   );

   return ret.seq;
}

在插入您的文档之前，获取下一个序列号:

BasicDBObject basicDbObject = new BasicDBObject();
basicDbObject.append( "$eval" , "getNextSequence()" );

CommandResult result = mongoTemplate.executeCommand(basicDbObject);
Object seqno = result.get("seq");

插入您的文档:

Mode model...
model.setId(Long.parseLong(seqno));
repository.save(model);

2. Optimistic Loop

创建自定义 insertDocument mongodb 函数

另一种方法要求您绕过 spring-data 进行插入。这一切都在一个 mongodb javascript 函数中完成(您需要创建)。

该函数将首先检索您要插入的文档的最大 _id:

var cursor = targetCollection.find( {}, { _id: 1 } ).sort( { _id: -1 } ).limit(1);function insertDocument(doc, targetCollection) {

var seq = cursor.hasNext() ? cursor.next()._id + 1 : 1;

然后，您可以使用序列号插入文档:

var results = targetCollection.insert(doc);

但是，由于序列号可能已被另一个同时插入使用，您需要检查错误，并在需要时重复该过程，这就是为什么整个函数在 while(1) 循环。

完整的功能:

function insertDocument(doc, targetCollection) {
    while (1) {

        var cursor = targetCollection.find( {}, { _id: 1 } ).sort( { _id: -1 } ).limit(1);

        var seq = cursor.hasNext() ? cursor.next()._id + 1 : 1;

        doc._id = seq;

        var results = targetCollection.insert(doc);

        if( results.hasWriteError() ) {
            if( results.writeError.code == 11000 /* dup key */ )
                continue;
            else
                print( "unexpected error inserting data: " + tojson( results ) );
        }

        break;
    }
}

调用insertDocument mongodb函数

现在，您可以从 java 调用 javascript 函数 insertDocument。要使用参数调用 mongo 存储过程，我相信 doEval 可能有用:

MongoClient mongoClient = new MongoClient();
DB db = mongoClient.getDB("dbName");
Model m = new Model(...);
CommandResult result = db.doEval("insertDocument", model);