如何使用更好的saveAsync扩展Mongoose原型

How to extend Mongooose prototype with better saveAsync

本文关键字：扩展 Mongoose 原型 saveAsync 何使用 更好      更新时间：2023-09-26

我使用猫鼬和用promisifyAll初始化的蓝鸟。这很酷，我喜欢承诺。但是mongoose的model.save()返回了像(err, user, numAffected)一样的函数签名，所以bluebird的model.saveAsync()返回了一个双元素数组，而不是单独返回明显的结果。这使得链接变得不方便，所以我开始制作这样的小助手函数：

function saveUser(user) {
    return user.saveAsync()
        .spread(function (user, numAffected) {
            return user;
        });
}

在mongoose原型中进行承诺时添加它会容易得多，并且在整个应用程序中只需执行一次。。。但我不知道怎么做。

这将在3.0中修复，但现在您可以使用过滤器和自定义承诺器：

// Promisifies only methods named "save"
Promise.promisifyAll(require("mongoose"), {
    filter: function(name) {
        return name === "save"
    },
    promisifier: function(save, defaultPromisifier) {
        var saveAsync = defaultPromisifier(save);
        return function() {
            return saveAsync.apply(this, arguments).spread(function(first) {
                return first;
            });
        }
    }
});
// Promisify rest of the mongoose... it will ignore save as it was already promisified
Promise.promisifyAll(require("mongoose"));

您可以修补user.saveAsync

var user_saveAsync = user.saveAsync;
user.saveAsync = function() {
    return user_saveAsync.bind(this)() //-> same as calling `user.saveAsync()`
        .spread(function(user) {      // but now you're returning spreaded result 
            return user;
        });
}

然后你就可以做了

function saveUser(user) {
    return user.saveAsync();
}
相关文章：