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

时间:2016-09-03 09:21:26

标签: java spring mongodb spring-data spring-data-mongodb

我在域对象@Id private Long id;

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

使用String作为@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

相关:

带字符串的当前代码

import org.springframework.data.annotation.Id;

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

}

1 个答案:

答案 0 :(得分:3)

MongoDB提出了两种处理该方案的方法:auto-increment-optimistic-loop

  
      
  1. 使用专柜收藏
  2.   

创建一个文档以保存序列号

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;
}

在插入文档之前,请获取nextSequence编号:

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);
  
      
  1. 乐观循环
  2.   

创建自定义insertDocument mongodb函数

另一种方法要求您绕过弹簧数据进行插入。这一切都是在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)循环中运行的原因。 / p>

完整的功能:

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);