mongodb字段值自增長實現

mongodb字段自增長

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

1.創建計數器集合

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

{
    "_id":1,
    "title": "標題",
    "content": "內容1",
    "type": "類型"
}

為此，創建 counters 集合，序列字段值可以實現自動長：

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可以對文檔的某個值為數字型（只能為滿足要求的數字）的鍵進行增減的操作；

db.collection.findAndModify({
    query: <document>, //定義關於哪些記錄需要修改的選擇標准
    sort: <document>, //確定選擇標准檢索多個文檔時應修改的文檔
    new: <boolean>, //表示將顯示修改后的文檔
    fields: <document>, //指定要返回的字段集
    upsert: <boolean> //如果選擇標准無法檢索文檔，則創建一個新文檔
    remove: <boolean> //為true，query指定的文檔將從數據庫中刪除
)}

 

3.測試

創建測試集合sms：

db.createCollection("sms")

在sms集合中新增文檔，實現_id自增長： 

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

查詢sms集合：

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

4.java實現

java實現以上功能，數據庫驅動版本不同運行效果有差異，僅供參考：

 
private MongoDatabase conn;
 
static{
    this.conn = getDatabase(databaseName);
}
 
/**
 * 連接數據庫
 * @param  databaseName 數據庫名稱
 * @return              數據庫連接對象
 */
private static MongoDatabase getDatabase(databaseName){
    MongoDatabase mongoDatabase = null;
    try{   
        // 連接到 mongodb 服務
        MongoClient mongoClient = new MongoClient( "localhost" , 27017 );
   
        // 連接到數據庫
        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", "標題" + 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());  
    }
}

5.總結

有了字段自增長功能，可以實現訂單流水號、編碼的流水號等功能，可以實現同MySQL自增字段、Oracle序列的相同效果。

 

轉載地址：https://blog.csdn.net/zxw08611/article/details/112555628