Mongoose - 如果存在，则在 Subdoc 上增加字段，否则创建新的

Question

我想要做什么。

我有一个userSchema，其中包含operationCountSchemaobjects 的列表。我要做的是创建一个静态方法，如果存在（由month_id 标识）字段，则更新这些操作计数子文档之一上的count 字段。如果当前月份不存在operationCountSchema 文档，则应创建一个新文档。有没有办法在猫鼬中实现这种行为？我曾尝试使用 upsert 无济于事。如何做到这一点？谢谢。

代码

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";
            });
        }
    });
};

让我知道我是否理解了您的问题或没有解决问题。

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
        }
    ]
}

首先，确保子文档存在，如果不只是创建一个使用$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
        }
    ]
}