我想 promise node-postgres 的 pg.connect
方法以及回调中提供的内部 connection.query
方法。
我可以 .promisify
后者,但我需要手动实现第一个(如果我在这里遗漏了什么,请解释)。
问题是,我不确定这段代码是正确的还是应该改进?代码正在运行,我只是想知道我是否按预期使用 Bluebird。
// aliases
var asPromise = Promise.promisify;
// save reference to original method
var connect = pg.connect.bind(pg);
// promisify method
pg.connect = function (data) {
var deferred = Promise.defer();
connect(data, function promisify(err, connection, release) {
if (err) return deferred.reject(err);
// promisify query factory
connection.query = asPromise(connection.query, connection);
// resolve promised connection
deferred.resolve([connection,release]);
});
return deferred.promise;
};
最佳答案
扔掉所有可怕的回调代码,然后在应用程序初始化的某个地方执行此操作:
var pg = require("pg");
var Promise = require("bluebird");
Object.keys(pg).forEach(function(key) {
var Class = pg[key];
if (typeof Class === "function") {
Promise.promisifyAll(Class.prototype);
Promise.promisifyAll(Class);
}
})
Promise.promisifyAll(pg);
稍后在任何地方你都可以使用 pg 模块,就好像它被设计为使用 promise 一样:
// Later
// Don't even need to require bluebird here
var pg = require("pg");
// Note how it's the pg API but with *Async suffix
pg.connectAsync(...).spread(function(connection, release) {
return connection.queryAsync("...")
.then(function(result) {
console.log("rows", result.rows);
})
.finally(function() {
// Creating a superfluous anonymous function cos I am
// unsure of your JS skill level
release();
});
});
关于node.js - 手动 promise pg.connect 与 Bluebird,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/23881058/