技術標籤：資料庫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序列的相同效果。