貓鼬 - 在Subdoc上增加字段如果存在，否則創建新的

Question

我正在嘗試做什麼。

我有一個userSchema其中包含一個operationCountSchema對象的列表。我試圖做的是創建一個靜態方法，如果它存在（由month_id標識）字段，則更新count字段中的某個操作計數子文檔。如果operationCountSchema文檔在當月不存在，它應該創建一個新文檔。有沒有辦法在貓鼬中實現這種行爲？我曾嘗試使用upsert無濟於事。如何做到這一點？謝謝。

CODE

var operationCountSchema = mongoose.Schema({ 
    month_id: String, 
    count: { type: Number, default: 0 } 
}, {_id : false}); 

var userSchema = mongoose.Schema({ 
    username : { type: String, unique: true, required: true }, 
    email: { type: String, unique: true, required: true }, 
    password: String, 
    operation_counts: [operationCountSchema] 
}); 

userSchema.statics.incrementOperationCount = function(userID, callback) { 
    var currDate = new Date(); 
    var dateIdentifier = currDate.getFullYear() + "-" + currDate.getMonth(); 
    //NEED TO INCREMENT OPERATION COUNT IF ONE FOR MONTH EXISTS, 
    //ELSE IF IT DOES NOT EXIST, CREATE A NEW ONE. 
} 

而且，在此功能可以實現替代方式中的任何建議，歡迎。

Answer 1

所以你可以有一個mongoose.find()或mongoose.findOne()並檢查子文檔是否存在。如果沒有，我們可以創建一個新對象，如果有，我們可以增加並保存。我在使用mongoose.findOne()。請參閱文檔here。

userSchema.statics.incrementOperationCount = function(userID, callback) { 
var currDate = new Date(); 
var dateIdentifier = currDate.getFullYear() + "-" + currDate.getMonth(); 
    //NEED TO INCREMENT OPERATION COUNT IF ONE FOR MONTH EXISTS, 
    //ELSE IF IT DOES NOT EXIST, CREATE A NEW ONE. 
    operationCountSchema.findOne({'month_id': dateIdentifier}, function(err, subDoc) { 
     // If there is an error in finding the document, catch them 
     if(err) { 
      // Handle errors 
      return err; 
     } 
     // If you find a document, increment the `count` and save 
     if(subDoc) { 
      subDoc.count += 1; 
      subDoc.save(function(err2) { 
       if(err2) { 
        // Handle errors 
        return err2; 
       } else { 
        return "Success"; 
       } 
      }); 
     } 
     // If no document is found, create a new one 
     else { 
      // Populate the values to create the object 
      var data = { 
       "month_id": dateIdentifier, 
       "count": 0 
      }; 
      operationCountSchema.create(data, function(err3, subDoc) { 
       if(err3) { 
        // Handle errors 
        return err3; 
       } 
       // Else return success 
       return "Success"; 
      }); 
     } 
    }); 
}; 

讓我知道如果我有你的問題worng或沒有處理的東西。

Answer 2

我想你想findOneAndUpdate()與upsert : true：

operationCountSchema.findOneAndUpdate({ 
    month_id : dateIdentifier, 
}, { 
    $inc : { count : 1 } 
}, { 
    upsert : true 
}, callback); 

（未經測試）

Answer 3

你可以做的兩個步驟，這裏是mongo shell一個例子：

mongos> db.collection.findOne()  
{ 
    "username" : "mark", 
    "email" : "admin@example.com", 
    "password" : "balalalala", 
    "operation_counts" : [ 
     { 
      "month_id" : "2016-05", 
      "count" : 6 
     } 
    ] 
} 

首先，確保subdoc存在，如果不是隻創建一個使用$addToSet。

mongos> db.collection.update({username:"mark", "operation_counts.month_id": {$ne:"2016-05"}}, {$addToSet: {"operation_counts":{month_id: "2016-05", count:0}}}) 
WriteResult({ "nMatched" : 0, "nUpserted" : 0, "nModified" : 0 }) 
// only update when the subdoc of specified month not exists 
mongos> db.collection.update({username:"mark", "operation_counts.month_id": {$ne:"2016-06"}}, {$addToSet: {"operation_counts":{month_id: "2016-06", count:0}}}) 
WriteResult({ "nMatched" : 1, "nUpserted" : 0, "nModified" : 1 }) 

mongos> db.collection.findOne() 
{ 
    "_id" : ObjectId("575636c21e9b27fe715df654"), 
    "username" : "mark", 
    "email" : "admin@example.com", 
    "password" : "balalalala", 
    "operation_counts" : [ 
     { 
      "month_id" : "2016-05", 
      "count" : 6 
     }, 
     { 
      "month_id" : "2016-06", 
      "count" : 0 
     } 
    ] 
} 

然後，增加count字段。

mongos> db.collection.update({username:"mark", "operation_counts.month_id": "2016-06"}, {$inc:{ "operation_counts.$.count":1 }}) 
WriteResult({ "nMatched" : 1, "nUpserted" : 0, "nModified" : 1 }) 

mongos> db.collection.findOne() 
{ 
    "_id" : ObjectId("575636c21e9b27fe715df654"), 
    "username" : "mark", 
    "email" : "admin@example.com", 
    "password" : "balalalala", 
    "operation_counts" : [ 
     { 
      "month_id" : "2016-05", 
      "count" : 6 
     }, 
     { 
      "month_id" : "2016-06", 
      "count" : 1 
     } 
    ] 
}