一尘不染

使用Bluebird手动实现pg.connect

node.js

我想在回调中提供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;
};

阅读 272

收藏
2020-07-07

共1个答案

一尘不染

丢掉所有可怕的回调代码,然后在应用程序初始化的某个位置执行此操作:

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模块,就好像它被设计为使用promises开头:

// 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();
         });
});
2020-07-07