快捷導(dǎo)航

mongodb字段值自增長實現(xiàn)代碼

更新時間：2021年01月15日 08:26:41 作者：zxw08611

這篇文章主要介紹了mongodb字段值自增長實現(xiàn),本文給大家介紹的非常詳細(xì)，對大家的學(xué)習(xí)或工作具有一定的參考借鑒價值，需要的朋友可以參考下

MongoDB 沒有像 SQL 一樣有自動增長的功能， MongoDB 的 _id 是系統(tǒng)自動生成的12字節(jié)唯一標(biāo)識。但在某些情況下，我們可能需要實現(xiàn) ObjectId 自動增長功能。由于 MongoDB 沒有實現(xiàn)這個功能，我們可以通過編程的方式來實現(xiàn)，以下我們將在 counters 集合中實現(xiàn)_id字段自動增長。

1.創(chuàng)建計數(shù)器集合

期望_id字段從1,2,3,4到n，啟動一個自動遞增的整數(shù)序列，如：

{
  "_id":1,
  "title": "標(biāo)題",
  "content": "內(nèi)容1",
  "type": "類型"
}

為此，創(chuàng)建 counters 集合，序列字段值可以實現(xiàn)自動長：

db.createCollection("counters")

初始化集合，以objId作為主鍵，sequence_value 字段是序列通過自動增長后的一個值：

db.counters.insert({_id:"objId",sequence_value:0})

2.查詢序列號

查詢返回更新后的序列號

db.counters.findAndModify({
  query: {_id: "objId" },
  update: {$inc:{sequence_value:1}},
  new: true
}).sequence_value;

操作符解釋：

$inc可以對文檔的某個值為數(shù)字型（只能為滿足要求的數(shù)字）的鍵進(jìn)行增減的操作；

db.collection.findAndModify({
  query: <document>, //定義關(guān)于哪些記錄需要修改的選擇標(biāo)準(zhǔn)
  sort: <document>, //確定選擇標(biāo)準(zhǔn)檢索多個文檔時應(yīng)修改的文檔
  new: <boolean>, //表示將顯示修改后的文檔
  fields: <document>, //指定要返回的字段集
  upsert: <boolean> //如果選擇標(biāo)準(zhǔn)無法檢索文檔，則創(chuàng)建一個新文檔
  remove: <boolean> //為true，query指定的文檔將從數(shù)據(jù)庫中刪除
)}

3.測試

創(chuàng)建測試集合sms：

db.createCollection("sms")

在sms集合中新增文檔，實現(xiàn)_id自增長：

db.sms.insert({
  _id: db.counters.findAndModify({query:{_id: "objId" },update: {$inc:{sequence_value:1}},"new":true}).sequence_value,
  title: "標(biāo)題1",
  content: "短信1",
  type: "1"
})

查詢sms集合：

db.sms.find({}).sort({_id:1})

4.java實現(xiàn)

java實現(xiàn)以上功能，數(shù)據(jù)庫驅(qū)動版本不同運行效果有差異，僅供參考：

private MongoDatabase conn;
 
static{
  this.conn = getDatabase(databaseName);
}
 
/**
 * 連接數(shù)據(jù)庫
 * @param databaseName 數(shù)據(jù)庫名稱
 * @return       數(shù)據(jù)庫連接對象
 */
private static MongoDatabase getDatabase(databaseName){
  MongoDatabase mongoDatabase = null;
  try{  
    // 連接到 mongodb 服務(wù)
    MongoClient mongoClient = new MongoClient( "localhost" , 27017 );
  
    // 連接到數(shù)據(jù)庫
    MongoDatabase mongoDatabase = mongoClient.getDatabase(databaseName); 
    System.out.println("Connect to database successfully");
    
  }catch(Exception e){
    System.err.println( e.getClass().getName() + ": " + e.getMessage() );
  }
  return mongoDatabase;
}
 
/**
 * 獲取最新序列號
 * @return 序列號
 */
private static int getNextSequenceValue(){
  DBCollection collection = conn.getCollection("counters");
  DBObject query = new BasicDBObject("_id", new BasicDBObject("$eq", "objId"));
  DBObject newDocument =new BasicDBObject();
  newDocument.put("$inc", new BasicDBObject().append("sequence_value", 1));
  newDocument.put("new": true);
  DBObject ret = collection.findAndModify(query, newDocument);
  if (ret == null){
    return 0;
  }else{
    return (Integer)ret.get("sequence_value");
  }
}
 
/**
 * 新增集合文檔
 */
public static void addSms(){
  int id = getNextSequenceValue();
  if(id != 0){
    DBCollection collection = conn.getCollection("sms");
    List<Document> documents = new ArrayList<Document>();
    for(int i = 0; i < 20; i++){
      int id = getNextSequenceValue();
      Document document = new Document("_id", id).
      append("title", "標(biāo)題" + i).
      append("content", "短信" + i). 
      append("type", 1);
      documents.add(document);
    }
    collection.insertMany(documents); 
    System.out.println("文檔插入成功"); 
 
  }
}
 
/**
 * 查詢集合
 */
public static void findSms(){
  DBCollection collection = conn.getCollection("sms");
  FindIterable<Document> findIterable = collection.find(); 
  MongoCursor<Document> mongoCursor = findIterable.iterator(); 
  while(mongoCursor.hasNext()){ 
    System.out.println(mongoCursor.next()); 
  }
}