数据库
首页 > 数据库> > java-MongoDB使用Spring Data Mongo在集合中具有自动递增的整数ID

java-MongoDB使用Spring Data Mongo在集合中具有自动递增的整数ID

作者:互联网

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

由以下原因引起:org.springframework.dao.InvalidDataAccessApiUsageException:无法为类型为…的实体自动生成类型为java.lang.Long的ID.

可以通过使用String作为@Id私有String ID来解决.

但是,如果我没有多线程(或没有适当地照顾它),也没有分布式MongoDB集群,那么我确实希望Id更加友好,
如何使用Spring Data MongoDB在Java中使整数自动递增MongoDB Id?

使用的版本:

> MongoDB 3.2
> Java mongodb驱动程序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;

}

解决方法:

MongoDB建议使用两种方法来处理这种情况: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;
}

插入文档之前,请获取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. Optimistic Loop

创建一个自定义的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)循环中运行的原因.

完整的功能:

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存储的proc,我相信doEval可能有用:

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

标签:mongodb,spring-data,spring-data-mongodb,spring,java
来源: https://codeday.me/bug/20191118/2026461.html